20 lines
442 B
Python
20 lines
442 B
Python
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()
|