Remove database connection and schema setup from the project

This commit is contained in:
2026-01-13 19:01:18 +00:00
parent 73a19f3ce3
commit 538ea9fe12
3 changed files with 1 additions and 88 deletions

View File

@@ -1,34 +0,0 @@
# To connect to PostgreSQL database
import psycopg2
from psycopg2.extras import RealDictCursor
from typing import Optional
class Database:
def __init__(self, db_name: str, user: str, password: str, host: str = 'localhost', port: int = 5432):
self.connection = psycopg2.connect(
dbname=db_name,
user=user,
password=password,
host=host,
port=port
)
self.connection.autocommit = True
def execute_query(self, query: str, params: Optional[tuple] = None):
with self.connection.cursor(cursor_factory=RealDictCursor) as cursor:
cursor.execute(query, params)
if cursor.description:
return cursor.fetchall()
return []
def execute_many(self, query: str, params_list: list[tuple]):
with self.connection.cursor(cursor_factory=RealDictCursor) as cursor:
cursor.executemany(query, params_list)
def close(self):
self.connection.close()
print("Database connection closed.")
def __exit__(self, exc_type, exc_val, exc_tb):
self.close()