UploadStatus: Handle exception ZeroDivisionError

2020-02-05 01:52:43 -08:00 · 2020-02-05 01:52:43 -08:00 · 96e623ef57
parent 05468196e3
commit 96e623ef57
1 changed files with 4 additions and 1 deletions
--- a/bot/helper/mirror_utils/status_utils/upload_status.py
+++ b/bot/helper/mirror_utils/status_utils/upload_status.py
@ -28,7 +28,10 @@ class UploadStatus(Status):
        return self.obj.name

    def progress_raw(self):
+        try:
            return self.obj.uploaded_bytes / self.__size * 100
+        except ZeroDivisionError:
+            return 0

    def progress(self):
        return f'{round(self.progress_raw(), 2)}%'