mirror of
https://github.com/cssgunc/compass.git
synced 2025-04-10 06:10:17 -04:00
Create to and from model methods for APIs for User and Tag Entities
This commit is contained in:
parent
f96beee4b2
commit
f5de6015bf
|
@ -38,4 +38,4 @@ class ResourceEntity(EntityBase):
|
||||||
program: Mapped[ProgramEnum] = mapped_column(Enum(ProgramEnum), nullable=False)
|
program: Mapped[ProgramEnum] = mapped_column(Enum(ProgramEnum), nullable=False)
|
||||||
|
|
||||||
#relationships
|
#relationships
|
||||||
resourceTags: Mapped[list["ServiceTagEntity"]] = relationship(back_populates="service", cascade="all,delete")
|
serivceTags: Mapped[list["ServiceTagEntity"]] = relationship(back_populates="service", cascade="all,delete")
|
||||||
|
|
|
@ -26,3 +26,33 @@ class TagEntity(EntityBase):
|
||||||
resourceTags: Mapped[list["ResourceTagEntity"]] = relationship(back_populates="tag", cascade="all,delete")
|
resourceTags: Mapped[list["ResourceTagEntity"]] = relationship(back_populates="tag", cascade="all,delete")
|
||||||
serviceTags: Mapped[list["ServiceTagEntity"]] = relationship(back_populates="tag", cascade="all,delete")
|
serviceTags: Mapped[list["ServiceTagEntity"]] = relationship(back_populates="tag", cascade="all,delete")
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def from_model(cls, model: Tag) -> 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,
|
||||||
|
content=model.id,
|
||||||
|
)
|
||||||
|
|
||||||
|
def to_model(self) -> Tag:
|
||||||
|
"""
|
||||||
|
Create a user model from entity
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
User: A User model for API usage
|
||||||
|
"""
|
||||||
|
|
||||||
|
return Tag(
|
||||||
|
id=self.id,
|
||||||
|
content=self.id,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -2,31 +2,42 @@
|
||||||
|
|
||||||
# Import our mapped SQL types from SQLAlchemy
|
# Import our mapped SQL types from SQLAlchemy
|
||||||
from sqlalchemy import Integer, String, DateTime, ARRAY
|
from sqlalchemy import Integer, String, DateTime, ARRAY
|
||||||
|
|
||||||
# Import mapping capabilities from the SQLAlchemy ORM
|
# Import mapping capabilities from the SQLAlchemy ORM
|
||||||
from sqlalchemy.orm import Mapped, mapped_column
|
from sqlalchemy.orm import Mapped, mapped_column
|
||||||
|
|
||||||
# Import the EntityBase that we are extending
|
# Import the EntityBase that we are extending
|
||||||
from .entity_base import EntityBase
|
from .entity_base import EntityBase
|
||||||
|
|
||||||
# Import datetime for created_at type
|
# Import datetime for created_at type
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
|
||||||
# Import enums for Role and Program
|
# Import enums for Role and Program
|
||||||
import enum
|
import enum
|
||||||
from sqlalchemy import Enum
|
from sqlalchemy import Enum
|
||||||
|
|
||||||
|
# Import self
|
||||||
|
from typing import Self
|
||||||
|
|
||||||
|
|
||||||
class RoleEnum(enum.Enum):
|
class RoleEnum(enum.Enum):
|
||||||
""" Determine role for User """
|
"""Determine role for User"""
|
||||||
ADMIN = 'ADMIN'
|
|
||||||
EMPLOYEE = 'EMPLOYEE'
|
ADMIN = "ADMIN"
|
||||||
VOLUNTEER = 'VOLUNTEER'
|
EMPLOYEE = "EMPLOYEE"
|
||||||
|
VOLUNTEER = "VOLUNTEER"
|
||||||
|
|
||||||
|
|
||||||
class ProgramEnum(enum.Enum):
|
class ProgramEnum(enum.Enum):
|
||||||
"""Determine program for User """
|
"""Determine program for User"""
|
||||||
DOMESTIC = 'DOMESTIC'
|
|
||||||
ECONOMIC = 'ECONOMIC'
|
DOMESTIC = "DOMESTIC"
|
||||||
COMMUNITY = 'COMMUNITY'
|
ECONOMIC = "ECONOMIC"
|
||||||
|
COMMUNITY = "COMMUNITY"
|
||||||
|
|
||||||
|
|
||||||
class UserEntity(EntityBase):
|
class UserEntity(EntityBase):
|
||||||
"""Serves as the databse model for User table"""
|
"""Serves as the database model for User table"""
|
||||||
|
|
||||||
# set table name to user in the database
|
# set table name to user in the database
|
||||||
__tablename__ = "user"
|
__tablename__ = "user"
|
||||||
|
@ -34,10 +45,52 @@ class UserEntity(EntityBase):
|
||||||
# set fields or 'columns' for the user table
|
# set fields or 'columns' for the user table
|
||||||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.now)
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.now)
|
||||||
username: Mapped[str] = mapped_column(String(32), nullable=False, default="", unique=True)
|
username: Mapped[str] = mapped_column(
|
||||||
role: Mapped[RoleEnum] = mapped_column(Enum(RoleEnum), nullable=False)
|
String(32), nullable=False, default="", unique=True
|
||||||
|
)
|
||||||
|
role: Mapped[RoleEnum] = mapped_column(Enum(RoleEnum), nullable=False)
|
||||||
email: Mapped[str] = mapped_column(String(50), nullable=False, unique=True)
|
email: Mapped[str] = mapped_column(String(50), nullable=False, unique=True)
|
||||||
program: Mapped[list[ProgramEnum]] = mapped_column(ARRAY(Enum(ProgramEnum)), nullable=False)
|
program: Mapped[list[ProgramEnum]] = mapped_column(
|
||||||
|
ARRAY(Enum(ProgramEnum)), nullable=False
|
||||||
|
)
|
||||||
experience: Mapped[int] = mapped_column(Integer, nullable=False)
|
experience: Mapped[int] = mapped_column(Integer, nullable=False)
|
||||||
group: Mapped[str] = mapped_column(String(50))
|
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,
|
||||||
|
)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user