def vendor_get_or_create(vendor): change_vendor_name = {'Bosch GmbH': 'Bosch', 'Bosch GmbH,Bosch': 'Bosch'} if vendor in change_vendor_name: vendor = change_vendor_name[vendor] elif vendor == None: return None try: return Vendor.objects.get(name=vendor.lower()) except Vendor.DoesNotExist: obj = Vendor(name=vendor.lower()) return obj.save()
def setUp(self): num_vendor = 3 num_medicine = 3 num_usable_item = 3 password = '******' self.my_admin = User.objects.create_superuser('myuser', '*****@*****.**', password) self.my_admin.save() self.c = Client() self.c.login(username=self.my_admin.username, password=password) for id in range(1, num_vendor): test = Vendor(id, f"test{id}", f"test{id} address", f"987654321{id}", f"test{id}@email.com") test.save() for id in range(1, num_medicine): test = Medicine(id=id, trade_name=f"test_trade{id}", generic_name=f"test_generic{id}", description=f"description{id}", purchasePrice=50.0, unit_selling=60.0, threshold_value=10, vendor=Vendor.objects.get(id=id)) test.save() for id in range(1, num_usable_item): test = UsableItem.objects.create( id=id, batch_id=f"test_batch{id}", expiry_date="2025-04-04", order_date="2021-04-04", quantity=50, medicine=Medicine.objects.get(id=id)) test.save() for id in range(1, num_usable_item): test = CustomerTransaction.objects.create( name="name1", phone="1234567890", transaction_id="1", quantity="1", date="2021-04-04", expiry_date="2025-04-04", batch_id="2", medicine=Medicine.objects.get(id=1)) test2 = VendorTransaction.objects.create( transaction_id="1", medicine_id="1", vendor=Vendor.objects.get(id=1)) self.username = '******' self.password = '******'
def setUp(self): num_vendor = 5 for id in range(1, num_vendor): test = Vendor( id, f"test{id}", f"test{id} address", f"987654321{id}", f"test{id}@email.com" ) test.save() password = '******' self.my_admin = User.objects.create_superuser( 'myuser', '*****@*****.**', password) self.my_admin.save() self.c = Client() self.c.login(username=self.my_admin.username, password=password)