39 lines
1004 B
Python
39 lines
1004 B
Python
import psycopg2
|
|
|
|
DATABASE_CONFIG = {
|
|
'dbname': 'messenger',
|
|
'user': 'andrei_cerbu',
|
|
'password': 'andrei',
|
|
'host': 'localhost',
|
|
'port': '5432'
|
|
}
|
|
|
|
|
|
def create_connection():
|
|
try:
|
|
connection = psycopg2.connect(**DATABASE_CONFIG)
|
|
return connection
|
|
except psycopg2.Error as e:
|
|
print(f"Error: Unable to connect to the database\n{e}")
|
|
return None
|
|
|
|
|
|
def execute_query(connection, query, params=None, fetchall=False):
|
|
try:
|
|
with connection.cursor() as cursor:
|
|
cursor.execute(query, params)
|
|
if fetchall:
|
|
result = cursor.fetchall()
|
|
return result
|
|
else:
|
|
connection.commit() # Commit the transaction for INSERT, UPDATE, DELETE queries
|
|
except psycopg2.Error as e:
|
|
print(f"Error: Unable to execute the query\n{e}")
|
|
return None
|
|
|
|
|
|
def close_connection(connection):
|
|
if connection:
|
|
connection.close()
|
|
print("Connection closed.")
|