FIX: Import Models in db.py
This commit is contained in:
parent
fc9b8a8f22
commit
875d7ee1cc
4
db.py
4
db.py
@ -1,10 +1,14 @@
|
|||||||
import os
|
import os
|
||||||
from sqlmodel import create_engine, SQLModel, Session
|
from sqlmodel import create_engine, SQLModel, Session
|
||||||
|
from models.station import Station
|
||||||
|
from models.measurement import IndoorMeasurement, OutdoorMeasurement
|
||||||
|
|
||||||
DATA_DIR = "data"
|
DATA_DIR = "data"
|
||||||
os.makedirs(DATA_DIR, exist_ok=True)
|
os.makedirs(DATA_DIR, exist_ok=True)
|
||||||
|
|
||||||
engine = create_engine(f"sqlite:///{DATA_DIR}/database.db")
|
engine = create_engine(f"sqlite:///{DATA_DIR}/database.db")
|
||||||
|
|
||||||
|
|
||||||
SQLModel.metadata.create_all(engine)
|
SQLModel.metadata.create_all(engine)
|
||||||
|
|
||||||
def get_session():
|
def get_session():
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user