Lock dispatcher groups. Fixes #255

pyrogram/client/ext/dispatcher.py

@@ -20,7 +20,7 @@ import logging
 import threading
 from collections import OrderedDict
 from queue import Queue
-from threading import Thread
+from threading import Thread, Lock
 
 import pyrogram
 from pyrogram.api import types
@@ -64,6 +64,8 @@ class Dispatcher:
         self.updates_queue = Queue()
         self.groups = OrderedDict()
 
+        self.lock = Lock()
+
         self.update_parsers = {
             Dispatcher.MESSAGE_UPDATES:
                 lambda upd, usr, cht: (pyrogram.Message._parse(self.client, upd.message, usr, cht), MessageHandler),
@@ -110,6 +112,7 @@ class Dispatcher:
         self.groups.clear()
 
     def add_handler(self, handler, group: int):
+        with self.lock:
             if group not in self.groups:
                 self.groups[group] = []
                 self.groups = OrderedDict(sorted(self.groups.items()))
@@ -117,6 +120,7 @@ class Dispatcher:
             self.groups[group].append(handler)
 
     def remove_handler(self, handler, group: int):
+        with self.lock:
             if group not in self.groups:
                 raise ValueError("Group {} does not exist. Handler was not removed.".format(group))
 
@@ -142,6 +146,7 @@ class Dispatcher:
                     else (None, type(None))
                 )
 
+                with self.lock:
                     for group in self.groups.values():
                         for handler in group:
                             args = None