Handle the case of maxSize missing in a discovery doc media section.
Reviewed in: http://codereview.appspot.com/4785041/
diff --git a/apiclient/discovery.py b/apiclient/discovery.py
index 8e14889..81242b8 100644
--- a/apiclient/discovery.py
+++ b/apiclient/discovery.py
@@ -250,6 +250,8 @@
def _media_size_to_long(maxSize):
"""Convert a string media size, such as 10GB or 3TB into an integer."""
+ if len(maxSize) < 2:
+ return 0
units = maxSize[-2:].upper()
multiplier = MULTIPLIERS.get(units, 0)
if multiplier:
@@ -284,7 +286,7 @@
mediaUpload = methodDesc['mediaUpload']
mediaPathUrl = mediaUpload['protocols']['simple']['path']
accept = mediaUpload['accept']
- maxSize = _media_size_to_long(mediaUpload['maxSize'])
+ maxSize = _media_size_to_long(mediaUpload.get('maxSize', ''))
if 'parameters' not in methodDesc:
methodDesc['parameters'] = {}