Merge pull request #251 from linode/bugfix/image-uplaod-check-max-upload-size

fix: Check Image upload size before attempting upload

Author: Dorthu

linodecli/plugins/image-upload.py

@@ -12,6 +12,7 @@
 from sys import exit
 
 PLUGIN_BASE = 'linode-cli image-upload'
+MAX_UPLOAD_SIZE = 5 * 1024 * 1024 * 1024 # 5GB
 
 
 def _progress(cur, total):
@@ -83,17 +84,22 @@ def call(args, context):
     # get default region populated
     context.client.config.update(parsed)
 
-    if not parsed.region:
-        print("No region provided.  Please set a default region or use --region")
-        exit(1)
-
     # make sure the file exists and is ready to upload
     filepath = os.path.expanduser(parsed.file)
 
     if not os.path.isfile(filepath):
         print("No file at {}; must be a path to a valid file.".format(filepath))
         exit(2)
 
+    # make sure it's not larger than the max upload size
+    if os.path.getsize(filepath) > MAX_UPLOAD_SIZE:
+        print("File {} is too large; compressed size must be less than 5GB".format(filepath))
+        exit(2)
+
+    if not parsed.region:
+        print("No region provided.  Please set a default region or use --region")
+        exit(1)
+
     label = parsed.label or os.path.basename(filepath)
 
     # generate an upload URL