separated user enum into its own file

This commit is contained in:
Jordweinstein 2024-03-25 19:29:37 -04:00
parent 8676a22ad2
commit 03f2204cb3
4 changed files with 14 additions and 12 deletions

View File

@ -18,7 +18,7 @@ from sqlalchemy import Enum
# Import self for to model
from typing import Self
from programtype_enum import ProgramType
from backend.entities.program_type_enum import ProgramType
class ResourceEntity(EntityBase):

View File

@ -17,7 +17,7 @@ import enum
from sqlalchemy import Enum
# Import ProgramType enumeration
from programtype_enum import ProgramType
from backend.entities.program_type_enum import ProgramType
class ResourceEntity(EntityBase):

View File

@ -15,15 +15,8 @@ from datetime import datetime
# Import enums for Role and Program
import enum
from sqlalchemy import Enum
from programtype_enumeration import ProgramType
class RoleEnum(enum.Enum):
"""Determine role for User"""
ADMIN = "ADMIN"
EMPLOYEE = "EMPLOYEE"
VOLUNTEER = "VOLUNTEER"
from program_type_enum import ProgramType
from user_enum import RoleType
class UserEntity(EntityBase):
@ -38,7 +31,7 @@ class UserEntity(EntityBase):
username: Mapped[str] = mapped_column(
String(32), nullable=False, default="", unique=True
)
role: Mapped[RoleEnum] = mapped_column(Enum(RoleEnum), nullable=False)
role: Mapped[RoleType] = mapped_column(Enum(RoleType), nullable=False)
email: Mapped[str] = mapped_column(String(50), nullable=False, unique=True)
program: Mapped[list[ProgramType]] = mapped_column(
ARRAY(Enum(ProgramType)), nullable=False

View File

@ -0,0 +1,9 @@
import enum
class RoleType(enum.Enum):
"""Determine role for User"""
ADMIN = "ADMIN"
EMPLOYEE = "EMPLOYEE"
VOLUNTEER = "VOLUNTEER"