mirror of
https://github.com/lemeow125/DocManagerBackend.git
synced 2025-04-27 18:21:23 +08:00
Add in requested changes
This commit is contained in:
parent
d81319c8ec
commit
844113d44f
7 changed files with 54 additions and 25 deletions
|
@ -8,7 +8,8 @@ class IsStaff(BasePermission):
|
|||
|
||||
def has_permission(self, request, view):
|
||||
return bool(
|
||||
request.user and request.user.role in ("head", "admin", "planning", "staff")
|
||||
request.user and request.user.role in (
|
||||
"head", "admin", "planning", "staff")
|
||||
)
|
||||
|
||||
|
||||
|
@ -18,7 +19,7 @@ class IsPlanning(BasePermission):
|
|||
"""
|
||||
|
||||
def has_permission(self, request, view):
|
||||
return bool(request.user and request.user.role in ("head", "admin", "planning"))
|
||||
return bool(request.user and request.user.role == "planning")
|
||||
|
||||
|
||||
class IsHead(BasePermission):
|
||||
|
|
|
@ -6,7 +6,7 @@ from rest_framework.settings import api_settings
|
|||
|
||||
|
||||
class CustomUserSerializer(serializers.ModelSerializer):
|
||||
birthday = serializers.DateField(format="%m-%d-%Y")
|
||||
birthday = serializers.DateField(format="%Y-%m-%d")
|
||||
|
||||
class Meta:
|
||||
model = CustomUser
|
||||
|
|
|
@ -10,19 +10,48 @@ from .models import CustomUser
|
|||
def create_admin_user(sender, **kwargs):
|
||||
# Programatically creates the administrator account
|
||||
if sender.name == "accounts":
|
||||
ADMIN_USER = CustomUser.objects.filter(
|
||||
email=get_secret("ADMIN_EMAIL")).first()
|
||||
if not ADMIN_USER:
|
||||
ADMIN_USER = CustomUser.objects.create_superuser(
|
||||
username=get_secret("ADMIN_EMAIL"),
|
||||
email=get_secret("ADMIN_EMAIL"),
|
||||
password=get_secret("ADMIN_PASSWORD"),
|
||||
sex="male",
|
||||
birthday=localdate(now()),
|
||||
)
|
||||
users = [{
|
||||
"email": get_secret("ADMIN_EMAIL"),
|
||||
"role": "head",
|
||||
"admin": True,
|
||||
}, {
|
||||
"email": "staff@test.com",
|
||||
"role": "staff",
|
||||
"admin": False,
|
||||
}, {
|
||||
"email": "planning@test.com",
|
||||
"role": "planning",
|
||||
"admin": False,
|
||||
}, {
|
||||
"email": "client@test.com",
|
||||
"role": "client",
|
||||
"admin": False,
|
||||
},]
|
||||
for user in users:
|
||||
USER = CustomUser.objects.filter(
|
||||
email=user["email"]).first()
|
||||
if not USER:
|
||||
if user["admin"]:
|
||||
USER = CustomUser.objects.create_superuser(
|
||||
username=user["email"],
|
||||
email=user["email"],
|
||||
password=get_secret("ADMIN_PASSWORD"),
|
||||
sex="male",
|
||||
birthday=localdate(now()),
|
||||
role=user["role"]
|
||||
)
|
||||
else:
|
||||
USER = CustomUser.objects.create_user(
|
||||
username=user["email"],
|
||||
email=user["email"],
|
||||
password=get_secret("ADMIN_PASSWORD"),
|
||||
sex="male",
|
||||
birthday=localdate(now()),
|
||||
role=user["role"]
|
||||
|
||||
print("Created administrator account:", ADMIN_USER.email)
|
||||
)
|
||||
print(f"Created {user['role']} account: {USER.email}")
|
||||
|
||||
ADMIN_USER.first_name = "Administrator"
|
||||
ADMIN_USER.is_active = True
|
||||
ADMIN_USER.save()
|
||||
USER.first_name = f"DEBUG_USER:{USER.email}"
|
||||
USER.is_active = True
|
||||
USER.save()
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue