spla/Mastodonplus.py
1
0
Fork 0
Codi Incidències Sol·licituds Pull repo.project Llançaments Paquets Wiki Activitat

Fix missing async in hashtag/list streams

Explorar la Font
This commit is contained in:
Lorenz Diener 2017-12-19 11:04:53 +01:00
pare 2420a5d6de
commit e5c50ea80d
S'han modificat 1 arxius amb 7 adicions i 5 eliminacions
Mostrar estadístiques Descarregar el fitxer de pedaç Descarregar el fitxer de diferències Expand all files Collapse all files

12
mastodon/Mastodon.py
Veure arxiu

@ -1387,7 +1387,7 @@ class Mastodon:
"""
if tag.startswith("#"):
raise MastodonIllegalArgumentError("Tag parameter should omit leading #")
return self.__stream("/api/v1/streaming/hashtag?tag={}".format(tag), listener)
return self.__stream("/api/v1/streaming/hashtag?tag={}".format(tag), listener, async=async)
@api_version("2.1.0", "2.1.0")
def stream_list(self, id, listener, async=False):
@ -1396,7 +1396,7 @@ class Mastodon:
list.
"""
id = self.__unpack_id(id)
return self.__stream("/api/v1/streaming/list?list={}".format(id), listener)
return self.__stream("/api/v1/streaming/list?list={}".format(id), listener, async=async)
###
# Internal helpers, dragons probably
@ -1670,9 +1670,11 @@ class Mastodon:
class __stream_handle():
def __init__(self, connection):
self.closed = False
self.connection = connection
def close(self):
self.closed = True
self.connection.close()
def is_alive(self):
@ -1682,10 +1684,10 @@ class Mastodon:
self._thread = threading.current_thread()
with closing(connection) as r:
try:
listener.handle_stream(r.iter_lines())
listener.handle_stream(r.iter_lines(chunk_size = 1, decode_unicode = True))
except AttributeError as e:
# Eat AttributeError from requests if user closes early
pass
if not self.closed:
raise e
return 0
handle = __stream_handle(connection)