column names from user could start with numbers

opendp · Nov 26, 2024 · ef97a5f · ef97a5f
1 parent b7669f5
commit ef97a5f
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/dp_wizard/utils/code_generators/__init__.py b/dp_wizard/utils/code_generators/__init__.py
@@ -244,8 +244,14 @@ def _snake_case(name: str):
     """
     >>> _snake_case("HW GRADE")
     'hw_grade'
+    >>> _snake_case("123")
+    '_123'
     """
-    return re.sub(r"\W+", "_", name.lower())
+    snake = re.sub(r"\W+", "_", name.lower())
+    # TODO: More validation in UI so we don't get zero-length strings.
+    if snake == "" or not re.match(r"[a-z]", snake[0]):
+        snake = f"_{snake}"
+    return snake
 
 
 def _make_imports():