ssc_codegen 0.7.5__tar.gz → 0.7.6__tar.gz

{ssc_codegen-0.7.5 → ssc_codegen-0.7.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ssc_codegen
-Version: 0.7.5
+Version: 0.7.6
 Summary: Python-dsl code converter to html parser for web scraping 
 Project-URL: Documentation, https://github.com/vypivshiy/selector_schema_codegen#readme
 Project-URL: Issues, https://github.com/vypivshiy/selector_schema_codegen/issues

{ssc_codegen-0.7.5 → ssc_codegen-0.7.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ssc_codegen"
-version = "0.7.5"
+version = "0.7.6"
 description = "Python-dsl code converter to html parser for web scraping "
 readme = "README.md"
 requires-python = ">=3.10"

{ssc_codegen-0.7.5 → ssc_codegen-0.7.6}/ssc_codegen/__init__.py

@@ -3,7 +3,7 @@ from .document import HTMLDocument, StringDocument, ArrayDocument, AssertDocumen
 from .json_struct import Json
 from .schema import ItemSchema, DictSchema, ListSchema, FlatListSchema
 
-VERSION = "0.7.5"
+VERSION = "0.7.6"
 
 class __MISSING(object):
     pass

{ssc_codegen-0.7.5 → ssc_codegen-0.7.6}/ssc_codegen/ast_build_utils.py

@@ -126,6 +126,10 @@ def extract_json_structs_from_module(module: ModuleType) -> list[Type[Json]]:
 def assert_ret_type_not_document(
     field: "BaseDocument", name: str, schema: Type[BaseSchema]
 ) -> None:
+    # always ret nil, skip
+    if name == "__PRE_VALIDATE__":
+        return
+
     if field.stack_last_ret == VariableType.DOCUMENT:
         msg = f"{schema.__name__}.{name} cannot return type {VariableType.DOCUMENT.name}"
         raise TypeError(msg)
@@ -222,6 +226,19 @@ def assert_field_document_variable_types(field: BaseDocument) -> None:
         if var_cursor == expr.accept_type:
             var_cursor = expr.ret_type
             continue
+        # currently used in to_len() operation
+        elif expr.accept_type == VariableType.LIST_ANY:
+            if var_cursor in (
+                VariableType.LIST_STRING,
+                VariableType.LIST_DOCUMENT,
+                VariableType.LIST_FLOAT,
+                VariableType.LIST_INT,
+            ):
+                var_cursor = expr.ret_type
+                continue
+            msg = f".to_len() expected LIST_ANY type, got '{var_cursor.name}'"
+            raise TypeError(msg)
+
         # this type always first (naive, used in DEFAULT and RETURN expr)
         elif expr.accept_type == VariableType.ANY:
             var_cursor = VariableType.DOCUMENT
@@ -232,5 +249,6 @@ def assert_field_document_variable_types(field: BaseDocument) -> None:
         elif var_cursor == VariableType.NESTED:
             raise TypeError("sub_parser not allowed next instructions")
 
+        # not covered cases
         msg = f"'{expr.kind.name}' expected type '{expr.accept_type.name}', got '{var_cursor.name}'"
         raise TypeError(msg)

{ssc_codegen-0.7.5 → ssc_codegen-0.7.6}/ssc_codegen/ast_ssc.py

@@ -427,12 +427,7 @@ class JoinExpression(BaseExpression):
 @dataclass(kw_only=True)
 class ArrayLengthExpression(BaseExpression):
     kind: ClassVar[TokenType] = TokenType.EXPR_LIST_LEN
-    accept_type: VariableType = (
-        VariableType.LIST_STRING
-        | VariableType.LIST_DOCUMENT
-        | VariableType.LIST_INT
-        | VariableType.LIST_FLOAT
-    )
+    accept_type: VariableType = VariableType.LIST_ANY
     ret_type: VariableType = VariableType.INT
 
 

{ssc_codegen-0.7.5 → ssc_codegen-0.7.6}/ssc_codegen/tokens.py

@@ -14,6 +14,10 @@ class VariableType(IntEnum):
     # mark accept any variable type
     # used in return and default expr
     ANY = auto()
+
+    # currently used for to_len() operation
+    LIST_ANY = auto()
+
     # nested functions
     NESTED = auto()
     INT = auto()