Fixed further requirements.

This commit is contained in:
Alvaro Arriola 2019-10-10 17:35:06 +02:00
parent 0b5214d247
commit 821d6c8672
11 changed files with 61 additions and 25 deletions

View file

@ -1,5 +1,6 @@
from dcim.models import Site, Platform, DeviceRole
from virtualization.models import Cluster, VirtualMachine
from virtualization.constants import VM_STATUS_CHOICES
from tenancy.models import Tenant
from extras.models import CustomField, CustomFieldValue
from ruamel.yaml import YAML
@ -42,6 +43,11 @@ with file.open('r') as stream:
params[assoc] = model.objects.get(**query)
if 'status' in params:
for vm_status in VM_STATUS_CHOICES:
if params['status'] in vm_status:
params['status'] = vm_status[0]
virtual_machine, created = VirtualMachine.objects.get_or_create(**params)
if created: