smartsheet-tools 0.0.2__py3-none-any.whl → 0.0.3__py3-none-any.whl

smartsheet_tools/__init__.py

@@ -1,5 +1,9 @@
+from datetime import datetime
 import re
-from smartsheet.models import Cell, Row
+from smartsheet.models import Cell, Row, Folder, Sheet
+
+# Cache for column types to minimize API calls when correcting date formats
+_COLUMN_TYPE_CACHE = {}
 
 def norm(v):
     if v is None:
@@ -34,6 +38,60 @@ def guard_row(row, *idxs):
     # ensure row has enough cells for all requested positions
     return max(idxs) < len(row.cells)
 
+def datetime_to_isoformat(dt):
+    if dt is None:
+        return None
+    return dt.replace(microsecond=0).isoformat() + 'Z'
+
+def standard_time_to_isoformat(st):
+    if st is None:
+        return None
+    return datetime_to_isoformat(datetime.strptime(st, "%m/%d/%Y"))
+
+def get_cached_column_type(column_id, sheet_obj):
+    if sheet_obj.id not in _COLUMN_TYPE_CACHE:
+        _COLUMN_TYPE_CACHE[sheet_obj.id] = {}
+        
+    if column_id not in _COLUMN_TYPE_CACHE[sheet_obj.id]:
+        _COLUMN_TYPE_CACHE[sheet_obj.id][column_id] = str(sheet_obj.get_column(column_id).type)
+    
+    return _COLUMN_TYPE_CACHE[sheet_obj.id][column_id]
+
+def get_col_names_of_date_cols(sheet_obj):
+    return [c.title for c in sheet_obj.columns if get_cached_column_type(c.id, sheet_obj) in ("DATE", "DATETIME")]
+
+def brute_force_date_string(s, nonetype_if_fail=False):
+    # attempt to parse a date string in common formats to ISO 8601
+    if isinstance(s, datetime):
+        return datetime_to_isoformat(s)
+    
+    if not isinstance(s, str):
+        return None if nonetype_if_fail else s
+    
+    s = s.split(" ")[0]
+    for fmt in ("%Y-%m-%d", "%m/%d/%Y", "%m/%d/%y"):
+        try:
+            return datetime_to_isoformat(datetime.strptime(s, fmt))
+        except ValueError:
+            continue
+    return None if nonetype_if_fail else s
+
+
+def is_date_col(column_id, sheet_obj):
+    column_type = get_cached_column_type(column_id, sheet_obj)
+    return column_type in ("DATE", "DATETIME")
+
+def correct_date_format(isoformat_datetime, column_id, sheet_obj):
+    if isinstance(isoformat_datetime, datetime):
+        isoformat_datetime = datetime_to_isoformat(isoformat_datetime)
+
+    column_type = get_cached_column_type(column_id, sheet_obj)
+    if column_type == "DATE":
+        return isoformat_datetime.split("T",1)[0]
+    elif column_type == "DATETIME":
+        return isoformat_datetime
+    return None
+
 def new_cell(column_id, value=None, strict=False, formula=None):
     new_cell = Cell()
     new_cell.column_id = column_id
@@ -45,12 +103,44 @@ def new_cell(column_id, value=None, strict=False, formula=None):
         new_cell.strict = True
     return new_cell
 
-def new_row(cells=None, parent_id=None, to_top=False):
+def new_row(cells=None, id=None, parent_id=None, to_top=False):
     new_row = Row()
     if cells:
         new_row.cells = cells
+    if id:
+        new_row.id = id
     if parent_id:
         new_row.parent_id = parent_id
     if to_top:
         new_row.to_top = to_top
-    return new_row
+    return new_row
+
+def walk_folder_for_sheets(smartsheet_client, folder_id):
+    for item in smartsheet_client.Folders.get_folder_children(folder_id).data:
+        if isinstance(item, Folder):
+            yield from walk_folder_for_sheets(smartsheet_client, item.id)
+        elif isinstance(item, Sheet):
+            yield item
+
+def walk_workspace_for_sheets(smartsheet_client, workspace_id):
+    for item in smartsheet_client.Workspaces.get_workspace_children(workspace_id).data:
+        if isinstance(item, Folder):
+            yield from walk_folder_for_sheets(smartsheet_client, item.id)
+        elif isinstance(item, Sheet):
+            yield item
+            
+def walk_folder_for_folders(smartsheet_client, folder_id):
+    for item in smartsheet_client.Folders.get_folder_children(folder_id).data:
+        if isinstance(item, Folder):
+            yield item
+            yield from walk_folder_for_folders(smartsheet_client, item.id)
+            
+def walk_workspace_for_folders(smartsheet_client, workspace_id):
+    for item in smartsheet_client.Workspaces.get_workspace_children(workspace_id).data:
+        if isinstance(item, Folder):
+            yield item
+            yield from walk_folder_for_folders(smartsheet_client, item.id)
+            
+def walk_sheet_names_from_workspace(smartsheet_client, workspace_id):
+    for sheet in walk_workspace_for_sheets(smartsheet_client, workspace_id):
+        yield sheet.name

{smartsheet_tools-0.0.2.dist-info → smartsheet_tools-0.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: smartsheet_tools
-Version: 0.0.2
+Version: 0.0.3
 Summary: A collection of convenience functions to aid with transitioning from simple-smartsheet to the SDK API and common tasks
 Author: Ashton Pooley
 Author-email: Ashton Pooley <ashton@ashi.digital>

smartsheet_tools-0.0.3.dist-info/RECORD

@@ -0,0 +1,6 @@
+smartsheet_tools/__init__.py,sha256=-0zRzn_7MUDL4sFfNzXRWTJj0MUY7NZ96O_5rSTmJiE,5268
+smartsheet_tools-0.0.3.dist-info/licenses/LICENSE,sha256=xshMXNQ83e1x1bG3-9fQ5U8hnMaJsv79ke3xuKmI2PI,31914
+smartsheet_tools-0.0.3.dist-info/METADATA,sha256=X_Tkf-hEspLuLBHreCq65BOSeKaBKi258LkNiL8sab8,834
+smartsheet_tools-0.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+smartsheet_tools-0.0.3.dist-info/top_level.txt,sha256=UXKUTK6mn1resx7hDN-MqSLi6ZnojUbkJ44VzmNmYi8,17
+smartsheet_tools-0.0.3.dist-info/RECORD,,

smartsheet_tools-0.0.2.dist-info/RECORD

@@ -1,6 +0,0 @@
-smartsheet_tools/__init__.py,sha256=LVXn-fARI98McLcctaxpF0SD7_RCh6Uu6RVxJmzMMTE,1683
-smartsheet_tools-0.0.2.dist-info/licenses/LICENSE,sha256=xshMXNQ83e1x1bG3-9fQ5U8hnMaJsv79ke3xuKmI2PI,31914
-smartsheet_tools-0.0.2.dist-info/METADATA,sha256=yD3p6zD6w9xi1oKAA6bszAzhThot4EOa_2aHnikX4sw,834
-smartsheet_tools-0.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-smartsheet_tools-0.0.2.dist-info/top_level.txt,sha256=UXKUTK6mn1resx7hDN-MqSLi6ZnojUbkJ44VzmNmYi8,17
-smartsheet_tools-0.0.2.dist-info/RECORD,,