From 3b579a97d9c1076aac682a8c9188c0710edddaeb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mi=C2=B3osz=20Stocki?= <milosz.stocki@gmail.com>
Date: Thu, 17 Mar 2022 11:50:15 +0100
Subject: [PATCH] handle multiple WMI objects

---
 tactical_agents_telegraf.py | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/tactical_agents_telegraf.py b/tactical_agents_telegraf.py
index 2189c83..bc42e66 100644
--- a/tactical_agents_telegraf.py
+++ b/tactical_agents_telegraf.py
@@ -63,12 +63,16 @@ class Agent(object):
             wmi_details = {}
             pass
         wmi_dict = {}
+        wmi_dict_keys = []
         for k, v in wmi_details.items():
-            try:
-                value = v[0][0]
-            except IndexError:
-                value = dict()
-            wmi_dict[k] = value
+            for index, item in enumerate(v):
+                try:
+                    value = item[0]
+                except IndexError:
+                    value = dict()
+                new_key = k + str(index)
+                wmi_dict[new_key] = value
+                wmi_dict_keys.append(new_key)
         return wmi_dict
 
     def get_agent_metrics(self):
@@ -116,4 +120,4 @@ telegraf_client = TelegrafClient(host=config['telegraf']['host'], port=config['t
 
 for agent in agent_list:
     a = Agent(agent, api)
-    a.send_agent_metrics(telegraf_client)
+    a.send_agent_metrics(telegraf_client)
\ No newline at end of file