def login_view(request):
username = request.data.get('username', '').lower()
password = request.data.get('password')
token = request.data.get('token')
try:
if token:
auth_user = Token.objects.get(key=token).user
else:
auth_user = authenticate(username=username, password=password)
if not auth_user and validate_email(username):
auth_user = get_user(email=username)
if not auth_user:
auth_user = authenticate(username=user.username, password=password)
if auth_user:
user = User.objects.get(auth_user=auth_user)
if user:
if auth_user.is_active:
response_data = UserSerializer(user).data
if Token.objects.filter(user=auth_user):
response_data['token'] = Token.objects.get(user=auth_user).key
return json_response(response_data, CODE_SUCCESS, MSG_LOGIN_SUCCESS)
else:
return invalid_token_response()
else:
return simple_json_response(CODE_NOT_ACTIVE, MSG_NOT_ACTIVE_USER)
else:
return simple_json_response(CODE_INCORRECT_USER_NAME_OR_PASSWORD, MSG_INCORRECT_USER_NAME_OR_PASSWORD)
else:
return simple_json_response(CODE_INCORRECT_USER_NAME_OR_PASSWORD, MSG_INCORRECT_USER_NAME_OR_PASSWORD)
except Exception as e:
return save_error_log(request, e)
def reset_password_with_verify_code_view(request):
code = request.data.get('verify_code')
email = request.data.get('email')
password = request.data.get('password')
try:
if not code:
return simple_json_response(CODE_EMPTY_VERIFY_CODE, MSG_EMPTY_VERIFY_CODE)
elif not email:
return simple_json_response(CODE_EMPTY_EMAIL, MSG_EMPTY_EMAIL)
elif not password:
return simple_json_response(CODE_EMPTY_PASSWORD, MSG_EMPTY_PASSWORD)
elif not validate_email(email):
return simple_json_response(CODE_INVALID_EMAIL, MSG_INVALID_EMAIL)
elif len(password) < MIN_PASSWORD_LEN:
return simple_json_response(CODE_INVALID_PASSWORD, MSG_INVALID_PASSWORD)
user = get_user(email=email)
if user:
user = User.objects.get(auth_user=user)
if Verify.objects.filter(user=user, email_verify_code=code.lower()):
verify = Verify.objects.filter(user=user, email_verify_code=code.lower())[0]
if (time.time() - float(format(verify.created, 'U'))) > VERIFY_CODE_EXPIRED_TIME:
return simple_json_response(CODE_EXPIRED_VERIFY_CODE, MSG_EXPIRED_VERIFY_CODE)
else:
user.auth_user.set_password(password)
user.auth_user.save()
user.save()
return simple_json_response(CODE_SUCCESS, MSG_RESET_PASSWORD_SUCCESS)
else:
return simple_json_response(CODE_INCORRECT_VERIFY_CODE, MSG_INCORRECT_VERIFY_CODE)
else:
return simple_json_response(CODE_USER_NOT_EXISTS, MSG_USER_NOT_EXISTS)
except Exception as e:
return save_error_log(request, e)
def update(self, request, *args, **kwargs):
token = request.data.get('token')
user_name = request.data.get('user_name')
phone = request.data.get('phone')
email = request.data.get('email')
gender = request.data.get('gender')
birthday = request.data.get('birthday')
hobbies = request.data.get('hobbies')
base64 = request.data.get('base64')
try:
user = get_user_by_token(token)
if user:
if AuthUser.objects.filter(user=user):
user = AuthUser.objects.get(user=user)
if email:
if AuthUser.objects.filter(
username__iexact=email
) or AuthUser.objects.filter(email__iexact=email):
return simple_json_response(
CODE_DUPLICATE_USER, MSG_DUPLICATE_EMAIL)
else:
user.email = email
user.save()
if user_name:
user.user_name = user_name
if phone:
if AuthUser.objects.filter(phone=phone):
return simple_json_response(
CODE_DUPLICATE_PHONE, MSG_DUPLICATE_PHONE)
else:
user.phone = phone
if gender:
user.gender = gender
if birthday:
user.birthday = birthday
if hobbies:
user.hobbies = hobbies
if base64:
image_name = user.username + time.strftime(
'%Y%m%d%H%M%S') + PROFILE_FOOTER_IMAGE
user.save()
response_data = UserSerializer(user).data
response_data['token'] = Token.objects.get(user=user).key
return json_response(response_data, CODE_SUCCESS,
MSG_UPDATE_USER_INFO_SUCCESS)
return invalid_token_response()
except Exception as e:
return save_error_log(request, e)
def destroy(self, request, *args, **kwargs):
try:
token = request.data.get('token')
auth_user = get_user_by_token(token)
if auth_user:
iaer = self.get_object()
if iaer:
try:
response = super(IaerViewSet,
self).destroy(request, *args,
**kwargs)
print("168: " + str(response.status_code))
if response.status_code != status.HTTP_204_NO_CONTENT:
iaer.id = -1
except Exception as e:
iaer.id = -1
save_error_log(request, e)
event_json = IaerSerializer(iaer).data
return json_response(event_json, CODE_SUCCESS,
MSG_DELETE_IAER_SUCCESS)
else:
return simple_json_response(CODE_NO_CONTENT, MSG_204)
else:
return invalid_token_response()
except Exception as e:
return save_error_log(request, e)
def destroy(self, request, *args, **kwargs):
try:
token = request.data.get('token')
user = get_user_by_token(token)
if user:
red_envelope = self.get_object()
if red_envelope:
try:
response = super(RedEnvelopeViewSet,
self).destroy(request, *args,
**kwargs)
if response.status_code != status.HTTP_204_NO_CONTENT:
red_envelope.id = -1
except Exception as e:
red_envelope.id = -1
save_error_log(request, e)
event_json = RedEnvelopeSerializer(red_envelope).data
return json_response(event_json, CODE_SUCCESS,
MSG_DELETE_RED_ENVELOPE_SUCCESS)
else:
return simple_json_response(CODE_NO_CONTENT, MSG_204)
else:
return invalid_token_response()
except Exception as e:
return save_error_log(request, e)
def create(self, request, *args, **kwargs):
try:
serializer = self.get_serializer(data=request.data)
username = request.data.get('username')
user_name = request.data.get('user_name')
password = request.data.get('password')
email = request.data.get('email')
first_name = request.data.get('first_name', '')
last_name = request.data.get('last_name', '')
base64 = request.data.get('base64', '')
if not username:
return simple_json_response(CODE_EMPTY_USER, MSG_EMPTY_USERNAME)
elif not user_name:
return simple_json_response(CODE_EMPTY_BABY_NAME, MSG_EMPTY_BABY_NAME)
elif not email:
return simple_json_response(CODE_EMPTY_EMAIL, MSG_EMPTY_EMAIL)
elif not password:
return simple_json_response(CODE_EMPTY_PASSWORD, MSG_EMPTY_PASSWORD)
elif not validate_email(email):
return simple_json_response(CODE_INVALID_EMAIL, MSG_INVALID_EMAIL)
elif len(password) < MIN_PASSWORD_LEN:
return simple_json_response(CODE_INVALID_PASSWORD, MSG_INVALID_PASSWORD)
elif AuthUser.objects.filter(username__iexact=username) or AuthUser.objects.filter(username__iexact=email):
return simple_json_response(CODE_DUPLICATE_USER, MSG_DUPLICATE_USER)
elif AuthUser.objects.filter(email__iexact=email) or AuthUser.objects.filter(email__iexact=username):
return simple_json_response(CODE_DUPLICATE_EMAIL, MSG_DUPLICATE_EMAIL)
elif serializer.is_valid():
auth_user = AuthUser()
auth_user.email = email.lower()
auth_user.is_active = True
auth_user.is_staff = True
auth_user.set_password(password)
auth_user.username = username.lower()
auth_user.first_name = first_name
auth_user.last_name = last_name
self.request.user = auth_user
self.perform_create(serializer)
response_data = serializer.data
response_data['token'] = Token.objects.create(user=auth_user).key
user = User.objects.get(auth_user = auth_user)
user.locale = response_data['locale']
user.created = timezone.now()
if base64:
image_name = username + time.strftime('%Y%m%d%H%M%S') + PROFILE_FOOTER_IMAGE
user.save()
return json_response(response_data, CODE_SUCCESS, MSG_CREATE_USER_SUCCESS)
else:
return simple_json_response(CODE_INVALID_REQUEST, MSG_400)
except Exception as e:
return save_error_log(request, e)
def send_verify_code_view(request):
email = request.data.get('email')
try:
if not email:
return simple_json_response(CODE_EMPTY_EMAIL, MSG_EMPTY_EMAIL)
elif not User.objects.filter(auth_user__email=email.lower()) and \
not User.objects.filter(auth_user__username=email.lower()):
return simple_json_response(CODE_INVALID_EMAIL, MSG_NO_SUCH_EMAIL)
user = None
if User.objects.filter(auth_user__email=email.lower()):
user = User.objects.get(auth_user__email=email.lower())
elif User.objects.filter(auth_user__username=email.lower()):
user = User.objects.get(auth_user__username=email.lower())
verify_code = get_random_string(length=6, allowed_chars='0123456789').lower()
send_email(user, email, verify_code)
return simple_json_response(CODE_SUCCESS, MSG_SEND_VERIFY_CODE_SUCCESS)
except Exception as e:
return save_error_log(request, e)