diff --git a/database.py b/database.py index 03eb88c..9742488 100644 --- a/database.py +++ b/database.py @@ -7,16 +7,18 @@ from sqlalchemy.orm import DeclarativeBase load_dotenv() + class Base(DeclarativeBase): pass + database_url = URL.create( - drivername = "mysql+pymysql", - username = os.getenv("MYSQL_USER"), - password = os.getenv("MYSQL_PASSWORD"), - host = os.getenv("MYSQL_HOST"), - port = int(os.getenv("MYSQL_PORT")), - database = os.getenv("MYSQL_DATABASE"), - ) + drivername="mysql+pymysql", + username=os.getenv("MYSQL_USER"), + password=os.getenv("MYSQL_PASSWORD"), + host=os.getenv("MYSQL_HOST"), + port=int(os.getenv("MYSQL_PORT")), + database=os.getenv("MYSQL_DATABASE"), +) engine = create_engine(database_url, pool_pre_ping=True) diff --git a/dependencies.py b/dependencies.py index 6d3937f..789de34 100644 --- a/dependencies.py +++ b/dependencies.py @@ -10,8 +10,5 @@ def get_session(): with Session(engine, expire_on_commit=False) as session: yield session -SessionDep = Annotated[ - Session, - Depends(get_session) - ] +SessionDep = Annotated[Session, Depends(get_session)] diff --git a/models.py b/models.py index 38462d6..8d3878a 100644 --- a/models.py +++ b/models.py @@ -8,17 +8,17 @@ from database import Base class Todo(Base): __tablename__ = "todos" id: Mapped[int] = mapped_column( - BIGINT(unsigned=True), - primary_key = True, - autoincrement = True, - ) + BIGINT(unsigned=True), + primary_key=True, + autoincrement=True, + ) task: Mapped[str] = mapped_column( - String(200), - nullable = False, - ) + String(200), + nullable=False, + ) completion: Mapped[bool] = mapped_column( - Boolean, - nullable = False, - default = False, - server_default = false(), - ) + Boolean, + nullable=False, + default=False, + server_default=false(), + )