diff --git a/db.py b/db.py new file mode 100644 index 0000000..c1f2687 --- /dev/null +++ b/db.py @@ -0,0 +1,18 @@ +from sqlalchemy import create_engine +from sqlalchemy.orm import sessionmaker + +from models import Base + +DB_PATH = "sqlite:///wordsearch.db" + + +def get_engine(db_url=None): + return create_engine(db_url or DB_PATH) + + +def get_session_factory(engine): + return sessionmaker(bind=engine) + + +def init_db(engine): + Base.metadata.create_all(engine)