This commit is contained in:
Cyberes 2024-03-06 14:28:20 -07:00
parent 94aadf1246
commit a9d6ef6783
2 changed files with 7 additions and 2 deletions

View File

@ -3,3 +3,4 @@ chardet==5.2.0
humanize==4.9.0
mmh3==4.1.0
flask==3.0.2
python-magic==0.4.27

View File

@ -1,4 +1,5 @@
import json
import os
import sqlite3
from datetime import datetime
from pathlib import Path
@ -8,9 +9,12 @@ from flask import Flask, render_template, send_from_directory
app = Flask(__name__)
if not os.environ.get('SQLITE3_DB'):
raise Exception('SQLITE3_DB not set.')
def get_db_connection():
conn = sqlite3.connect('emails.db')
conn = sqlite3.connect(os.environ.get('SQLITE3_DB'))
conn.row_factory = sqlite3.Row
return conn