database.py

from sqlmodel import Field, SQLModel, create_engine, Session


# model for users
class User(SQLModel, table=True):
    id: int | None = Field(default=None, primary_key=True)
    username: str
    password: str

# model for tasks
class Task(SQLModel, table=True):
    # table=True stores the class metadate in SQLModel.metadata
    id: int | None = Field(default=None,primary_key=True)
    name: str 
    description: str | None = None
    user_id: int = Field(foreign_key="user.id")

sqlite_file_name = "database.sqlite"
sqlite_url = f"sqlite:///{sqlite_file_name}"

# holds network connections to the db
engine = create_engine(sqlite_url)

def create_db_tables():
    SQLModel.metadata.create_all(engine)