Merge pull request #39 from mcilya/next

Handle ConnectionResetError
ada-af · Nov 29, 2020 · eb46ce6 · eb46ce6
2 parents aad9e0b + c54c491
commit eb46ce6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/dermod/imgloader.py b/dermod/imgloader.py
@@ -34,7 +34,11 @@ def __init__(self, url, fileid, fileform=''):
 
     def run(self):
         print(f"Receiving {self.id}") if self.format == '' else ''
-        self.get_raw_image()
+        try:
+            self.get_raw_image()
+        except:
+            self.readiness = 1
+            quit(0)
         print(f"Wiritng {self.id}") if self.format == '' else ''
         self.writer()
         self.readiness = 1