Merge "fix: vlan id for cloud history"

redhat-performance · Nov 24, 2020 · 9d3222a · 9d3222a
2 parents 59d6e62 + 6acc048
commit 9d3222a
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/quads/model.py b/quads/model.py
@@ -81,10 +81,17 @@ class CloudHistory(Document):
 
     @staticmethod
     def prep_data(data):
-        for flag in ['provisioned', 'validated', 'vlan', 'last_redefined']:
+        for flag in ['provisioned', 'validated', 'last_redefined']:
             if flag in data:
                 data.pop(flag)
 
+        if 'vlan' in data:
+            if type(data.get('vlan')) == str:
+                data['vlan_id'] = int(data.get('vlan'))
+            elif type(data.get('vlan')) == Vlan:
+                data['vlan_id'] = data.get('vlan').vlan_id
+            data.pop('vlan')
+
         params = ['name', 'description', 'owner', 'ticket']
         result, data = param_check(data, params)