create base models(users,tasks)

This commit is contained in:
IluaAir
2025-05-31 13:29:23 +03:00
parent b33a04f8fc
commit 1951260788
21 changed files with 1589 additions and 2 deletions

7
src/models/__init__.py Normal file
View File

@@ -0,0 +1,7 @@
from src.models.users import UsersORM
from src.models.tasks import TasksORM
__all__ = [
"UsersORM",
"TasksORM",
]

28
src/models/tasks.py Normal file
View File

@@ -0,0 +1,28 @@
from datetime import date
from typing import Optional, TYPE_CHECKING
from sqlalchemy import ForeignKey, Text, Date, Enum, String
from sqlalchemy.orm import Mapped, mapped_column, relationship
from src.db.database import Base
status_enum = Enum("open", "closed", "in_progress", "todo", name="status_enum")
priority_enum = Enum("low", "medium", "high", "critical", name="priority_enum")
class TasksORM(Base):
if TYPE_CHECKING:
from src.models.users import UsersORM
__tablename__ = "tasks"
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
user_id: Mapped[int] = mapped_column(ForeignKey("users.id"))
title: Mapped[str] = mapped_column(String(100))
description: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
due_date: Mapped[Optional[date]] = mapped_column(Date, nullable=True)
status: Mapped[str] = mapped_column(status_enum, default="open")
priority: Mapped[str] = mapped_column(priority_enum, default="medium")
time_spent: Mapped[int] = mapped_column(default=0)
user: Mapped["Users"] = relationship(back_populates="tasks")

22
src/models/users.py Normal file
View File

@@ -0,0 +1,22 @@
from typing import Optional, TYPE_CHECKING
from fastapi_users_db_sqlalchemy import SQLAlchemyBaseUserTable
from sqlalchemy import String, BigInteger, Integer
from sqlalchemy.orm import Mapped, mapped_column, relationship
from src.db.database import Base
if TYPE_CHECKING:
from src.models.tasks import TasksORM
class UsersORM(SQLAlchemyBaseUserTable[int], Base):
__tablename__ = "users"
id: Mapped[int] = mapped_column(Integer, primary_key=True)
username: Mapped[Optional[str]] = mapped_column(
String(30), nullable=False, unique=True, index=True
)
telegram_id: Mapped[Optional[int]] = mapped_column(BigInteger, nullable=True)
avatar_path: Mapped[Optional[str]] = mapped_column(String(255), nullable=True)
tasks: Mapped[list["TasksORM"]] = relationship(back_populates="user")