aboutsummaryrefslogtreecommitdiff
path: root/database.py
blob: 544ee0510a3c6f0c3c16846155c915a084d84817 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
import os

# Create 'data' directory at root if it doesn't exist
if not os.path.exists("data"):
    os.makedirs("data")

engine = create_engine("sqlite:///data/data.db")
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

Base = declarative_base()