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