fix: Safely check DG hierarchy info

PaloAltoNetworks · Sep 11, 2021 · b143999 · b143999
1 parent d3769a8
commit b143999
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/panos/panorama.py b/panos/panorama.py
@@ -123,7 +123,7 @@ def refresh(self):
 
         dev = self.obj.panorama()
         state = dev.opstate.dg_hierarchy.fetch()
-        self.parent = state[self.obj.uid]
+        self.parent = state.get(self.obj.uid)
 
     def update(self):
         """Change this device group's hierarchical parent.