mirror of
https://github.com/cssgunc/compass.git
synced 2025-04-17 17:30:17 -04:00
Create entites for each table to setup schema
This commit is contained in:
parent
035ab5fadb
commit
b8903adeb8
38
backend/entities/resource_entity.py
Normal file
38
backend/entities/resource_entity.py
Normal file
|
@ -0,0 +1,38 @@
|
||||||
|
""" Defines the table for storing resources """
|
||||||
|
# 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, 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")
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
32
backend/entities/resource_tag_entity.py
Normal file
32
backend/entities/resource_tag_entity.py
Normal file
|
@ -0,0 +1,32 @@
|
||||||
|
""" Defines the table for resource tags """
|
||||||
|
|
||||||
|
# Import our mapped SQL types from SQLAlchemy
|
||||||
|
from sqlalchemy import ForeignKey, Integer, String, DateTime, ARRAY
|
||||||
|
|
||||||
|
# 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 Role and Program
|
||||||
|
import enum
|
||||||
|
from sqlalchemy import Enum
|
||||||
|
|
||||||
|
|
||||||
|
class ResourceTagEntity(EntityBase):
|
||||||
|
|
||||||
|
# set table name to user in the database
|
||||||
|
__tablename__ = "resourceTag"
|
||||||
|
|
||||||
|
# set fields or 'columns' for the user table
|
||||||
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||||||
|
resourceId: Mapped[int] = mapped_column(ForeignKey("event.id"), primary_key=True)
|
||||||
|
tagId: Mapped[int] = mapped_column(ForeignKey("user.pid"), primary_key=True)
|
||||||
|
|
||||||
|
# relationships
|
||||||
|
resource: Mapped["ResourceEntity"] = relationship(back_populates="resourceTags")
|
||||||
|
tag: Mapped["TagEntity"] = relationship(back_populates="resourceTags")
|
41
backend/entities/serivce_entity.py
Normal file
41
backend/entities/serivce_entity.py
Normal file
|
@ -0,0 +1,41 @@
|
||||||
|
""" Defines the table for storing services """
|
||||||
|
|
||||||
|
# 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, 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 Service"""
|
||||||
|
DOMESTIC = "DOMESTIC"
|
||||||
|
ECONOMIC = "ECONOMIC"
|
||||||
|
COMMUNITY = "COMMUNITY"
|
||||||
|
|
||||||
|
|
||||||
|
class ResourceEntity(EntityBase):
|
||||||
|
|
||||||
|
#set table name
|
||||||
|
__tablename__ = "service"
|
||||||
|
|
||||||
|
#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)
|
||||||
|
requirements: Mapped[list[str]] = mapped_column(ARRAY(String))
|
||||||
|
program: Mapped[ProgramEnum] = mapped_column(Enum(ProgramEnum), nullable=False)
|
||||||
|
|
||||||
|
#relationships
|
||||||
|
resourceTags: Mapped[list["ServiceTagEntity"]] = relationship(back_populates="service", cascade="all,delete")
|
25
backend/entities/service_tag_entity.py
Normal file
25
backend/entities/service_tag_entity.py
Normal file
|
@ -0,0 +1,25 @@
|
||||||
|
""" Defines the table for service tags """
|
||||||
|
|
||||||
|
# Import our mapped SQL types from SQLAlchemy
|
||||||
|
from sqlalchemy import ForeignKey, Integer
|
||||||
|
|
||||||
|
# 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
|
||||||
|
|
||||||
|
|
||||||
|
class ResourceTagEntity(EntityBase):
|
||||||
|
|
||||||
|
# set table name to user in the database
|
||||||
|
__tablename__ = "serviceTag"
|
||||||
|
|
||||||
|
# set fields or 'columns' for the user table
|
||||||
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||||||
|
sericeId: Mapped[int] = mapped_column(ForeignKey("event.id"), primary_key=True)
|
||||||
|
tagId: Mapped[int] = mapped_column(ForeignKey("user.pid"), primary_key=True)
|
||||||
|
|
||||||
|
# relationships
|
||||||
|
service: Mapped["ServiceEntity"] = relationship(back_populates="resourceTags")
|
||||||
|
tag: Mapped["TagEntity"] = relationship(back_populates="resourceTags")
|
28
backend/entities/tag_entity.py
Normal file
28
backend/entities/tag_entity.py
Normal file
|
@ -0,0 +1,28 @@
|
||||||
|
""" Defines the table for storing tags """
|
||||||
|
|
||||||
|
# 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
|
||||||
|
|
||||||
|
class TagEntity(EntityBase):
|
||||||
|
|
||||||
|
#set table name
|
||||||
|
__tablename__ = "tag"
|
||||||
|
|
||||||
|
#set fields
|
||||||
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||||||
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.now)
|
||||||
|
content: Mapped[str] = mapped_column(String(100), nullable=False)
|
||||||
|
|
||||||
|
#relationships
|
||||||
|
resourceTags: Mapped[list["ResourceTagEntity"]] = relationship(back_populates="tag", cascade="all,delete")
|
||||||
|
serviceTags: Mapped[list["ServiceTagEntity"]] = relationship(back_populates="tag", cascade="all,delete")
|
||||||
|
|
43
backend/entities/user_entity.py
Normal file
43
backend/entities/user_entity.py
Normal file
|
@ -0,0 +1,43 @@
|
||||||
|
""" 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
|
||||||
|
import enum
|
||||||
|
from sqlalchemy import Enum
|
||||||
|
|
||||||
|
class RoleEnum(enum.Enum):
|
||||||
|
""" Determine role for User """
|
||||||
|
ADMIN = 'ADMIN'
|
||||||
|
EMPLOYEE = 'EMPLOYEE'
|
||||||
|
VOLUNTEER = 'VOLUNTEER'
|
||||||
|
|
||||||
|
class ProgramEnum(enum.Enum):
|
||||||
|
"""Determine program for User """
|
||||||
|
DOMESTIC = 'DOMESTIC'
|
||||||
|
ECONOMIC = 'ECONOMIC'
|
||||||
|
COMMUNITY = 'COMMUNITY'
|
||||||
|
|
||||||
|
|
||||||
|
class UserEntity(EntityBase):
|
||||||
|
"""Serves as the databse 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(Enum(RoleEnum), nullable=False)
|
||||||
|
email: Mapped[str] = mapped_column(String(50), nullable=False, unique=True)
|
||||||
|
program: Mapped[list[ProgramEnum]] = mapped_column(ARRAY(Enum(ProgramEnum)), nullable=False)
|
||||||
|
experience: Mapped[int] = mapped_column(Integer, nullable=False)
|
||||||
|
group: Mapped[str] = mapped_column(String(50))
|
||||||
|
|
19
backend/test/entities/tag_entity_test.py
Normal file
19
backend/test/entities/tag_entity_test.py
Normal file
|
@ -0,0 +1,19 @@
|
||||||
|
""" Testing Tag Entity """
|
||||||
|
|
||||||
|
from sqlalchemy import Engine
|
||||||
|
from ... import entities
|
||||||
|
from ...entities.tag_entity import TagEntity
|
||||||
|
|
||||||
|
|
||||||
|
def test_add_sample_data_tag(session: Engine):
|
||||||
|
|
||||||
|
"""Inserts a sample data point and verifies it is in the database"""
|
||||||
|
entity = TagEntity(content="Test tag")
|
||||||
|
session.add(entity)
|
||||||
|
session.commit()
|
||||||
|
data = session.get(TagEntity, 1)
|
||||||
|
assert data.id == 1
|
||||||
|
assert data.content == "Test tag"
|
||||||
|
|
||||||
|
|
||||||
|
|
24
backend/test/entities/user_entity_test.py
Normal file
24
backend/test/entities/user_entity_test.py
Normal file
|
@ -0,0 +1,24 @@
|
||||||
|
""" Testing User Entity """
|
||||||
|
|
||||||
|
from sqlalchemy import Engine
|
||||||
|
from ... import entities
|
||||||
|
from ...entities.user_entity import UserEntity
|
||||||
|
from ...entities.user_entity import RoleEnum
|
||||||
|
from ...entities.user_entity import ProgramEnum
|
||||||
|
|
||||||
|
def test_add_sample_data_user(session: Engine):
|
||||||
|
|
||||||
|
|
||||||
|
"""Inserts a sample data point and verifies it is in the database"""
|
||||||
|
entity = UserEntity(id=1, username="emmalynf", role=RoleEnum.ADMIN, email="efoster@unc.edu", program=[ProgramEnum.COMMUNITY, ProgramEnum.DOMESTIC, ProgramEnum.ECONOMIC], experience=10, group="group")
|
||||||
|
session.add(entity)
|
||||||
|
session.commit()
|
||||||
|
data = session.get(UserEntity, 1)
|
||||||
|
assert data.id == 1
|
||||||
|
assert data.username == "emmalynf"
|
||||||
|
assert data.email == "efoster@unc.edu"
|
||||||
|
assert data.experience == 10
|
||||||
|
assert data.role == RoleEnum.ADMIN
|
||||||
|
assert data.program == [ProgramEnum.COMMUNITY, ProgramEnum.DOMESTIC, ProgramEnum.ECONOMIC]
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user