hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-08 21:10:29 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix workers not stopping on .disconnect(), start them on login

Browse Source
This commit is contained in:
Lonami Exo
2017-10-01 19:56:24 +02:00
parent 62ea77cbea
commit d98fd6a424
2 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
telethon/update_state.py
View File

@@ -32,7 +32,6 @@ class UpdateState:
# https://core.telegram.org/api/updates
self._state = tl.updates.State(0, 0, datetime.now(), 0, 0)
self._setup_workers()
def can_poll(self):
"""Returns True if a call to .poll() won't lock"""
@@ -66,16 +65,16 @@ class UpdateState:
"""Changes the number of workers running.
If 'n is None', clears all pending updates from memory.
"""
self._stop_workers()
self.stop_workers()
self._workers = n
if n is None:
self._updates.clear()
else:
self._setup_workers()
self.setup_workers()
workers = property(fget=get_workers, fset=set_workers)
def _stop_workers(self):
def stop_workers(self):
"""Raises "StopIterationException" on the worker threads to stop them,
and also clears all of them off the list
"""
@@ -93,7 +92,7 @@ class UpdateState:
self._worker_threads.clear()
def _setup_workers(self):
def setup_workers(self):
if self._worker_threads or not self._workers:
# There already are workers, or workers is None or 0. Do nothing.
return