88 lines
4.3 KiB
Python
88 lines
4.3 KiB
Python
from sqlalchemy import Column, Integer, String, Boolean, Float, DateTime, Date, ForeignKey, Enum, UniqueConstraint
|
||
from fastapi import Depends
|
||
from .db import Base, engine
|
||
from sqlalchemy.orm import relationship
|
||
|
||
|
||
# импортируем модуль для создания перечислений
|
||
import enum
|
||
|
||
# # класс, необходимый для создания перечисления
|
||
# class State(enum.Enum):
|
||
# published = 1
|
||
# taken = 2
|
||
# obsolete = 3
|
||
|
||
|
||
class User(Base):#класс пользователя
|
||
__tablename__ = "users"
|
||
|
||
id = Column(Integer, primary_key=True, index=True, unique=True)#айди пользователя
|
||
email = Column(String)#электронная почта пользователя
|
||
hashed_password = Column(String)
|
||
name = Column(String, nullable=True)#имя пользователя
|
||
surname = Column(String)#фамилия пользователя
|
||
disabled = Column(Boolean, default=False)
|
||
rating = Column(Integer, default=0) #баллы пользователя
|
||
num_of_ratings = Column(Integer, default=0) # количество оценок (т.е. то, сколько раз другие пользователи оценили текущего)
|
||
reg_date = Column(Date) # дата регистрации
|
||
|
||
announcements = relationship("Announcement", back_populates="user")
|
||
|
||
class Announcement(Base): #класс объявления
|
||
__tablename__ = "announcements"
|
||
|
||
id = Column(Integer, primary_key=True, index=True) # айди объявления
|
||
user_id = Column(Integer, ForeignKey("users.id")) # айди создателя объявления
|
||
name = Column(String) # название объявления
|
||
category = Column(String) #категория продукта из объявления
|
||
best_by = Column(Date) #срок годности продукта из объявления
|
||
address = Column(String)
|
||
longtitude = Column(Integer)
|
||
latitude = Column(Integer)
|
||
description = Column(String) #описание продукта в объявлении
|
||
src = Column(String, nullable=True) #изображение продукта в объявлении
|
||
metro = Column(String) #ближайщее метро от адреса нахождения продукта
|
||
trashId = Column(Integer, nullable=True)
|
||
booked_by = Column(Integer) #статус бронирования (либо -1, либо айди бронирующего)
|
||
# state = Column(Enum(State), default=State.published) # состояние объявления (опубликовано, забронировано, устарело)
|
||
obsolete = Column(Boolean, default=False) # состояние объявления (по-умолчанию считаем его актуальным)
|
||
|
||
user = relationship("User", back_populates="announcements")
|
||
|
||
|
||
class Trashbox(Base): #класс мусорных баков
|
||
__tablename__ = "trashboxes"
|
||
|
||
id = Column(Integer, primary_key=True, index=True)#айди
|
||
name = Column(String, nullable=True)#имя пользователя
|
||
address = Column(String)
|
||
latitude = Column(Integer)
|
||
longtitude = Column(Integer)
|
||
category = Column(String) #категория продукта из объявления
|
||
|
||
|
||
class Poems(Base):#класс поэзии
|
||
__tablename__ = "poems"
|
||
|
||
id = Column(Integer, primary_key=True, index=True) #айди
|
||
title = Column(String) # название стихотворения
|
||
text = Column(String) # текст стихотворения
|
||
author = Column(String) # автор стихотворения
|
||
|
||
# from typing import AsyncGenerator
|
||
# from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
|
||
# from fastapi_users.db import SQLAlchemyBaseUserTableUUID, SQLAlchemyUserDatabase
|
||
# # This function can be called during the initialization of the FastAPI app.
|
||
# async def create_db_and_tables():
|
||
# async with engine.begin() as conn:
|
||
# await conn.run_sync(Base.metadata.create_all)
|
||
|
||
|
||
# async def get_async_session() -> AsyncGenerator[AsyncSession, None]:
|
||
# async with async_session_maker() as session:
|
||
# yield session
|
||
|
||
|
||
# async def get_user_db(session: AsyncSession = Depends(get_async_session)):
|
||
# yield SQLAlchemyUserDatabase(session, User) |