Fix case if agent is an general agent (has no protocol attribute in obsconfig

Alexis Koralewski
1 parent ac8a42fa
Showing 1 changed file with 3 additions and 0 deletions Show diff stats
src/core/pyros_django/obs_config/obsconfig_class.py
@@ -1380,6 +1380,9 @@ class OBSConfig:
         print(f"{computer_agents=}")
         for computer_agent_name in computer_agents:
             protocol_name = self.get_agent_information(self.unit_name, computer_agent_name).get("protocol")
+            if protocol_name is None:
+                # Agent has no protocol, might be an general agent of PyROS, return the name of the class (i.e. do nothing)
+                return agent_name
             agent_real_name, _ =  os.path.splitext(protocol_name)
             if agent_name in agent_real_name:
                 return agent_name