From 0a86fde91576441be68d0a58f4614daffea91728 Mon Sep 17 00:00:00 2001
From: Blake Ridgway <blake@blakeridgway.com>
Date: Tue, 11 Feb 2025 20:55:54 -0600
Subject: [PATCH] (feat): Added dotenv and updated the database to run with
 DATABASE_URL

---
 database.py      | 10 ++++++++--
 requirements.txt |  3 ++-
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/database.py b/database.py
index b62eea4..8271efc 100644
--- a/database.py
+++ b/database.py
@@ -1,7 +1,13 @@
 import sqlite3
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+DATABASE_URL = os.getenv("DATABASE_URL")
 
 def init_db():
-    conn = sqlite3.connect('subscribers.db')
+    conn = sqlite3.connect(DATABASE_URL)
     cursor = conn.cursor()
     cursor.execute("""
         CREATE TABLE IF NOT EXISTS subscribers (
@@ -14,7 +20,7 @@ def init_db():
 
 def add_email(email):
     try:
-        conn = sqlite3.connect('subscribers.db')
+        conn = sqlite3.connect(DATABASE_URL)
         cursor = conn.cursor()
         cursor.execute("INSERT INTO subscribers (email) VALUES (?)", (email,))
         conn.commit()
diff --git a/requirements.txt b/requirements.txt
index 8ab6294..5a4a2b9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1 +1,2 @@
-flask
\ No newline at end of file
+flask
+python-dotenv
\ No newline at end of file