mirror of
https://github.com/MrSedan/neuro-reply-bot-reworked.git
synced 2026-01-14 13:39:42 +03:00
67 lines
2.5 KiB
Python
67 lines
2.5 KiB
Python
import os
|
|
from datetime import datetime
|
|
from os.path import dirname, join
|
|
from typing import List, Optional
|
|
from uuid import UUID
|
|
|
|
from dotenv import load_dotenv
|
|
from sqlalchemy import Column, DateTime, ForeignKey, create_engine
|
|
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, relationship
|
|
|
|
load_dotenv(join(dirname(__file__), '..', '.env'))
|
|
DATABASE_PASSWORD=os.getenv('DATABASE_PASSWORD')
|
|
DATABASE_NAME=os.getenv('DATABASE_NAME')
|
|
DATABASE_USER=os.getenv('DATABASE_USER')
|
|
DATABASE_PORT=os.getenv('DATABASE_PORT')
|
|
DATABASE_HOST=os.getenv('DATABASE_HOST')
|
|
engine = create_engine(f"postgresql+psycopg2://{DATABASE_USER}:{DATABASE_PASSWORD}@{DATABASE_HOST}:{DATABASE_PORT}/{DATABASE_NAME}", echo=True)
|
|
|
|
|
|
class Base(DeclarativeBase):
|
|
pass
|
|
|
|
class User(Base):
|
|
__tablename__ = 'user'
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True)
|
|
user_name: Mapped[Optional[str]]
|
|
|
|
admin: Mapped['Admin'] = relationship(back_populates='user', cascade='all, delete-orphan')
|
|
|
|
def __repr__(self) -> str:
|
|
return f'User(id={self.id!r}, user_name={self.user_name!r})'
|
|
|
|
def __str__(self) -> str:
|
|
return f'User(id={self.id!r}, user_name={self.user_name!r})'
|
|
|
|
class Admin(Base):
|
|
__tablename__ = 'admin'
|
|
|
|
user_id: Mapped[int] = mapped_column(ForeignKey('user.id'), primary_key=True)
|
|
user: Mapped['User'] = relationship(back_populates='admin')
|
|
posts: Mapped[List['Post']] = relationship(back_populates='user', cascade='all, delete')
|
|
|
|
class Post(Base):
|
|
__tablename__ = 'post'
|
|
|
|
uuid: Mapped[UUID] = mapped_column(primary_key=True)
|
|
posted: Mapped[bool] = mapped_column(default=False)
|
|
from_user_id: Mapped[int] = mapped_column(ForeignKey('admin.user_id'))
|
|
user: Mapped['Admin'] = relationship(back_populates='posts')
|
|
text: Mapped[str]
|
|
media_group_id: Mapped[str]
|
|
images: Mapped[List['Image']] = relationship(back_populates='post', cascade='all, delete')
|
|
timestamp = Column(DateTime, default=datetime.utcnow)
|
|
|
|
def __repr__(self) -> str:
|
|
return f"UUID: {self.uuid}. From_user_id: {self.from_user_id}. media group id: {self.media_group_id}. Text: {self.text}"
|
|
|
|
class Image(Base):
|
|
__tablename__ = 'photo'
|
|
message_id: Mapped[int] = mapped_column(primary_key=True)
|
|
post: Mapped['Post'] = relationship(back_populates='images')
|
|
post_id: Mapped[int] = mapped_column(ForeignKey('post.uuid'))
|
|
file_id: Mapped[str]
|
|
|
|
if __name__ == '__main__':
|
|
Base.metadata.create_all(engine) |