54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
import datetime as dt
|
|
|
|
from sqlalchemy import BigInteger, DateTime, ForeignKey, create_engine, func
|
|
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, sessionmaker
|
|
|
|
|
|
class Base(DeclarativeBase): ...
|
|
|
|
|
|
class User(Base):
|
|
__tablename__ = "user"
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
fullname: Mapped[str] = mapped_column()
|
|
tg_id: Mapped[int] = mapped_column(BigInteger)
|
|
phone: Mapped[str] = mapped_column(nullable=True)
|
|
|
|
|
|
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.date] = mapped_column(nullable=True)
|
|
end_date: Mapped[dt.date] = mapped_column(nullable=True)
|
|
|
|
|
|
class UserEvent(Base):
|
|
__tablename__ = "user_event"
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
user: Mapped[int] = mapped_column(ForeignKey("user.id"))
|
|
event: Mapped[int] = mapped_column(ForeignKey("event.id"))
|
|
date: Mapped[dt.date] = mapped_column(
|
|
DateTime(timezone=True), server_default=func.now()
|
|
)
|
|
status: Mapped[str] = mapped_column(default="not confirmed")
|
|
|
|
if __name__ == "__main__":
|
|
engine = create_engine("sqlite:///test.db")
|
|
SessionMaker = sessionmaker(engine)
|
|
Base.metadata.drop_all(engine)
|
|
Base.metadata.create_all(engine)
|
|
|
|
with SessionMaker() as session:
|
|
user = User(fullname = "Shevkunov Daniil", tg_id=12023)
|
|
event = Event(title="Event")
|
|
|
|
session.add_all([user, event])
|
|
session.flush()
|
|
user_event = UserEvent(user=user.id, event=event.id)
|
|
session.add(user_event)
|
|
session.commit()
|
|
# with SessionMaker as session:
|
|
|
|
|