useful list of entityIDs

Razumain · Jan 23, 2018 · f982bd7 · f982bd7
1 parent 31473f4
commit f982bd7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/pyff/store.py b/src/pyff/store.py
@@ -45,6 +45,9 @@ def update(self, t, tid=None, ts=None, merge_strategy=None):
     def reset(self):
         raise NotImplementedError()
 
+    def entity_ids(self):
+        return set(e.get('entityID') for e in self.lookup('entities'))
+
 
 class MemoryStore(StoreBase):
     def __init__(self):