Psycopg2 – Return dictionary like values
In this article, we will discuss how to use psycopg2 to return dictionary-like values.
We can return dictionary-like values by using the psycopg2 PostgreSQL driver with and without using the extras module and both will be discussed in this article. psycopg.extras provide Miscellaneous goodies for Psycopg2. It holds helper functions and classes until a better place in distribution is found.
Without using psycopg. extras module:
First import the required packages and form a connection to the PostgreSQL database using the psycopg2.connect() method. and a cursor with the use of the cursor() class, after that we execute the select SQL statement to retrieve rows from the table. The cursor.fetchall() method gives the output as tuples, we use slicing to fetch values and print them out in the form of a dictionary.
Example: Use psycopg2 to return dictionary like values
Python3
import psycopg2
conn = psycopg2.connect(
database = "codes" , user = 'postgres' , password = 'pass' ,
host = '127.0.0.1' , port = '5432'
)
conn.autocommit = True
cursor = conn.cursor()
sql =
cursor.execute(sql)
sql2 =
cursor.execute(sql2)
sql3 =
cursor.execute(sql3)
results = cursor.fetchall()
for row in results:
print ( "code: {}" . format (row[ 0 ]))
print ( "name: {}" . format (row[ 1 ]))
print ( "concatenated_column: {}" . format (row[ 2 ]))
conn.commit()
conn.close()
|
Output :
code: AF
name: Africa
concatenated_column: AFAfrica
code: NA
name: North America
concatenated_column: NANorth America
code: OC
name: Oceania
concatenated_column: OCOceania
code: AN
name: Antartica
concatenated_column: ANAntartica
code: AS
name: Asia
concatenated_column: ASAsia
code: EU
name: Europe
concatenated_column: EUEurope
code: SA
name: South America
concatenated_column: SASouth America
Using psycopg.extras module
RealDictCursor is a cursor that uses a real dict as the base type for rows. This cursor doesn’t allow normal indexing to fetch data. data can be fetched only with the help of keys of column names of the table.
Syntax:
cursor = conn.cursor(cursor_factory = psycopg2.extras.RealDictCursor)
Cursor subclasses are passed as cursor_factory argument to connect() so that the connection’s cursor() method will generate objects of this class. cursor_factory is an argument in the conn.cursor() class.
Syntax:
psycopg2.extras.RealDictCursor(*args, **kwargs)
This cursor used realdict as the base type for rows.
Example: Use psycopg2 to return dictionary like values
Python3
import psycopg2.extras
conn = psycopg2.connect(
database = "codes" , user = 'postgres' , password = 'pass' ,
host = '127.0.0.1' , port = '5432'
)
conn.autocommit = True
cursor = conn.cursor(cursor_factory = psycopg2.extras.RealDictCursor)
sql =
cursor.execute(sql)
sql2 =
cursor.execute(sql2)
sql3 =
cursor.execute(sql3)
results = cursor.fetchall()
for row in results:
print ( "code: {}" . format (row[ 'code' ]))
print ( "name: {}" . format (row[ 'name' ]))
print ( "concatenated_column: {}" . format (row[ 'concatenated_column' ]))
conn.commit()
conn.close()
|
Output:
code: AF
name: Africa
concatenated_column: AFAfrica
code: NA
name: North America
concatenated_column: NANorth America
code: OC
name: Oceania
concatenated_column: OCOceania
code: AN
name: Antartica
concatenated_column: ANAntartica
code: AS
name: Asia
concatenated_column: ASAsia
code: EU
name: Europe
concatenated_column: EUEurope
code: SA
name: South America
concatenated_column: SASouth America
Last Updated :
28 Nov, 2021
Like Article
Save Article
Share your thoughts in the comments
Please Login to comment...