class Login(ViewSet):
""" API Login """
authentication_classes = []
permission_classes = [
partial(APIAccessPermission,
API().get_api_name('auth', 'login'))
]
renderer_classes = [JSONRenderer]
serializer_class = LoginSerializer
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.token = ''
self.status = Status()
self.commons = Commons()
self.message = _('You are successfully logged in.')
self.error_msg = _('Something wrong. Please try again.')
def create(self, request):
self.commons.active_language(language=request.META.get(
'HTTP_LANGUAGE', getattr(settings, 'LANGUAGE_CODE')))
serializer = self.serializer_class(data=request.data)
if serializer.is_valid():
obj_user = authenticate(username=serializer.data['username'],
password=serializer.data['password'])
if obj_user is not None:
data = {'token': self.commons.init_token(obj_user)}
login(request, obj_user)
self.commons.logs(level=1,
message=str(obj_user) +
' has successfully logged in.',
name=__name__)
return self.commons.response(_status=self.status.HTTP_2000_OK,
data=data,
message=self.message)
self.error_msg = _('User does not exists.')
else:
self.error_msg = serializer.errors
self.commons.logs(level=2, message=self.error_msg, name=__name__)
return self.commons.response(_status=self.status.HTTP_4000_BAD_REQUEST,
error_msg=self.error_msg)
class Register(ViewSet):
""" Register your account. """
authentication_classes = []
permission_classes = [partial(APIAccessPermission, API().get_api_name('auth', 'register'))]
renderer_classes = [JSONRenderer]
serializer_class = RegisterSerializer
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.status = Status()
self.commons = Commons()
self.message = _('Congratulations. You have successfully registered.')
self.error_msg = _('Something wrong. Please try again.')
@transaction.atomic()
def create(self, request):
self.commons.active_language(language=request.META.get('HTTP_LANGUAGE', getattr(settings, 'LANGUAGE_CODE')))
serializer = self.serializer_class(data=self.request.data)
if serializer.is_valid():
if not serializer.compare_pwd(data=request.data):
obj_user, created = Users.objects.get_or_create(username=serializer.data['username'], defaults={
'password': make_password(serializer.data['password'])
})
if created:
data = {
'token': self.commons.init_token(obj_user)
}
self.commons.logs(level=1, message=str(obj_user) + ' has successfully registered.', name=__name__)
return self.commons.response(_status=self.status.HTTP_2000_OK, data=data, message=self.message)
self.error_msg = _('User always exists.')
else:
self.error_msg = _('Password does not match.')
else:
self.error_msg = serializer.errors
self.commons.logs(level=2, message=self.error_msg, name=__name__)
return self.commons.response(_status=self.status.HTTP_4000_BAD_REQUEST, error_msg=self.error_msg)