xenofem/python-zulip-api
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

zulip: Add support for additional args for Client.call_on_each_*.

Browse source 
- Added `**kwargs` typed `object` in the definition of
  `Client.call_on_each_*` .

- Replaced `**kwargs` type from `Any` to `object` in the
  definition of `Client.register`.

Fixes part of #647
This commit is contained in:
LoopThrough-i-j 2021-03-10 22:46:56 +05:30 committed by Tim Abbott
parent 262c926bbf
commit 70f457f82a
1 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
zulip/zulip/__init__.py
View file

@ -617,17 +617,17 @@ class Client:
callback: Callable[[Dict[str, Any]], None],
event_types: Optional[List[str]] = None,
narrow: Optional[List[List[str]]] = None,
**kwargs: object,
) -> None:
if narrow is None:
narrow = []
def do_register() -> Tuple[str, int]:
while True:
if event_types is None:
res = self.register()
else:
res = self.register(event_types=event_types, narrow=narrow)
if event_types is None:
narrow = None
while True:
res = self.register(event_types, narrow, **kwargs)
if 'error' in res['result']:
if self.verbose:
print("Server returned error:\n%s" % (res['msg'],))
@ -679,11 +679,11 @@ class Client:
last_event_id = max(last_event_id, int(event['id']))
callback(event)
def call_on_each_message(self, callback: Callable[[Dict[str, Any]], None]) -> None:
def call_on_each_message(self, callback: Callable[[Dict[str, Any]], None], **kwargs: object) -> None:
def event_callback(event: Dict[str, Any]) -> None:
if event['type'] == 'message':
callback(event['message'])
self.call_on_each_event(event_callback, ['message'])
self.call_on_each_event(event_callback, ['message'], None, **kwargs)
def get_messages(self, message_filters: Dict[str, Any]) -> Dict[str, Any]:
'''
@ -1029,7 +1029,7 @@ class Client:
self,
event_types: Optional[Iterable[str]] = None,
narrow: Optional[List[List[str]]] = None,
**kwargs: Any
**kwargs: object
) -> Dict[str, Any]:
'''
Example usage: