Skip to content

Commit 3976b3b

Browse files
committed
Fix streams and put regex and metadata search into them.
1 parent 0412496 commit 3976b3b

1 file changed

Lines changed: 11 additions & 3 deletions

File tree

sentenai/__init__.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -870,15 +870,23 @@ def put(self, stream, event, id=None, timestamp=None):
870870
raise SentenaiException("something went wrong")
871871

872872

873-
def streams(self):
873+
def streams(self, name=None, meta={}):
874874
"""Get list of available streams."""
875875
url = "/".join([self.host, "streams"])
876876
headers = {'auth-key': self.auth_key}
877877
resp = requests.get(url, headers=headers)
878878
status_codes(resp.status_code)
879-
print(resp.json())
879+
880+
def filtered(s):
881+
f = True
882+
if name:
883+
f = bool(re.search(name, s['name']))
884+
for k,v in meta.items():
885+
f = f and s.get('meta', {}).get(k) == v
886+
return f
887+
880888
try:
881-
return [stream(**v) for v in resp.json()]
889+
return [stream(**v) for v in resp.json() if filtered(v)]
882890
except:
883891
raise SentenaiException("Something went wrong")
884892

0 commit comments

Comments
 (0)