raise error on failed upload

racerxdl · Feb 11, 2019 · 5a47ead · 5a47ead
1 parent b3b8058
commit 5a47ead
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/instapy_cli/cli.py b/instapy_cli/cli.py
@@ -3,6 +3,7 @@
 from instapy_cli.api import InstapySession
 from instapy_cli.media import Media
 
+
 class InstapyCli(object):
     def __init__(self, username, password):
         self._session = InstapySession()
@@ -27,9 +28,12 @@ def upload(self, file, caption=''):
             print('\nSomething went bad.\nPlease retry or send an issue on https://github.com/b3nab/instapy-cli\n')
             upload_completed = False
         finally:
-            if upload_completed:
-                print('Done.')
+
             # media_status = 'YES' if media.isDownloaded() else 'NO'
             # print('Media is a downloaded file? ' + media_status)
             if media.isDownloaded():
-                media.removeMedia()
+                media.removeMedia()
+            if upload_completed:
+                print('Done.')
+            else:
+                raise IOError("Unable to upload.")