GitLab is used only for code review, issue tracking and project management. Canonical locations for source code are still and

Commit 9acad379 authored by juga  's avatar juga
Browse files

chg: state: Add method to count list values

parent 26409a3d
......@@ -90,3 +90,18 @@ class State:
def __contains__(self, item):
self._state = self._read()
return self._state.__contains__(item)
def count(self, k):
Returns the length if the key value is a list
or the sum of number if the key value is a list of list
or the key value
or None if the state doesn't have the key.
if self.get(k):
if isinstance(self._state[k], list):
if isinstance(self._state[k][0], list):
return sum(map(lambda x: x[1], self._state[k]))
return len(self._state[k])
return self.get(k)
return None
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment