""" Defines the table for storing users """ # Import our mapped SQL types from SQLAlchemy from sqlalchemy import Integer, String, DateTime, ARRAY # Import mapping capabilities from the SQLAlchemy ORM from sqlalchemy.orm import Mapped, mapped_column # Import the EntityBase that we are extending from .entity_base import EntityBase # Import datetime for created_at type from datetime import datetime # Import enums for Role and Program from backend.entities.program_enum import ProgramEnum from .user_enum import RoleEnum class UserEntity(EntityBase): """Serves as the database model for User table""" # set table name to user in the database __tablename__ = "user" # set fields or 'columns' for the user table id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True) created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.now) username: Mapped[str] = mapped_column( String(32), nullable=False, default="", unique=True ) role: Mapped[RoleEnum] = mapped_column(RoleEnum, nullable=False) username: Mapped[str] = mapped_column( String(32), nullable=False, default="", unique=True ) role: Mapped[RoleEnum] = mapped_column(RoleEnum, nullable=False) email: Mapped[str] = mapped_column(String(50), nullable=False, unique=True) program: Mapped[list[ProgramEnum]] = mapped_column( ARRAY(ProgramEnum), nullable=False ) program: Mapped[list[ProgramEnum]] = mapped_column( ARRAY(ProgramEnum), nullable=False ) experience: Mapped[int] = mapped_column(Integer, nullable=False) group: Mapped[str] = mapped_column(String(50)) """ @classmethod def from_model(cls, model: User) -> Self: Create a user entity from model Args: model (User): the model to create the entity from Returns: self: The entity return cls( id=model.id, username=model.username, role=model.role, email=model.email, program=model.program, experience=model.experience, group=model.group, ) def to_model(self) -> User: Create a user model from entity Returns: User: A User model for API usage return User( id=self.id, username=self.id, role=self.role, email=self.email, program=self.program, experience=self.experience, group=self.group, ) """