added spreadsheet creation functionality

condad · condad · commit b3560d1f8016 · 2017-06-06T15:26:46.000-04:00
diff --git a/google_objects/sheets.py b/google_objects/sheets.py
@@ -23,6 +23,34 @@
     # iii/ change .from_existing to .from_raw
 
 
+def _value_to_cell(val):
+    if str(val).isdigit():
+        try:
+            return {'userEnteredValue': {'numberValue': float(val)}}
+        except:
+            return {'userEnteredValue': {'numberValue': int(val)}}
+    return {'userEnteredValue': {'stringValue': str(val)}}
+
+
+def _cells_to_row(cells):
+    return [_value_to_cell(cell) for cell in cells]
+
+
+def _format_sheet(sheet):
+    title = sheet.get('title', 'NEW SHEET')
+    values = sheet.get('values', [])
+    return {
+        'properties': {
+            'title': title
+        },
+        'data': {
+            'rowData': [
+                {'values': _cells_to_row(val)} for val in values
+            ]
+        }
+    }
+
+
 def _grid_to_a1(sheet_name, start, end):
     start_row, start_col = start
     end_row, end_col = end
@@ -62,6 +90,16 @@ def get_spreadsheet(self, id):
 
         return Spreadsheet.from_existing(data, self)
 
+    def create_spreadsheet(self, sheets=[], **kwargs):
+        data = self._resource.spreadsheets().create(
+            body={
+                'properties': kwargs,
+                'sheets': [_format_sheet(s) for s in sheets]
+            }
+        ).execute()
+
+        return Spreadsheet.from_existing(data, self)
+
     def get_values(self, spreadsheet_id, range_name):
         """Initialize a new block and return it"""
 
@@ -122,26 +160,31 @@ def __init__(self, client=None, **kwargs):
         # initalize the other properties
         super(self.__class__, self).__init__(**kwargs)
 
-    @classmethod
-    def from_existing(cls, data, client=None):
-        """initiates using existing Spreadsheet resource"""
-
-        new_data = keys_to_snake(data)
-        return cls(client, **new_data)
-
     def __iter__(self):
         return self.yield_sheets()
 
+    def __getitem__(self, key):
+        try:
+            if key.isdigit():
+                return self.get_sheet_by_id(key)
+            else:
+                return self.get_sheet_by_name(key)
+        except ValueError:
+            raise TypeError('Sheet not found')
+
     def __enter__(self):
         return self
 
     def __exit__(self, exception_type, exception_value, traceback):
-        # if self.__updates:
-            # self.client.push_updates(self._id, self._updates)
-            # # TODO: add success handlers
-            # del self._updates[:]
         self.update()
 
+    @classmethod
+    def from_existing(cls, data, client=None):
+        """initiates using existing Spreadsheet resource"""
+
+        new_data = keys_to_snake(data)
+        return cls(client, **new_data)
+
     @property
     def id(self):
         return self._spreadsheet_id
@@ -216,15 +259,6 @@ def update(self):
             # TODO: add success handlers
             del self._updates[:]
 
-    def __getitem__(self, key):
-        try:
-            if key.isdigit():
-                return self.get_sheet_by_id(key)
-            else:
-                return self.get_sheet_by_name(key)
-        except ValueError:
-            raise TypeError('Sheet not found')
-
     class NamedRange(object):
         """represents a NamedRange resource, can
         return it's range in A1 notation
