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
