FEAT: Update Community model to include role field and modify admin creation logic to use orm

Author: IkeSalmonson

app/routers/admin/routes.py

@@ -22,13 +22,15 @@ async def create_admin(session: AsyncSession):
     hashed_password = auth.hash_password(password)
     community = DBCommunity(
         username=ADMIN_USER,
-        email="ADMIN_USER",
+        email="ADMIN_USER@mail.com",
         password=hashed_password,
         role="admin",
     )
-    session.add(community)
-    await session.commit()
-    await session.refresh(community)
+    await create_community(session=session, community=community)
+
+    # session.add(community)
+    # await session.commit()
+    # await session.refresh(community)
     return {"msg": "Admin successfully created"}
 
 

app/schemas.py

@@ -34,9 +34,10 @@ class Community(BaseModel):
     email: str
 
 
-# Extends Community Class with hashed password
+# Extends Community Class with hashed password and role
 class CommunityInDB(Community):
     password: str
+    role: str
 
 
 class News(BaseModel):

app/services/database/models/communities.py

@@ -17,3 +17,4 @@ class Community(SQLModel, table=True):
         default_factory=datetime.now,
         sa_column_kwargs={"onupdate": datetime.now},
     )
+    role: str = Field(default="user")  # user or admin