Dump changes

app/user/service.py

@@ -1,15 +1,18 @@
+from datetime import date
 from typing import Optional
 
 from passlib.context import CryptContext
 from sqlalchemy.orm import Session
 
 from app.user.model import User
+from app.role import service as role_service
 
 pwd_context = CryptContext(schemes=["bcrypt"])
 
 
 def create_user(db: Session, user: User) -> User:
-    if get_user_by_username_or_email(db, user.username, user.email) is not None:
+    existing_user = get_user_by_username_or_email(db, user.username, user.email)
+    if existing_user is not None:
         raise Exception()
 
     user.password = pwd_context.hash(user.password)
@@ -31,8 +34,28 @@ def get_user_by_username_or_email(db: Session, username: str, email: str) -> Opt
     return db.query(User).filter(User.username == username, User.email == email).one_or_none()
 
 
+def create_root_user(db: Session, root_email: str, root_password: str):
+    root_user = get_user_by_username(db, "root")
+    admin_role = role_service.get_role_by_name(db, "admin")
+
+    if root_user is None:
+        root_user = User(
+            username="root",
+            email=root_email,
+            password=root_password,
+            given_name="Root",
+            role=admin_role,
+            birthdate=date.today()
+        )
+        create_user(db, root_user)
+
+
 def delete_user_by_id(db: Session, id: int):
     user = get_user_by_id(db, id)
+
+    if user.username == "root":
+        raise Exception()
+
     if user is not None:
         db.delete(user)
         db.commit()