73 lines
2.7 KiB
Python
73 lines
2.7 KiB
Python
import datetime as dt
|
|
|
|
from sqlalchemy import BigInteger, DateTime, ForeignKey, UniqueConstraint, func
|
|
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, relationship
|
|
|
|
from .enums import EventUserStatus, UserRole
|
|
|
|
|
|
class Base(DeclarativeBase):
|
|
pass
|
|
|
|
|
|
class User(Base):
|
|
__tablename__ = "user"
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
tg_id: Mapped[int] = mapped_column(BigInteger, unique=True)
|
|
fullname: Mapped[str] = mapped_column()
|
|
role: Mapped[str] = mapped_column(default=UserRole.USER.value)
|
|
phone: Mapped[str] = mapped_column(nullable=True)
|
|
address: Mapped[str] = mapped_column(nullable=True)
|
|
|
|
events = relationship("UserEvent", back_populates="user_rel")
|
|
certificates = relationship("Certificate", back_populates="user_rel")
|
|
|
|
|
|
class Event(Base):
|
|
__tablename__ = "event"
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
title: Mapped[str] = mapped_column()
|
|
description: Mapped[str] = mapped_column(default="")
|
|
start_date: Mapped[dt.datetime] = mapped_column(nullable=True)
|
|
end_date: Mapped[dt.datetime] = mapped_column(nullable=True)
|
|
created_at: Mapped[dt.datetime] = mapped_column(
|
|
DateTime(timezone=True), server_default=func.now()
|
|
)
|
|
is_active: Mapped[bool] = mapped_column(default=True)
|
|
|
|
user_events = relationship("UserEvent", back_populates="event_rel")
|
|
|
|
|
|
class UserEvent(Base):
|
|
__tablename__ = "user_event"
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
user_id: Mapped[int] = mapped_column(ForeignKey("user.id"))
|
|
event_id: Mapped[int] = mapped_column(ForeignKey("event.id"))
|
|
date: Mapped[dt.datetime] = mapped_column(
|
|
DateTime(timezone=True), server_default=func.now()
|
|
)
|
|
status: Mapped[str] = mapped_column(default=EventUserStatus.NOT_CONFIRMED.value)
|
|
|
|
user_rel = relationship("User", back_populates="events")
|
|
event_rel = relationship("Event", back_populates="user_events")
|
|
|
|
__table_args__ = (UniqueConstraint("user_id", "event_id", name="uq_user_event"),)
|
|
|
|
|
|
class Course(Base):
|
|
__tablename__ = "course"
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
title: Mapped[str] = mapped_column()
|
|
description: Mapped[str] = mapped_column(nullable=True)
|
|
|
|
certificates = relationship("Certificate", back_populates="course_rel")
|
|
|
|
|
|
class Certificate(Base):
|
|
__tablename__ = "certificate"
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
user_id: Mapped[int] = mapped_column(ForeignKey("user.id"))
|
|
course_id: Mapped[int] = mapped_column(ForeignKey("course.id"))
|
|
date: Mapped[dt.datetime] = mapped_column()
|
|
__table_args__ = (UniqueConstraint("user_id", "course_id", name="uq_user_course"),)
|