(feat): Changed from SQLite3 to PostgreSQL

database.py

@@ -1,41 +1,52 @@
-import sqlite3
 import os
+import psycopg2
+from psycopg2 import IntegrityError
 from dotenv import load_dotenv
 
 load_dotenv()
 
-DATABASE_URL = os.getenv("DATABASE_FILE")
+def get_connection():
+    """Return a database connection."""
+    return psycopg2.connect(
+        host=os.getenv("PG_HOST"),
+        port=os.getenv("PG_PORT"),
+        dbname=os.getenv("PG_DATABASE"),
+        user=os.getenv("PG_USER"),
+        password=os.getenv("PG_PASSWORD"),
+        connect_timeout=10
+    )
 
 def init_db():
-    conn = sqlite3.connect(DATABASE_URL)
+    conn = get_connection()
     cursor = conn.cursor()
     cursor.execute("""
         CREATE TABLE IF NOT EXISTS subscribers (
-            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            id SERIAL PRIMARY KEY,
             email TEXT UNIQUE NOT NULL
         )
     """)
     conn.commit()
+    cursor.close()
     conn.close()
 
 def add_email(email):
     try:
-        with sqlite3.connect(DATABASE_URL, timeout=10) as conn:
+        with get_connection() as conn:
-            cursor = conn.cursor()
+            with conn.cursor() as cursor:
-            cursor.execute("INSERT INTO subscribers (email) VALUES (?)", (email,))
+                cursor.execute("INSERT INTO subscribers (email) VALUES (%s)", (email,))
                 conn.commit()
         return True
-    except sqlite3.IntegrityError:
+    except IntegrityError:
         return False
-    except sqlite3.OperationalError as e:
+    except psycopg2.OperationalError as e:
-        print(f"Operational Error: {e}")
+        print(f"Error: {e}")
         return False
 
 def remove_email(email):
     try:
-        with sqlite3.connect(DATABASE_URL, timeout=10) as conn:
+        with get_connection() as conn:
-            cursor = conn.cursor()
+            with conn.cursor() as cursor:
-            cursor.execute("DELETE FROM subscribers WHERE email = ?", (email,))
+                cursor.execute("DELETE FROM subscribers WHERE email = %s", (email,))
                 conn.commit()
                 if cursor.rowcount > 0:
                     return True

requirements.txt

@@ -1,2 +1,3 @@
 flask
 python-dotenv
+psycopg2-binary