ssc_codegen 0.2.5__tar.gz → 0.2.7__tar.gz

{ssc_codegen-0.2.5 → ssc_codegen-0.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ssc_codegen
-Version: 0.2.5
+Version: 0.2.7
 Summary: generate selector schemas classes from dsl-like language based on python
 License: MIT
 Author: vypivshiy

{ssc_codegen-0.2.5 → ssc_codegen-0.2.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ssc_codegen"
-version = "0.2.5"
+version = "0.2.7"
 description = "generate selector schemas classes from dsl-like language based on python"
 authors = ["vypivshiy"]
 readme = "README.md"

{ssc_codegen-0.2.5 → ssc_codegen-0.2.7}/ssc_codegen/document.py

@@ -1,6 +1,6 @@
 import warnings
 from contextlib import contextmanager
-from typing import Optional
+from typing import Optional, Union
 
 from ssc_codegen.objects import VariableState, Expression, TokenType
 from ssc_codegen.queries import (
@@ -373,4 +373,14 @@ class Document(
 ):
     """Base document-like attribute provide DSL parser logic for generating code"""
 
-    pass
+    def __getitem__(self, item: Union[int, str]):
+        """if item str invoke HtmlOpDocument.attr(item)
+        if item int - ArrayOpDocument.index(item)
+        """
+        if isinstance(item, str):
+            return self.attr(item)
+        elif isinstance(item, int):
+            return self.index(item)
+        wrong_type = type(item).__name__
+        msg = f"should be str or int, not '{wrong_type}'"
+        raise TypeError(msg)

{ssc_codegen-0.2.5 → ssc_codegen-0.2.7}/ssc_codegen/objects.py

@@ -8,19 +8,19 @@ if TYPE_CHECKING:
 
 
 class VariableState(IntEnum):
-    DOCUMENT = 0
-    LIST_DOCUMENT = 1
-    STRING = 2
-    LIST_STRING = 3
-    NONE = 4
-    DOCSTRING = 5  # ///
+    DOCUMENT = 1
+    LIST_DOCUMENT = 2
+    STRING = 3
+    LIST_STRING = 4
+    NONE = 5
+    DOCSTRING = 6  # ///
 
 
 class TokenType(IntEnum):
     """all command enum representation"""
 
     # SELECTORS
-    OP_XPATH = 0
+    OP_XPATH = 300
     OP_XPATH_ALL = 1
     OP_CSS = 2
     OP_CSS_ALL = 3