def setUp(self): self.splunk_venture = Venture( name='Splunk unknown usage', venture_id=666, symbol='splunk_unknown_usage', ) self.splunk_venture.save() venture1 = Venture(name='venture1', venture_id=111, symbol='venture1') venture1.save() venture2 = Venture(name='venture2', venture_id=222, symbol='venture2') venture2.save() self.device1 = Device(name='test_host1', device_id=1) self.device1.save() self.device2 = Device(name='test_host2', device_id=2) self.device2.save() daily_device1 = DailyDevice( date=datetime.datetime.today(), name='test_host1', pricing_venture=venture1, pricing_device=self.device1, ) daily_device1.save() daily_device2 = DailyDevice( date=datetime.datetime.today(), name='test_host2', pricing_venture=venture2, pricing_device=self.device2, ) daily_device2.save()
def update_assets(data, date): created = False if not data['ralph_id']: return False try: old_device = Device.objects.exclude(device_id=data['ralph_id'], ).get( asset_id=data['asset_id'], ) except Device.DoesNotExist: pass else: old_device.asset_id = None old_device.save() try: device = Device.objects.get(device_id=data['ralph_id']) except Device.DoesNotExist: created = True device = Device() device.device_id = data['ralph_id'] device.asset_id = data['asset_id'] device.slots = data['slots'] device.sn = data['sn'] device.barcode = data['barcode'] device.save() daily, daily_created = DailyDevice.objects.get_or_create( date=date, pricing_device=device, ) daily.price = data['price'] daily.deprecation_rate = data['deprecation_rate'] daily.is_deprecated = data['is_deprecated'] daily.save() return created