Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import psycopg2.extras
- import os
- def connect_db(autocommit=False):
- """ Connects to a data base and returns a connection and a cursor. """
- conn = psycopg2.connect(
- database=os.environ.get('DATABASE_DBNAME'),
- user=os.environ.get('DATABASE_USERNAME'),
- host=os.environ.get('DATABASE_HOST'),
- password=os.environ.get('DATABASE_PASSWORD')
- )
- conn.autocommit = autocommit
- cursor = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
- return {'conn': conn, 'cursor': cursor}
- def execute_query(a_query, autocommit=True):
- """ Executes a query, printing out rows returned if any, returning the results, the connection and the cursor. """
- db = connect_db(autocommit=autocommit)
- conn, cursor = db.get('conn'), db.get('cursor')
- cursor.execute(a_query)
- result = cursor.fetchall()
- if result:
- print(len(result), "records fetched")
- return {'result':result, 'conn':conn, 'cursor':cursor}
- if if __name__ == '__main__':
- q = """ Select * from table limit 1 """
- res = execute_query(q)
- print(res)
Add Comment
Please, Sign In to add comment