chore: minor improvements (#103)

Minor fixes.

platformics/codegen/generator.py

@@ -5,7 +5,9 @@
 
 import logging
 import os
+import re
 
+import jinja2.ext
 from jinja2 import Environment, FileSystemLoader
 from linkml_runtime.utils.schemaview import SchemaView
 
@@ -122,6 +124,16 @@ def generate_entity_import_files(
             print(f"... wrote {filename}")
 
 
+def regex_replace(txt, rgx, val, ignorecase=False, multiline=False):
+    flag = 0
+    if ignorecase:
+        flag |= re.I
+    if multiline:
+        flag |= re.M
+    compiled_rgx = re.compile(rgx, flag)
+    return compiled_rgx.sub(val, txt)
+
+
 def generate(schemafile: str, output_prefix: str, render_files: bool, template_override_paths: tuple[str]) -> None:
     """
     Launch code generation
@@ -130,7 +142,8 @@ def generate(schemafile: str, output_prefix: str, render_files: bool, template_o
     template_paths.append(
         os.path.join(os.path.abspath(os.path.dirname(__file__)), "templates/"),
     )  # default template path
-    environment = Environment(loader=FileSystemLoader(template_paths))
+    environment = Environment(loader=FileSystemLoader(template_paths), extensions=[jinja2.ext.loopcontrols])
+    environment.filters["regex_replace"] = regex_replace
     view = SchemaView(schemafile)
     view.imports_closure()
     wrapped_view = ViewWrapper(view)

platformics/codegen/templates/database/migrations/env.py.j2

@@ -43,6 +43,8 @@ def run_migrations_offline() -> None:
     )
 
     with context.begin_transaction():
+        context.get_context()._ensure_version_table()  # pylint: disable=protected-access
+        connection.execute(sa.sql.text("LOCK TABLE alembic_version IN ACCESS EXCLUSIVE MODE"))
         context.run_migrations()
 
 

platformics/codegen/templates/support/enums.py.j2

@@ -13,6 +13,7 @@ import enum
 @strawberry.enum
 class {{enum.name}}(enum.Enum):
     {%- for value in enum.permissible_values %}
-    {{value}} = "{{value}}"
+    {#- SQLAlchemy freaks out about spaces in enum values :'( #}
+    {{value | regex_replace('[^0-9A-Za-z]', '_') }} = "{{value | regex_replace('[^0-9A-Za-z]', '_') }}"
     {%- endfor %}
 {%- endfor %}