can now init. sheets from pandas DataFrames

condad · condad · commit 325638d55ad9 · 2017-07-19T16:42:13.000-04:00
diff --git a/README.md b/README.md
@@ -33,18 +33,18 @@ print about.photo
 
 ```python
 files_by_type = {
-	'slides': gdrive.list_files('presentation'),
-	'folders': gdrive.list_files('folder'),
-	'spreadsheets': gdrive.list_files('spreadsheets'),
+    'slides': gdrive.list_files('presentation'),
+    'folders': gdrive.list_files('folder'),
+    'spreadsheets': gdrive.list_files('spreadsheets'),
 }
 
 for file in files_by_type['folders']:
-	print file.id
-	print file.name
+    print file.id
+    print file.name
 
 for file in files_by_type['spreadsheets']:
-	# prints list of parent folder IDs
-	print file.parents
+    # prints list of parent folder IDs
+    print file.parents
 
 # ...
 ```
@@ -74,32 +74,32 @@ presentation = gslides.get_presentation('PRESENTATION_ID')
 
 # print slides attributes
 for slide in presentation:
-	print slide.id
+    print slide.id
 
-	for element in slide: # equivalent to 'for element in presentation.elements()' 	
-		print element.type 
-		# Shape, Table, etc
+    for element in slide: # equivalent to 'for element in presentation.elements()'  
+      print element.type 
+      # Shape, Table, etc
 ```
 
 - [x] Check text in shape:
 
 ```python
 shape = presentation.get_element_by_id('SHAPE_ID')
 for segment in shape.text:
-	print segment.text
+    print segment.text
 ```
 
 - [x] Batch update every cell in table:
 
 ```python
 # use with to perform batch updates in block
 with presentation as pres:
-	table = pres.get_element_by_id('TABLE_ID')
-	for cell in table:
-		print cell.location # tuple containing cell location
-		for segment in cell.text:
-			# update cell
-			segment.text = 'UPDATED_VALUE'
+    table = pres.get_element_by_id('TABLE_ID')
+    for cell in table:
+        print cell.location # tuple containing cell location
+        for segment in cell.text:
+            # update cell
+            segment.text = 'UPDATED_VALUE'
 ```
 
 ### Google Sheets v4
@@ -113,22 +113,22 @@ gsheets = SheetsClient(OAUTHLIB_CREDS)
 spreadsheet = gsheets.get_spreadsheet('SPREADSHEET_ID')
 
 for sheet in spreadsheet:
-	print sheet.id, sheet.title
+    print sheet.id, sheet.title
 ```
 
 - [x] Get sheet by name and return its full block of values:
 
 ```python
 sheet = spreadsheet['Sheet 1']
-values = sheet.values()	
+values = sheet.values() 
 ```
 
 - [x] Get named range value block:
 
 ```python
 named_ranges = spreadsheet.named_ranges('SHEET_NAME!A:C')
 for rng in named_range:
-	values = named_range.get_block()
+    values = named_range.get_block()
 ```
 
 - [x] Update values block:
@@ -137,8 +137,8 @@ for rng in named_range:
 values = spreadsheet.get_range('SHEET_NAME!A:C')
 # loop through rows
 for i, row in enumerate(values):
-	values[i] = [1, 2, 3]
-	print row
+    values[i] = [1, 2, 3]
+    print row
 values.update()
 
 # you can also use the slice syntax for updating..
@@ -150,5 +150,5 @@ values.update()
 
 ```python
 to_append = [[1, 2, 3], [4, 5, 6]]
-values.append(to_append)	
+values.append(to_append)  
 ```
diff --git a/google_objects/__init__.py b/google_objects/__init__.py
@@ -6,8 +6,8 @@
 
 from apiclient import discovery
 
-from .utils import set_private_attrs
-from .utils import keys_to_snake, keys_to_camel
+from google_objects.utils import set_private_attrs
+from google_objects.utils import keys_to_snake, keys_to_camel
 
 
 # sets default logging handler to avoid "No handler found" warnings.
diff --git a/google_objects/sheets/core.py b/google_objects/sheets/core.py
@@ -22,7 +22,6 @@
 
 
 def _value_to_cell(val):
-    val = val.encode('utf-8').strip()
     if val.isdigit():
         try:
             return {'userEnteredValue': {'numberValue': float(val)}}
@@ -89,6 +88,13 @@ def get_spreadsheet(self, id):
 
         return Spreadsheet.from_existing(data, self)
 
+    def create_spreadsheet_from_dataframes(self, *frames, **options):
+        if not frames:
+            raise ValueError
+
+        formatted_frames = [frm.values.tolist() for frm in frames]
+        return self.create_spreadsheet(self, formatted_frames, **options)
+
     def create_spreadsheet(self, sheets=[], **kwargs):
         data = self._resource.spreadsheets().create(
             body={
diff --git a/google_objects/sheets/updates.py b/google_objects/sheets/updates.py
@@ -1,4 +1,4 @@
-def FORMAT_ROW(sheet_id, start, end, red=0, green=0, blue=0):
+def FORMAT_ROW(sheet_id, start, end, rgba):
     return {
         "repeatCell": {
             "range": {
@@ -9,9 +9,10 @@ def FORMAT_ROW(sheet_id, start, end, red=0, green=0, blue=0):
             "cell": {
                 "userEnteredFormat": {
                     "backgroundColor": {
-                        "red": red,
-                        "green": green,
-                        "blue": blue
+                        "red": rgba[0],
+                        "green": rgba[1],
+                        "blue": rgba[2],
+                        'alpha': rgba[3]
                     }
                 }
             },
diff --git a/google_objects/utils.py b/google_objects/utils.py
@@ -92,3 +92,11 @@ def keys_to_camel(dt):
         dt[new_key] = dt.pop(key)
 
     return dt
+
+if __name__ == '__main__':
+    dt = {
+        'pageElements': 'test',
+        'otherPresentation': 3
+    }
+
+    print(keys_to_snake(dt))
