From bafc3ec4b2686cdec4ac1c33f69f7607f368d4ce Mon Sep 17 00:00:00 2001
From: Dan Klein <danklein10@gmail.com>
Date: Wed, 7 Oct 2015 12:02:13 +0300
Subject: updated stats handling implementation
---
.../trex_control_plane/common/trex_stats.py | 38 +++++++++++++++-------
1 file changed, 26 insertions(+), 12 deletions(-)
(limited to 'scripts/automation')
diff --git a/scripts/automation/trex_control_plane/common/trex_stats.py b/scripts/automation/trex_control_plane/common/trex_stats.py
index 62c3a890..33015f01 100755
--- a/scripts/automation/trex_control_plane/common/trex_stats.py
+++ b/scripts/automation/trex_control_plane/common/trex_stats.py
@@ -3,18 +3,32 @@ import copy
class CTRexStatsManager(object):
- def __init__(self):
- self._stats = {}
- pass
-
- def update(self, obj_id, stats_obj):
- assert isinstance(stats_obj, CTRexStats)
- self._stats[obj_id] = stats_obj
-
- def get_stats(self, obj_id):
- return copy.copy(self._stats.pop(obj_id))
-
-
+ def __init__(self, *args):
+ for stat_type in args:
+ # register stat handler for each stats type
+ setattr(self, stat_type, CTRexStatsManager.CSingleStatsHandler())
+
+ def __getitem__(self, item):
+ stats_obj = getattr(self,item)
+ if stats_obj:
+ return stats_obj.get_stats()
+ else:
+ return None
+
+ class CSingleStatsHandler(object):
+
+ def __init__(self):
+ self._stats = {}
+
+ def update(self, obj_id, stats_obj):
+ assert isinstance(stats_obj, CTRexStats)
+ self._stats[obj_id] = stats_obj
+
+ def get_stats(self, obj_id=None):
+ if obj_id:
+ return copy.copy(self._stats.pop(obj_id))
+ else:
+ return copy.copy(self._stats)
class CTRexStats(object):
--
cgit