feat: admin user

alquimistas-org · Sep 5, 2023 · f908b92 · f908b92
1 parent db80e9b
commit f908b92
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/models.py b/models.py
@@ -5,7 +5,7 @@
 from sqlalchemy import ForeignKey, Table
 from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, relationship
 
-from utils import Status
+from utils import Status, UserRole
 
 
 class Base(DeclarativeBase):
@@ -20,9 +20,7 @@ class User(Base):
     last_name: Mapped[str]
     email: Mapped[str] = mapped_column(unique=True)
     active: Mapped[bool] = mapped_column(default=False)
-
-
-# TODO: create admin user
+    role: Mapped[str] = mapped_column(SqlAlchemyEnum(UserRole), default=UserRole.USER)
 
 
 class Metric(Base):

diff --git a/utils.py b/utils.py
@@ -4,3 +4,8 @@
 class Status(enum.Enum):
     ACTIVE = "active"
     COMPLETED = "completed"
+
+
+class UserRole(enum.Enum):
+    ADMIN = "admin"
+    USER = "user"