1
0
mirror of https://github.com/ilri/dspace-statistics-api.git synced 2024-11-25 07:40:17 +01:00

database.py: Use psycopg2.extras.DictCursor

This allows us to access records using their column name. I didn't
notice that this was not working, as I had been testing the wrong
server!

See: http://initd.org/psycopg/docs/extras.html
This commit is contained in:
Alan Orth 2018-09-25 02:03:49 +03:00
parent 44884140e5
commit 6e4bc630f7
Signed by: alanorth
GPG Key ID: 0FB860CC9C45B1B9

View File

@ -3,9 +3,10 @@ from config import DATABASE_USER
from config import DATABASE_PASS from config import DATABASE_PASS
from config import DATABASE_HOST from config import DATABASE_HOST
import psycopg2 import psycopg2
import psycopg2.extras
def database_connection(): def database_connection():
connection = psycopg2.connect("dbname={} user={} password={} host='{}'".format(DATABASE_NAME, DATABASE_USER, DATABASE_PASS, DATABASE_HOST)) connection = psycopg2.connect("dbname={} user={} password={} host='{}'".format(DATABASE_NAME, DATABASE_USER, DATABASE_PASS, DATABASE_HOST), cursor_factory=psycopg2.extras.DictCursor)
return connection return connection