Updated GoogleDoc class to work with new(er) spreadsheets

etc/gdocs.py

@@ -27,11 +27,17 @@ class GoogleDoc(object):
     key = None
     file_format = 'xlsx'
     file_name = 'copy'
-    gid = '0'
+    gid = None
 
     # You can change these with kwargs but it's not recommended.
-    spreadsheet_url = 'https://spreadsheets.google.com/feeds/download/spreadsheets/Export?key=%(key)s&exportFormat=%(format)s&gid=%(gid)s'
-    new_spreadsheet_url = 'https://docs.google.com/spreadsheets/d/%(key)s/export?format=%(format)s&id=%(key)s&gid=%(gid)s'
+    spreadsheet_url = 'https://spreadsheets.google.com/feeds/download/spreadsheets/Export?key=%(key)s&exportFormat=%(format)s'
+    if gid:
+        spreadsheet_url = spreadsheet_url + '&gid=%(gid)s'
+
+    new_spreadsheet_url = 'https://docs.google.com/spreadsheets/d/%(key)s/export?format=%(format)s&id=%(key)s'
+    if gid:
+        new_spreadsheet_url = new_spreadsheet_url + '&gid=%(gid)s'
+
     auth = None
     email = os.environ.get('APPS_GOOGLE_EMAIL', None)
     password = os.environ.get('APPS_GOOGLE_PASS', None)

fabfile/text.py

@@ -11,7 +11,7 @@
 from etc.gdocs import GoogleDoc
 
 @task(default=True)
-def update():
+def update(gid=None):
     """
     Downloads a Google Doc as an Excel file.
     """
@@ -25,6 +25,9 @@ def update():
         bits = bits[1].split('&')
         doc['key'] = bits[0]
 
+        if gid:
+            doc['gid'] = gid
+
         g = GoogleDoc(**doc)
         g.get_auth()
         g.get_document()