Fix all Service and Service Tag typos and import entities to init

This commit is contained in:
emmalynf 2024-03-25 19:08:38 -04:00
parent d9ae511deb
commit ea8c60f271
5 changed files with 24 additions and 11 deletions

View File

@ -1,2 +1,9 @@
from .entity_base import EntityBase from .entity_base import EntityBase
from .sample_entity import SampleEntity from .sample_entity import SampleEntity
from .tag_entity import TagEntity
from .user_entity import UserEntity
from .resource_entity import ResourceEntity
from .resource_tag_entity import ResourceTagEntity
from .service_entity import ServiceEntity
from .service_tag_entity import ServiceTagEntity

View File

@ -24,7 +24,7 @@ class ProgramEnum(enum.Enum):
COMMUNITY = "COMMUNITY" COMMUNITY = "COMMUNITY"
class ResourceEntity(EntityBase): class ServiceEntity(EntityBase):
#set table name #set table name
__tablename__ = "service" __tablename__ = "service"

View File

@ -10,7 +10,7 @@ from sqlalchemy.orm import Mapped, mapped_column, relationship
from .entity_base import EntityBase from .entity_base import EntityBase
class ResourceTagEntity(EntityBase): class ServiceTagEntity(EntityBase):
# set table name to user in the database # set table name to user in the database
__tablename__ = "serviceTag" __tablename__ = "serviceTag"

View File

@ -26,33 +26,37 @@ 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 @classmethod
def from_model(cls, model: Tag) -> Self: def from_model(cls, model: Tag) -> Self:
"""
Create a user entity from model Create a user entity from model
Args: model (User): the model to create the entity from Args: model (User): the model to create the entity from
Returns: Returns:
self: The entity self: The entity
"""
return cls( return cls(
id=model.id, id=model.id,
content=model.id, content=model.id,
) )
def to_model(self) -> Tag: def to_model(self) -> Tag:
"""
Create a user model from entity Create a user model from entity
Returns: Returns:
User: A User model for API usage User: A User model for API usage
"""
return Tag( return Tag(
id=self.id, id=self.id,
content=self.id, content=self.id,
) )
"""

View File

@ -56,16 +56,17 @@ class UserEntity(EntityBase):
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 @classmethod
def from_model(cls, model: User) -> Self: def from_model(cls, model: User) -> Self:
"""
Create a user entity from model Create a user entity from model
Args: model (User): the model to create the entity from Args: model (User): the model to create the entity from
Returns: Returns:
self: The entity self: The entity
"""
return cls( return cls(
id=model.id, id=model.id,
@ -78,12 +79,12 @@ class UserEntity(EntityBase):
) )
def to_model(self) -> User: def to_model(self) -> User:
"""
Create a user model from entity Create a user model from entity
Returns: Returns:
User: A User model for API usage User: A User model for API usage
"""
return User( return User(
id=self.id, id=self.id,
@ -94,3 +95,4 @@ class UserEntity(EntityBase):
experience=self.experience, experience=self.experience,
group=self.group, group=self.group,
) )
"""