Fix deadlock

Uwjshhs78 · May 12, 2020 · 1670b61 · 1670b61
1 parent a196ce5
commit 1670b61
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bot/helper/mirror_utils/download_utils/aria2_download.py b/bot/helper/mirror_utils/download_utils/aria2_download.py
@@ -34,7 +34,7 @@ def __onDownloadComplete(self, api: API, gid):
                         download_dict[self.__listener.uid] = AriaDownloadStatus(self, self.__listener)
                         if download.is_torrent:
                             download_dict[self.__listener.uid].is_torrent = True
-                        update_all_messages()
+                    update_all_messages()
                     LOGGER.info(f'Changed gid from {gid} to {self.gid}')
                 else:
                     self.__listener.onDownloadComplete()
@@ -70,7 +70,7 @@ def add_download(self, link: str, path):
         with self._resource_lock:
             self.gid = download.gid
         with download_dict_lock:
-            download_dict[self._listener.uid] = AriaDownloadStatus(self.gid, self._listener)
+            download_dict[self.__listener.uid] = AriaDownloadStatus(self, self.__listener)
             LOGGER.info(f"Started: {self.gid} DIR:{download.dir} ")
 
     def cancel_download(self):