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

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

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

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "smartsheet_tools"                       # from setup.py
-version = "0.0.3"                       # from setup.py
+version = "0.0.6"                       # 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.3 → smartsheet_tools-0.0.6}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup
 
 setup(
     name="smartsheet_tools",
-    version="0.0.3",
+    version="0.0.6",
     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.3 → smartsheet_tools-0.0.6}/smartsheet_tools/__init__.py

@@ -1,9 +1,12 @@
 from datetime import datetime
 import re
 from smartsheet.models import Cell, Row, Folder, Sheet
+from smartsheet.models import Column
 
 # Cache for column types to minimize API calls when correcting date formats
 _COLUMN_TYPE_CACHE = {}
+_TITLE_TO_ID_CACHE = {}
+_ID_TO_INDEX_CACHE = {}
 
 def norm(v):
     if v is None:
@@ -18,11 +21,15 @@ def disp_or_val(cell):
 
 def title_to_index(sheet):
     # authoritative positions from Smartsheet (not Python enumerate order)
-    return {c.title: c.index for c in sheet.columns}
+    if sheet.id not in _TITLE_TO_ID_CACHE:
+        _TITLE_TO_ID_CACHE[sheet.id] = {c.title: c.index for c in sheet.columns}
+    return _TITLE_TO_ID_CACHE[sheet.id]
 
 def index_to_id(sheet):
     # authoritative positions from Smartsheet (not Python enumerate order)
-    return {c.index: c.id for c in sheet.columns}
+    if sheet.id not in _ID_TO_INDEX_CACHE:
+        _ID_TO_INDEX_CACHE[sheet.id] = {c.index: c.id for c in sheet.columns}
+    return _ID_TO_INDEX_CACHE[sheet.id]
 
 def id_to_index(sheet):
     # authoritative positions from Smartsheet (not Python enumerate order)
@@ -48,17 +55,20 @@ def standard_time_to_isoformat(st):
         return None
     return datetime_to_isoformat(datetime.strptime(st, "%m/%d/%Y"))
 
-def get_cached_column_type(column_id, sheet_obj):
+def get_cached_column_type(column_id, sheet_obj, prefill=False):
     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)
+        if not prefill:
+            _COLUMN_TYPE_CACHE[sheet_obj.id][column_id] = str(sheet_obj.get_column(column_id).type)
+        else:
+            _COLUMN_TYPE_CACHE[sheet_obj.id][column_id] = prefill
     
     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")]
+    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 brute_force_date_string(s, nonetype_if_fail=False):
     # attempt to parse a date string in common formats to ISO 8601
@@ -92,18 +102,18 @@ def correct_date_format(isoformat_datetime, column_id, sheet_obj):
         return isoformat_datetime
     return None
 
-def new_cell(column_id, value=None, strict=False, formula=None):
+def new_cell(column_id=None, value=None, strict=False, formula=None):
     new_cell = Cell()
-    new_cell.column_id = column_id
+    if column_id is not None:
+        new_cell.column_id = column_id
     if formula is not None:
         new_cell.formula = formula
     else:
         new_cell.value = value
-    if strict:
-        new_cell.strict = True
+    new_cell.strict = strict
     return new_cell
 
-def new_row(cells=None, id=None, parent_id=None, to_top=False):
+def new_row(cells=None, id=None, parent_id=None, to_top=False, locked=False):
     new_row = Row()
     if cells:
         new_row.cells = cells
@@ -113,6 +123,8 @@ def new_row(cells=None, id=None, parent_id=None, to_top=False):
         new_row.parent_id = parent_id
     if to_top:
         new_row.to_top = to_top
+    if locked:
+        new_row.locked = locked
     return new_row
 
 def walk_folder_for_sheets(smartsheet_client, folder_id):
@@ -143,4 +155,25 @@ 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
+        yield sheet.name
+        
+def new_column(column_type, title, index=None, id=None, options=None, symbol=None, primary=False, hidden=False, locked=False):
+    new_column = Column()
+    
+    new_column.type = column_type
+    new_column.title = title
+    if index is not None:
+        new_column.index = index
+    if id is not None:
+        new_column.id = id
+    if options is not None and column_type in ("PICKLIST", "MULTI_PICKLIST"):
+        new_column.options = options
+    if symbol is not None and column_type == "CHECKBOX":
+        new_column.symbol = symbol
+    if primary:
+        new_column.primary = True
+    if hidden:
+        new_column.hidden = True
+    if locked:
+        new_column.locked = True
+    return new_column

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

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