def post(self, request):
serializer = UserAvatarSerializer(data=request.data,
instance=request.user)
if serializer.is_valid(raise_exception=True):
serializer.save()
return success_response(serializer.data, '')
else:
return error_response(serializer.errors, 400)
def create(self, request, *args, **kwargs):
obj = dict(request.data)
obj['user_id'] = request.user.id
serializer = self.get_serializer(data=obj)
if serializer.is_valid(raise_exception=True):
serializer.save()
return success_response(data=serializer.data, message='', status=200)
def respond(request):
responded = User.objects.filter(id=request.data.get('user_id')).get()
if not responded.is_company:
raise error_response('You cannot respond to user', 400)
serializer = RespondSerializer(data=request.data)
if serializer.is_valid():
serializer.save()
return success_response(serializer.data, '')
def invite(request):
invited = User.objects.filter(id=request.data.get('user_id')).get()
if invited.is_company:
raise error_response('You cannot invite company', 400)
serializer = InviteSerializer(data=request.data)
if serializer.is_valid():
serializer.save()
return success_response(serializer.data, '')
def register(request):
serializer = UserSerializer(data=request.data)
if serializer.is_valid(raise_exception=True):
if serializer.data['is_company']:
User.objects.create_company(serializer.data['email'],
serializer.data['password'])
else:
User.objects.create_user(serializer.data['email'],
serializer.data['password'])
return success_response(data=serializer.data,
message="User created",
status=201)
else:
return error_response(message=serializer.errors, status=400)
def list(self, request, *args, **kwargs):
queryset = self.get_queryset()
serializer = self.get_serializer(queryset, many=True)
return success_response(data=serializer.data, message='', status=200)
def create(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
self.perform_create(serializer)
return success_response(data=serializer.data, message='', status=201)
def create_contacts(request):
response = Company.objects.add_contact(request.data.get('company_id'),
request.data.get('value'))
return success_response(response, '', 201)
def retrieve(self, request, *args, **kwargs):
queryset = self.get_object()
serializer = self.get_serializer(queryset, many=False)
return success_response(data=serializer.data, message='', status=201)
