def create(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
device = FCMDevice()
user_info = serializer.validated_data['user_info']
provider = serializer.data.get('provider')
try:
user = User.objects.get(email=user_info['email'])
device.user = user
device.registration_id = request.data.get('registration_id')
device.save()
except ObjectDoesNotExist:
user = User()
if provider == 'google':
user.email = user_info['email']
user.name = user_info['name']
user.avatar = user_info['picture']
customer = stripe.Customer.create(
email=user.email,
name=user.name,
)
user.customer_id = customer.id
user.save()
device.user = user
device.registration_id = request.data.get('registration_id')
device.save()
if not user.is_active:
return Response(
{
"result": False,
"errorCode": 13,
"errorMsg": "User account is disabled."
},
status=status.HTTP_500_INTERNAL_SERVER_ERROR)
try:
token = Token.objects.get(user=user)
except ObjectDoesNotExist:
token = Token.objects.create(user=user)
response = {
"result": True,
"data": {
"token": token.key,
"user": {
"id": user.id,
"email": user.email,
"username": user.username,
"name": user.name,
"avatar": user.avatar,
"address": user.address,
"phone": user.phone
}
}
}
return Response(response, status=status.HTTP_201_CREATED)
def addUser(request): password = hashlib.sha256(request.POST['password']).hexdigest() user = User(account=request.POST['account'],password=password) user.name = request.POST['name'] user.phone = request.POST['phone'] user.address = request.POST['address'] if 'avatar' in request.FILES: user.avatar = request.FILES['avatar'] user.save() return HttpResponse(200)
def store_user(request):
"""
This view creates user and Store user in DB
"""
if request.user.is_admin:
formData = User()
name = request.POST.get("name")
email = request.POST.get("email")
password = request.POST.get("password")
phone = request.POST.get("phone")
designation = request.POST.get("designation")
isActive = request.POST.get("user_status")
role = request.POST.get("role_id")
if 'avatar' in request.FILES:
avatar_file = request.FILES["avatar"]
save_path = os.path.join(settings.MEDIA_ROOT, avatar_file.name)
default_storage.save(save_path, avatar_file)
formData.avatar = avatar_file.name
formData.name = name
formData.email = email
formData.password = make_password(password)
formData.phone = phone
formData.designation = designation
formData.status = isActive
if isActive == 'Active':
formData.is_active = True
else :
formData.is_active = False
if role == "s":
formData.is_staff = True
elif role == "a" :
formData.is_admin = True
elif role == "v" :
formData.is_viewer = True
# save_path = os.path.join(settings.MEDIA_ROOT, avatar_file.name)
# default_storage.save(save_path, avatar_file)
#
# formData.avatar = avatar_file.name
formData.save()
return redirect("/create-user/")
else:
return redirect('/dashboard/')
