30 lines
No EOL
717 B
Python
30 lines
No EOL
717 B
Python
import sqlite3
|
|
import os
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
DATABASE_URL = os.getenv("DATABASE_URL")
|
|
|
|
def init_db():
|
|
conn = sqlite3.connect(DATABASE_URL)
|
|
cursor = conn.cursor()
|
|
cursor.execute("""
|
|
CREATE TABLE IF NOT EXISTS subscribers (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
email TEXT UNIQUE NOT NULL
|
|
)
|
|
""")
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
def add_email(email):
|
|
try:
|
|
conn = sqlite3.connect(DATABASE_URL)
|
|
cursor = conn.cursor()
|
|
cursor.execute("INSERT INTO subscribers (email) VALUES (?)", (email,))
|
|
conn.commit()
|
|
conn.close()
|
|
return True
|
|
except sqlite3.IntegrityError:
|
|
return False |