def create(self,request):
serializer = VendorCreateSerializer(data=request.data)
if serializer.is_valid():
vendor = Vendor(name = serializer.validated_data.get('name'),
service_type=serializer.validated_data.get('service_type'),
address = serializer.validated_data.get('address'),
contact_no_1 = serializer.validated_data.get('contact_no_1'),
contact_no_2=serializer.validated_data.get('contact_no_2'),
description = serializer.validated_data.get('description'),
created_by =serializer.validated_data.get('created_by'),
created_on=serializer.validated_data.get('created_on'),
)
vendor.put()
return Response(serializer.data, status=status.HTTP_201_CREATED)
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
def post(self):
user = users.get_current_user()
isVendor = False
if self.request.get("entity_key"):
memento_user_key = ndb.Key(urlsafe=self.request.get("entity_key"))
memento_user = memento_user_key.get()
memento_user.user_name = user.nickname()
isVendor = False
if 'isVendor' in self.request.POST:
isVendor = True
memento_user.isVendor = isVendor
memento_user.put()
if memento_user.isVendor:
self.redirect('/VendorHub')
else:
self.redirect('/HRHub')
else:
if 'isVendor' in self.request.POST:
isVendor = True
new_memento_user = MementoUser(parent = MEMENTO_USER_KEY,
user_name = user.nickname(),
isVendor = isVendor,
user_data = None)
new_memento_user.put()
if new_memento_user.isVendor:
new_Vendor = Vendor(parent = new_memento_user.key, company_name=self.request.get("company"), inventory=[])
new_Vendor.put()
new_memento_user.user_data = new_Vendor.key
new_memento_user.put()
self.redirect('/VendorHub')
else:
new_HR = HRUser(parent = new_memento_user.key, company_name=self.request.get("company"), mementos=[])
new_HR.put()
new_memento_user.user_data = new_HR.key
new_memento_user.put()
self.redirect('/HRHub')
