diff --git a/backup/database/connect.py b/backup/database/connect.py new file mode 100644 index 0000000000000000000000000000000000000000..ae94c0ccdd8b3dcd9cbff6f605604ec1fc3453cd --- /dev/null +++ b/backup/database/connect.py @@ -0,0 +1,26 @@ +import sqlalchemy as sa +from sqlalchemy.ext.declarative import declarative_base +from sqlalchemy import MetaData +from sqlalchemy.orm import sessionmaker, scoped_session +from sqlalchemy_utils import database_exists, create_database + + +def create_db(): + if not database_exists("sqlite:///database.db"): + create_database("sqlite:///database.db") + + +def create_tables(): + engine, base, session, meta = connection() + base.metadata.create_all(engine) + + +base = declarative_base() +meta = MetaData() + + +def connection(): + engine = sa.create_engine("sqlite:///database.db") + base.metadata.bind = engine + session = scoped_session(sessionmaker())(bind=engine) + return engine, base, session, meta