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')