Version bump 0.3.26 → 0.3.27, ATP readonly classifier: add CHECKPOINT to write keywords, regex-based whole-word matching, test coverage

Author: FanaticPythoner

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "kuzualchemy"
-version = "0.3.26"
+version = "0.3.27"
 description = "SQLAlchemy-like ORM for Kuzu graph database"
 readme = "README.md"
 license = { file = "LICENSE" }

src/kuzualchemy/atp_integration.py

@@ -228,11 +228,11 @@ def _is_readonly_query(self, query: str) -> bool:
         # Write clause keywords that indicate a non-readonly query
         write_keywords = (
             'CREATE', 'DELETE', 'DETACH', 'SET', 'REMOVE', 'MERGE',
-            'DROP', 'ALTER', 'COPY', 'CALL', 'BEGIN', 'COMMIT', 'ROLLBACK'
+            'DROP', 'ALTER', 'COPY', 'CALL', 'BEGIN', 'COMMIT', 'ROLLBACK',
+            'CHECKPOINT'
         )
         for kw in write_keywords:
-            # Check for keyword as whole word (not substring)
-            if f' {kw} ' in f' {q} ' or q.startswith(f'{kw} ') or f' {kw}(' in f' {q} ':
+            if re.search(rf"(^|\s){kw}(\s|\(|;|$)", q):
                 return False
         return True
 

tests/test_atp_readonly_classifier.py

@@ -0,0 +1,10 @@
+from __future__ import annotations
+
+from kuzualchemy.atp_integration import ATPIntegration
+
+
+def test_checkpoint_is_not_classified_as_readonly() -> None:
+    integration = object.__new__(ATPIntegration)
+
+    assert integration._is_readonly_query("CHECKPOINT;") is False
+    assert integration._is_readonly_query("MATCH (n) RETURN n") is True