smartsheet-tools 0.0.6__tar.gz → 0.0.8__tar.gz

{smartsheet_tools-0.0.6/smartsheet_tools.egg-info → smartsheet_tools-0.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: smartsheet_tools
-Version: 0.0.6
+Version: 0.0.8
 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.6 → smartsheet_tools-0.0.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "smartsheet_tools"                       # from setup.py
-version = "0.0.6"                       # from setup.py
+version = "0.0.8"                       # from setup.py
 description = "A collection of convenience functions to aid with transitioning from simple-smartsheet to the SDK API and common tasks"
 readme = "README.md"
 requires-python = ">=3.9"               # matches classifiers (3.9–3.12)

{smartsheet_tools-0.0.6 → smartsheet_tools-0.0.8}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup
 
 setup(
     name="smartsheet_tools",
-    version="0.0.6",
+    version="0.0.8",
     description="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@ashi.digital",

{smartsheet_tools-0.0.6 → smartsheet_tools-0.0.8}/smartsheet_tools/__init__.py

@@ -1,6 +1,8 @@
 from datetime import datetime
+import time
 import re
-from smartsheet.models import Cell, Row, Folder, Sheet
+import warnings
+from smartsheet.models import Cell, Row, Folder, Sheet, Error
 from smartsheet.models import Column
 
 # Cache for column types to minimize API calls when correcting date formats
@@ -61,7 +63,11 @@ def get_cached_column_type(column_id, sheet_obj, prefill=False):
         
     if column_id not in _COLUMN_TYPE_CACHE[sheet_obj.id]:
         if not prefill:
-            _COLUMN_TYPE_CACHE[sheet_obj.id][column_id] = str(sheet_obj.get_column(column_id).type)
+            
+            # Value is not in there and no prefill, so look it up
+            for col in sheet_obj.columns:
+                _COLUMN_TYPE_CACHE[sheet_obj.id][column_id] = col.type
+
         else:
             _COLUMN_TYPE_CACHE[sheet_obj.id][column_id] = prefill
     
@@ -70,6 +76,9 @@ def get_cached_column_type(column_id, sheet_obj, prefill=False):
 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, prefill=c.type) in ("DATE", "DATETIME")]
 
+def get_col_names_of_bool_cols(sheet_obj):
+    return [c.title for c in sheet_obj.columns if get_cached_column_type(c.id, sheet_obj, prefill=c.type) == "CHECKBOX"]
+
 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):
@@ -91,16 +100,16 @@ 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)
+def correct_date_format(value, column_id, sheet_obj, nonetype_if_fail=False):
+    if isinstance(value, datetime):
+        value = datetime_to_isoformat(value)
 
     column_type = get_cached_column_type(column_id, sheet_obj)
     if column_type == "DATE":
-        return isoformat_datetime.split("T",1)[0]
+        return value.split("T",1)[0]
     elif column_type == "DATETIME":
-        return isoformat_datetime
-    return None
+        return value
+    return None if nonetype_if_fail else value
 
 def new_cell(column_id=None, value=None, strict=False, formula=None):
     new_cell = Cell()
@@ -156,6 +165,27 @@ def walk_workspace_for_folders(smartsheet_client, workspace_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
+
+def safe_grab_sheet_by_name(name, smartsheet_client, max_tries=5, delay_seconds=15):
+    last_error = None
+    for attempt in range(1, max_tries + 1):
+        try:
+            result = smartsheet_client.Sheets.get_sheet_by_name(name)
+        except Exception as exc:
+            last_error = exc
+            result = None
+
+        if isinstance(result, Sheet):
+            return result
+
+        if isinstance(result, Error):
+            last_error = result
+
+        if attempt < max_tries:
+            warnings.warn(f"failed to grab sheet {name} trying again in {delay_seconds}s")
+            time.sleep(delay_seconds)
+
+    raise RuntimeError(f"failed to grab sheet {name} after {max_tries} tries") from (last_error if isinstance(last_error, Exception) else None)
         
 def new_column(column_type, title, index=None, id=None, options=None, symbol=None, primary=False, hidden=False, locked=False):
     new_column = Column()
@@ -176,4 +206,4 @@ def new_column(column_type, title, index=None, id=None, options=None, symbol=Non
         new_column.hidden = True
     if locked:
         new_column.locked = True
-    return new_column
+    return new_column

{smartsheet_tools-0.0.6 → smartsheet_tools-0.0.8/smartsheet_tools.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: smartsheet_tools
-Version: 0.0.6
+Version: 0.0.8
 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>