def get_role(request):
user = request.user
result = {'user': UserSerializer(user).data}
if user.is_superuser:
result['role'] = 'admin'
return HttpResponse(json.dumps(result),
content_type='application/json')
roles = get_user_roles(user)
permissions = []
for role in roles:
for p, granted in role.available_permissions.items():
if granted:
permissions.append(p)
result['permissions'] = permissions
roles = []
if has_role(user, 'purchasing_agent'):
roles.append('purchasing_agent')
if has_role(user, 'finance'):
roles.append('finance')
if has_role(user, 'godown_manager'):
roles.append('godown_manager')
if has_role(user, 'operator'):
roles.append('operator')
result['role'] = roles[0]
return HttpResponse(json.dumps(result), content_type='application/json')
def put(self):
obj = User.query.filter(User.username==current_identity.username).first()
logo_form = LogoUpdateForm(request.files, csrf_enabled=False)
if logo_form.validate():
file = logo_form.logo.data
file_name = secure_filename(file.filename)
file.save(os.path.join(app.config['UPLOAD_FOLDER'], file_name))
url = app.config['STATIC_FOLDER']+file_name
logo_form.save(obj, url)
return jsonify({"status": "success","data": UserSerializer().dump(obj).data})
return logo_form.errors
def register_user(request):
data = request.data
serializer = UserSerializer(data = data)
if serializer.is_valid():
user = User.objects.create_user(**request.data)
if user:
data['user'] = user.id
serializer = MerchantSerializer(data = data)
if serializer.is_valid():
m_user = serializer.save()
if m_user:
html = REGISTRATION_HTML.format(name = data.get('first_name'), username = data.get('username'))
send_mail(html, data.get('email'))
return Response({'username':data.get('username'), 'email':data.get('email')}, status = status.HTTP_200_OK)
else:
user.delete()
print serializer.errors
return Response(serializer.errors, status = status.HTTP_400_BAD_REQUEST)
else:
print serializer.errors,'---'
return Response(serializer.errors, status = status.HTTP_400_BAD_REQUEST)
def register_user(request):
data = request.data
serializer = UserSerializer(data=data)
if serializer.is_valid():
user = User.objects.create_user(**request.data)
if user:
data['user'] = user.id
serializer = MerchantSerializer(data=data)
if serializer.is_valid():
m_user = serializer.save()
if m_user:
html = REGISTRATION_HTML.format(name=data.get('first_name'),
username=data.get('username'))
send_mail(html, data.get('email'))
return Response(
{
'username': data.get('username'),
'email': data.get('email')
},
status=status.HTTP_200_OK)
else:
user.delete()
print serializer.errors
return Response(serializer.errors,
status=status.HTTP_400_BAD_REQUEST)
else:
print serializer.errors, '---'
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
def get_user_total(**kwargs):
result = User.objects
if kwargs.get('user_number'):
user_number = kwargs.get('user_number')
kwargs.pop('user_number')
result = result.filter(user_number__contains=user_number)
if kwargs.get('user_name'):
user_name = kwargs.get('user_name')
kwargs.pop('user_name')
result = result.filter(user_name__contains=user_name)
if kwargs.get('user_phone'):
user_phone = kwargs.get('user_phone')
kwargs.pop('user_phone')
result = result.filter(user_phone__contains=user_phone)
result = result.filter(**kwargs)
return len(UserSerializer(result, many=True).data)
def get_user(page_num, page_size, **kwargs):
start = (page_num - 1) * page_size
end = start + page_size
result = User.objects
if kwargs.get('user_number'):
user_number = kwargs.get('user_number')
kwargs.pop('user_number')
result = result.filter(user_number__contains=user_number)
if kwargs.get('user_name'):
user_name = kwargs.get('user_name')
kwargs.pop('user_name')
result = result.filter(user_name__contains=user_name)
if kwargs.get('user_phone'):
user_phone = kwargs.get('user_phone')
kwargs.pop('user_phone')
result = result.filter(user_phone__contains=user_phone)
result = result.filter(**kwargs).order_by('user_number')[start:end]
return UserSerializer(result, many=True).data
def index(request):
serializer = UserSerializer(request.user)
return Response(serializer.data, status=status.HTTP_200_OK)
def create_user(**kwargs):
serializer = UserSerializer(data=kwargs)
if serializer.is_valid(raise_exception=True):
serializer.save(**kwargs)
def get(self):
user = User.query.filter(User.username==current_identity.username).first()
return jsonify({"data": UserSerializer().dump(user).data})
def get(self, user_id):
abort_if_user_doesnt_exist(user_id)
sub_account_validate(user_id)
user = User.query.get(user_id)
return jsonify({"data": UserSerializer().dump(user).data})
def process_request(self, request):
if request.user.is_authenticated():
request.user.user_json = UserSerializer(request.user,
many=False).data
return None
