Write to postgres DB

Author: MatthewCline-git

src/db/postgres_client.py

@@ -29,8 +29,8 @@ def __init__(self, database_url: str | None = None) -> None:
 
     @contextmanager
     def _get_connection(self) -> Generator[psycopg2.extensions.connection, None, None]:
-        conn = psycopg2.connect(self.database_url)
         try:
+            conn = psycopg2.connect(self.database_url)
             yield conn
         finally:
             conn.close()
@@ -151,7 +151,7 @@ def insert_entry(
                 """,
                 (db_user_id, user_activity_id, date, duration_minutes, raw_input),
             )
-        connection.commit()
+            connection.commit()
 
     def get_user_id_from_telegram(self, telegram_id: int) -> str:
         with self._get_connection() as connection, connection.cursor() as cursor:

src/main.py

@@ -6,7 +6,7 @@
 
 from src.activities.parser import OpenAIActivityParser
 from src.activities.tracker import ActivityTracker
-from src.db.client import SQLiteClient
+from src.db.postgres_client import PostgresClient
 from src.logging_config.logging_config import setup_logging
 from src.messaging.telegram_handler import TelegramHandler
 from src.sheets.client import GoogleSheetsClient
@@ -68,7 +68,7 @@ def main() -> None:
         spreadsheet_id=config["SPREADSHEET_ID"],
     )
 
-    db_client = SQLiteClient()
+    db_client = PostgresClient()
 
     activity_parser = OpenAIActivityParser(api_key=config["OPENAI_API_KEY"], confidence_threshold=0.7)