handle null facility and metro in ip/vlan responses

Signed-off-by: Marques Johansson <[email protected]>

Author: displague

packet/IPAddress.py

@@ -23,15 +23,18 @@ def __init__(self, data, manager):
         self.customdata = data.get("customdata")
         self.project = data.get("project")
         self.project_lite = data.get("project_lite")
-        self.facility = Facility(data.get("facility"))
-        self.metro = Metro(data.get("metro"))
         self.details = data.get("details")
         self.assigned_to = data.get("assigned_to")
         self.interface = data.get("interface")
         self.network = data.get("network")
         self.address = data.get("address")
         self.gateway = data.get("gateway")
 
+        facility = data.get("facility", None)
+        self.facility = Facility(facility) if facility else None
+        metro = data.get("metro", None)
+        self.metro = Metro(metro) if metro else None
+
     def delete(self):
         return self.manager.call_api("ips/%s" % self.id, type="DELETE")
 

packet/Vlan.py

@@ -18,9 +18,11 @@ def __init__(self, data, manager):
         self.facility_code = data.get("facility_code")
         self.metro_code = data.get("metro_code")
         self.created_at = data.get("created_at")
+        facility = data.get("facility", None)
+        self.facility = Facility(facility) if facility else None
+        metro = data.get("metro", None)
+        self.metro = Metro(metro) if metro else None
 
-        self.facility = Facility(data.get("facility"))
-        self.metro = Metro(data.get("metro"))
         try:
             project_data = self.manager.call_api(
                 data["assigned_to"]["href"], type="GET"