File size: 1,689 Bytes
ab250f8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
from app.backend.controllers.base_controller import engine
from app.backend.models.base_model import Base
from app.backend.models.chats import Chat
from sqlalchemy.orm import relationship, Session
from sqlalchemy import Column, String
class User(Base):
'''
Base model for users table
'''
__tablename__ = "users"
id = Column("id", String, primary_key=True, unique=True)
language = Column("language", String, default="English", nullable=False)
theme = Column("theme", String, default="light", nullable=False)
chats = relationship("Chat", back_populates="user")
def add_new_user(id: str) -> User:
with Session(autoflush=False, bind=engine, expire_on_commit=False) as db:
new_user = User(id=id)
db.add(new_user)
db.commit()
return new_user
def find_user_by_id(id: str) -> User | None:
with Session(autoflush=False, bind=engine) as db:
return db.query(User).where(User.id == id).first()
def update_user(user: User, language: str = None, theme: str = None) -> None:
with Session(autoflush=False, bind=engine) as db:
user = db.merge(user)
if language:
user.language = language
if theme:
user.theme = theme
db.commit()
def get_user_chats(user: User) -> list[Chat]:
with Session(autoflush=False, bind=engine) as db:
user = db.get(User, user.id)
return user.chats
def get_user_last_chat(user: User) -> Chat | None:
with Session(autoflush=False, bind=engine) as db:
user = db.get(User, user.id)
chats = user.chats
if chats is not None and len(chats):
return chats[-1]
return None
|