19
db.py
Normal file
19
db.py
Normal file
@@ -0,0 +1,19 @@
|
||||
import sqlite3, os
|
||||
from contextlib import contextmanager
|
||||
|
||||
DB_PATH = os.environ.get("DATABASE_PATH", "readitlater.db")
|
||||
|
||||
def init_db():
|
||||
conn = sqlite3.connect(DB_PATH)
|
||||
conn.executescript(open("schema.sql", "r", encoding="utf-8").read())
|
||||
conn.commit()
|
||||
conn.close()
|
||||
|
||||
@contextmanager
|
||||
def get_db():
|
||||
conn = sqlite3.connect(DB_PATH)
|
||||
conn.row_factory = sqlite3.Row
|
||||
try:
|
||||
yield conn
|
||||
finally:
|
||||
conn.close()
|
||||
Reference in New Issue
Block a user