Spaces:
Sleeping
Sleeping
File size: 1,978 Bytes
e53c2d7 |
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 |
from sqlalchemy import Column, String, Integer
from sqlalchemy.orm import relationship, Session
from app.backend.models.base_model import Base
from app.backend.controllers.base_controller import engine
class User(Base):
__tablename__ = "users"
id = Column("id", Integer, autoincrement=True, primary_key=True, unique=True)
email = Column("email", String, unique=True, nullable=False)
password_hash = Column("password_hash", String, nullable=False)
language = Column("language", String, default="English", nullable=False)
theme = Column("theme", String, default="light", nullable=False)
access_string_hash = Column("access_string_hash", String, nullable=True)
chats = relationship("Chat", back_populates="user")
def add_new_user(email: str, password_hash: str, access_string_hash: str) -> None:
with Session(autoflush=False, bind=engine) as db:
db.add(User(email=email, password_hash=password_hash, access_string_hash=access_string_hash))
db.commit()
def find_user_by_id(id: int) -> User | None:
with Session(autoflush=False, bind=engine) as db:
return db.query(User).where(User.id == id).first()
def find_user_by_email(email: str) -> User | None:
with Session(autoflush=False, bind=engine) as db:
return db.query(User).where(User.email == email).first()
def find_user_by_access_string(access_string_hash: str) -> User | None:
with Session(autoflush=False, bind=engine) as db:
return db.query(User).where(User.access_string_hash == access_string_hash).first()
def update_user(user: User, language: str = None, theme: str = None, access_string_hash: 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
if access_string_hash:
user.access_string_hash = access_string_hash
db.commit()
|