mirror of
https://github.com/cssgunc/compass.git
synced 2025-04-10 06:10:17 -04:00
69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
""" Defines the table for storing resources """
|
|
# Import our mapped SQL types from SQLAlchemy
|
|
from sqlalchemy import Integer, String, DateTime
|
|
# Import mapping capabilities from the SQLAlchemy ORM
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
# 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 Program
|
|
import enum
|
|
from sqlalchemy import Enum
|
|
|
|
class ProgramEnum(enum.Enum):
|
|
""" Determine program for Resource """
|
|
DOMESTIC = 'DOMESTIC'
|
|
ECONOMIC = 'ECONOMIC'
|
|
COMMUNITY = 'COMMUNITY'
|
|
|
|
class ResourceEntity(EntityBase):
|
|
|
|
#set table name
|
|
__tablename__ = "resource"
|
|
|
|
#set fields
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.now)
|
|
name: Mapped[str] = mapped_column(String(32), nullable=False)
|
|
summary: Mapped[str] = mapped_column(String(100), nullable=False)
|
|
link: Mapped[str] = mapped_column(String, nullable=False)
|
|
program: Mapped[ProgramEnum] = mapped_column(Enum(ProgramEnum), nullable=False)
|
|
|
|
#relationships
|
|
resourceTags: Mapped[list["ResourceTagEntity"]] = relationship(back_populates="resource", cascade="all,delete")
|
|
|
|
@classmethod
|
|
def from_model(cls, model: user_model) -> Self:
|
|
"""
|
|
Create a UserEntity from a User model.
|
|
|
|
Args:
|
|
model (User): The model to create the entity from.
|
|
|
|
Returns:
|
|
Self: The entity (not yet persisted).
|
|
"""
|
|
|
|
return cls (
|
|
id = model.id
|
|
created_at = model.created_at
|
|
name = model.name
|
|
summary = model.summary
|
|
link = model.link
|
|
program = model.program
|
|
)
|
|
|
|
def to_model(self) -> user_model:
|
|
return user_model (
|
|
id = self.id
|
|
created_at = self.created_at
|
|
name = self.name
|
|
summary = self.summary
|
|
link = self.link
|
|
program = self.program
|
|
)
|
|
|
|
|
|
|