Commit e04df445 authored by Gladys Forte's avatar Gladys Forte

Merge branch 'RMSv2' of http://42.61.118.105:7990/scm/rms/api-main-service into core-dev-gladys

parents 6ec24683 8c97b2d2
...@@ -55,6 +55,7 @@ class UserViewSet(viewsets.ModelViewSet): ...@@ -55,6 +55,7 @@ class UserViewSet(viewsets.ModelViewSet):
'email', 'contact_no' 'email', 'contact_no'
) )
@rms.AccountValidation
@rms.user_create @rms.user_create
@transaction.atomic @transaction.atomic
def create(self, request, *args, **kwargs): def create(self, request, *args, **kwargs):
......
...@@ -88,6 +88,22 @@ class rms: ...@@ -88,6 +88,22 @@ class rms:
return function(self, request, *args, **kwargs) return function(self, request, *args, **kwargs)
return wrapper return wrapper
@staticmethod
def AccountValidation(function):
@wraps(function)
def wrapper(self, request, *args, **kwargs):
payload = request.data
username = self.queryset.filter(username=payload['username'])
if username.count() >= 1:
return Response(
{
"message": f"username {username.first().username} is already taken",
},status=status.HTTP_400_BAD_REQUEST
)
return function(self, request, *args, **kwargs)
return wrapper
@staticmethod @staticmethod
def reset_password(function): def reset_password(function):
@wraps(function) @wraps(function)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment