vex-ast 0.2.2__py3-none-any.whl → 0.2.3__py3-none-any.whl

vex_ast/ast/vex_nodes.py

@@ -91,8 +91,21 @@ class VexAPICall(FunctionCall):
             return False, self._validation_error
         
         # Convert args and kwargs to appropriate format
-        arg_values = [arg for arg in self.args]
-        kwarg_values = {kw.name: kw.value for kw in self.keywords or []}
+        arg_values = []
+        for arg in self.args:
+            # For string literals, use their actual string value for validation
+            if hasattr(arg, 'value') and hasattr(arg, '__class__') and arg.__class__.__name__ == 'StringLiteral':
+                arg_values.append(arg.value)
+            else:
+                arg_values.append(arg)
+        
+        kwarg_values = {}
+        for kw in (self.keywords or []):
+            # For string literals, use their actual string value for validation
+            if hasattr(kw.value, 'value') and hasattr(kw.value, '__class__') and kw.value.__class__.__name__ == 'StringLiteral':
+                kwarg_values[kw.name] = kw.value.value
+            else:
+                kwarg_values[kw.name] = kw.value
         
         # Validate against the signature
         valid, error = signature.validate_arguments(arg_values, kwarg_values)

vex_ast/parser/factory.py

@@ -123,6 +123,12 @@ class NodeFactory:
         """Create a function definition node."""
         return FunctionDefinition(name, args, body, return_annotation, location)
     
+    def create_argument(self, name: str, annotation: Optional[Expression] = None,
+                       default: Optional[Expression] = None,
+                       location: Optional[SourceLocation] = None) -> Argument:
+        """Create an argument node."""
+        return Argument(name, annotation, default, location)
+    
     def create_return_statement(self, value: Optional[Expression] = None,
                                location: Optional[SourceLocation] = None) -> ReturnStatement:
         """Create a return statement node."""
@@ -176,4 +182,4 @@ class NodeFactory:
         return Program(body, location)
 
 # Global factory instance for simple use cases
-default_factory = NodeFactory()
+default_factory = NodeFactory()

vex_ast/parser/python_parser.py

@@ -692,7 +692,13 @@ class PythonParser(BaseParser):
                 f"Syntax error: {e.msg}",
                 loc
             )
-            raise VexSyntaxError(f"Syntax error: {e.msg}", loc) from e
+            
+            # Only raise if the error handler is configured to do so
+            if self.error_handler._raise_on_error:
+                raise VexSyntaxError(f"Syntax error: {e.msg}", loc) from e
+            
+            # Return an empty program if we're not raising
+            return self.factory.create_program([])
             
         except Exception as e:
             # Handle other parsing errors

vex_ast/registry/signature.py

@@ -138,6 +138,49 @@ class VexFunctionSignature:
             if param_name in kwargs:
                 return False, f"Duplicate argument '{param_name}' for {self.name}"
         
-        # TODO: Add type checking for arguments
+        # Type checking for arguments
+        from ..types.type_checker import type_checker
+        from ..types.enums import EnumType
+        
+        # Check positional arguments
+        for i, arg in enumerate(args):
+            if i >= len(self.parameters):
+                break
+                
+            param = self.parameters[i]
+            expected_type = param.type
+            
+            # Handle string literals for enum types
+            if isinstance(expected_type, EnumType) and isinstance(arg, str):
+                if arg not in expected_type.values:
+                    return False, f"Invalid enum value '{arg}' for parameter '{param.name}' in {self.name}"
+                continue
+                
+            # Handle other types
+            if hasattr(arg, 'get_type'):
+                arg_type = arg.get_type()
+                if arg_type and not type_checker.is_compatible(arg_type, expected_type):
+                    return False, f"Type mismatch for parameter '{param.name}' in {self.name}: expected {expected_type}, got {arg_type}"
+        
+        # Check keyword arguments
+        for kwarg_name, kwarg_value in kwargs.items():
+            # Find the parameter
+            param = next((p for p in self.parameters if p.name == kwarg_name), None)
+            if not param:
+                continue  # Already checked for unknown kwargs above
+                
+            expected_type = param.type
+            
+            # Handle string literals for enum types
+            if isinstance(expected_type, EnumType) and isinstance(kwarg_value, str):
+                if kwarg_value not in expected_type.values:
+                    return False, f"Invalid enum value '{kwarg_value}' for parameter '{param.name}' in {self.name}"
+                continue
+                
+            # Handle other types
+            if hasattr(kwarg_value, 'get_type'):
+                kwarg_type = kwarg_value.get_type()
+                if kwarg_type and not type_checker.is_compatible(kwarg_type, expected_type):
+                    return False, f"Type mismatch for parameter '{param.name}' in {self.name}: expected {expected_type}, got {kwarg_type}"
         
         return True, None

vex_ast/serialization/json_deserializer.py

@@ -10,7 +10,7 @@ from typing import Any, Dict, List, Optional, Type, Union, cast
 from ..ast.interfaces import IAstNode
 from ..parser.factory import NodeFactory
 from ..utils.source_location import SourceLocation
-from ..utils.errors import ErrorHandler
+from ..utils.errors import ErrorHandler, ErrorType
 
 
 class DeserializationFactory:
@@ -153,6 +153,7 @@ class DeserializationFactory:
             "WhileLoop": self.node_factory.create_while_loop,
             "ForLoop": self.node_factory.create_for_loop,
             "FunctionDefinition": self.node_factory.create_function_definition,
+            "Argument": self.node_factory.create_argument,
             "ReturnStatement": self.node_factory.create_return_statement,
             "BreakStatement": self.node_factory.create_break_statement,
             "ContinueStatement": self.node_factory.create_continue_statement,
@@ -198,11 +199,21 @@ class DeserializationFactory:
         except TypeError as e:
             # If the factory method doesn't accept the kwargs, report an error
             if self.error_handler:
-                self.error_handler.report_error(f"Failed to create {node_type}: {str(e)}")
+                self.error_handler.add_error(
+                    error_type=ErrorType.INTERNAL_ERROR,
+                    message=f"Failed to create {node_type}: {str(e)}"
+                )
             raise ValueError(f"Failed to deserialize {node_type}: {str(e)}")
     
     # Specific node creation methods for complex cases
     
+    def _create_attributeaccess(self, data: Dict[str, Any],
+                               location: Optional[SourceLocation]) -> IAstNode:
+        """Create an AttributeAccess node from serialized data."""
+        object_expr = self._deserialize_value(data.get("object"))
+        attribute = data.get("attribute", "")
+        return self.node_factory.create_attribute_access(object_expr, attribute, location)
+    
     def _create_program(self, data: Dict[str, Any], 
                        location: Optional[SourceLocation]) -> IAstNode:
         """Create a Program node from serialized data."""

vex_ast/serialization/schema.py

@@ -453,16 +453,15 @@ def _generate_definitions() -> Dict[str, Any]:
     return definitions
 
 
-def export_schema_to_file(filepath: str, indent: int = 2) -> None:
+def export_schema_to_file(schema: Dict[str, Any], filepath: str, indent: int = 2) -> None:
     """
-    Save the generated schema to a file.
+    Save the schema to a file.
     
     Args:
+        schema: The schema to save
         filepath: The path to save the schema to
         indent: The indentation level for pretty-printing
     """
-    schema = generate_ast_schema()
-    
     # Ensure the directory exists
     os.makedirs(os.path.dirname(os.path.abspath(filepath)), exist_ok=True)
     

vex_ast/utils/errors.py

@@ -50,7 +50,7 @@ class ErrorObserver(Protocol[T_Error]):
 class ErrorHandler:
     """Manages error collection and notification."""
     
-    def __init__(self, raise_on_error: bool = False):
+    def __init__(self, raise_on_error: bool = True):
         self._errors: List[Error] = []
         self._raise_on_error = raise_on_error
         self._observers: List[ErrorObserver] = []
@@ -109,4 +109,4 @@ class VexSyntaxError(VexAstError):
     
     def __init__(self, message: str, location: Optional[SourceLocation] = None):
         self.location = location
-        super().__init__(message)
+        super().__init__(message)

{vex_ast-0.2.2.dist-info → vex_ast-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vex_ast
-Version: 0.2.2
+Version: 0.2.3
 Summary: A Python package for generating Abstract Syntax Trees for VEX V5 code.
 Home-page: https://github.com/heartx2/vex_ast
 Author: Chaze

{vex_ast-0.2.2.dist-info → vex_ast-0.2.3.dist-info}/RECORD

@@ -11,12 +11,12 @@ vex_ast/ast/navigator.py,sha256=9DaVXrknBbBr4omAAMHQnZL9Wpj5wjtoCS6_lni8MYM,7529
 vex_ast/ast/operators.py,sha256=I-yWvhsrz-OxmBZs5zIss_GTZF5S-nwcSmIzvAVtddM,3160
 vex_ast/ast/statements.py,sha256=OWRthjYGmTuNozYAHjh_Enp5t-hR1PphtPnFg31FeWw,11841
 vex_ast/ast/validators.py,sha256=ySpi5H0XRdGXXMFAfmuVEyc3Q9mMXhubZAkP4N8NL3c,4717
-vex_ast/ast/vex_nodes.py,sha256=HmUPfiogz0rnObyhg1XEsxPIDdM5xbNxl_kW9NND8pQ,11345
+vex_ast/ast/vex_nodes.py,sha256=8Wq9vl8Mp70E3fxEut9tB6WZE1z0Q_i5V0qh_5Zoeag,11981
 vex_ast/parser/README.md,sha256=P1qq_97skpgluLDpNu9V_pAdkuF9StjkzOEXJJYpEuM,2565
 vex_ast/parser/__init__.py,sha256=LGHnFm3UzR4Nw7royGH3c_2RqeY66y8O6DdXHbm9yL4,504
-vex_ast/parser/factory.py,sha256=gaje0jOPu5lpndgX9c3VuQ_q-FJuFz9yC5y2bD69lL4,8770
+vex_ast/parser/factory.py,sha256=ryTs8dQej8GbATyc7H4NUUISL1Bq65eIjyq84HaFTlo,9098
 vex_ast/parser/interfaces.py,sha256=Ttc0bD_5X420ZCT9MLUf_wE1aZLWLkaJRQqCwBy9wIs,956
-vex_ast/parser/python_parser.py,sha256=e1owk-c6Rb4Bt5SYcgB-urZzqsX5xC9tzcK1sB8gkr4,29687
+vex_ast/parser/python_parser.py,sha256=EHcro2NqVLScOBBG0XuWUMiT7Dmqep4M9scGOxpJoYw,29947
 vex_ast/parser/strategies.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vex_ast/registry/README.md,sha256=J8xEXEscEZhWa6wmn05kjE395AjrV7p2h7nq4SXI5uE,1535
 vex_ast/registry/__init__.py,sha256=dXT8OS7uLyBSdQKZDWrtbANzgSWcg9HUQDA1EaLOvjQ,1188
@@ -24,7 +24,7 @@ vex_ast/registry/api.py,sha256=yyW-TsCIULTzePAoyMvEqD_RNlkFSb4PLpiSkw6ueQk,5460
 vex_ast/registry/categories.py,sha256=y5tnCzHQ9PPg9hOzOMgyE1ZAhcsjgeFSrgqdJrdqHEg,4885
 vex_ast/registry/language_map.py,sha256=eekN3k-K9g90GFi4--If3nfJCyarUJxDWDCDvLm7mdU,2656
 vex_ast/registry/registry.py,sha256=gvsQKyeXWp84T2uDB4O_WQgKCeq8W9udl8TD5MnPZzc,6661
-vex_ast/registry/signature.py,sha256=m8gQKdc5IaORDdfdBX6_qH1OL5LoTzveGoWlIIaOES0,5763
+vex_ast/registry/signature.py,sha256=_Lde3bteVVdYG31fN4v62UViWSVyxi4U3mYT1-9b4A0,7881
 vex_ast/registry/simulation_behavior.py,sha256=WwkVKae0owtv3WTrfnVH93rpsh9lbqJ_RSAVmpIVA_k,313
 vex_ast/registry/validation.py,sha256=xq5UjSWYAcRxjxsCGg3WU5RtPN-DB-JAqKqDNGvnJGk,1973
 vex_ast/registry/functions/__init__.py,sha256=X9aw0-Y9Q6Gcx01zepP8G_20ybfwC-LDPQfkK2RDOrE,207
@@ -35,9 +35,9 @@ vex_ast/registry/functions/motor.py,sha256=Vby44hV1goNFgFQEFDlv7z_9uGQeuvLMABplO
 vex_ast/registry/functions/sensors.py,sha256=BdhLV5gSWD5HO-o3JRXhbGQF7s1Ybk3b6VLLqVau1i0,6410
 vex_ast/registry/functions/timing.py,sha256=DqwMVPk7VKDeZB4S3NI7wOYgM1R52dcdgaVIb9135Jg,3204
 vex_ast/serialization/__init__.py,sha256=qPTEiMjU8hpVxNH5z4MY7Yj60AxuFurjXZStjhWWf6o,835
-vex_ast/serialization/json_deserializer.py,sha256=m_xiEUEoFXGfFylK3M8BVs7F2EieQJ9wbUa_ZTpkHtQ,10523
+vex_ast/serialization/json_deserializer.py,sha256=GCfm8XAF_90FswlizFy59fyZwjZtE94yPR_FWWWAE20,11108
 vex_ast/serialization/json_serializer.py,sha256=YvMRUpqXjtbxlIvathEIywUqbH3Ne6GSRODfFB0QCGM,4465
-vex_ast/serialization/schema.py,sha256=zpKujT7u3Qw2kNbwjV09lNzeQAsCxvMx-uAdaRMEstA,14431
+vex_ast/serialization/schema.py,sha256=BNBYruht3fmuGoVGv5fuMwdIEqLu7OZ1XZS8gzdom0w,14440
 vex_ast/types/README.md,sha256=Wd3jBShiNXNc3iJ69Qps5_0mBq8QVEd_Gz5OachfS34,1029
 vex_ast/types/__init__.py,sha256=naLOT_-qHWxzYj4nwxjCB5dfL6tcIt7TjMEaRaXMWbU,2163
 vex_ast/types/base.py,sha256=hCPCeBNnD5p29Mim-ouRTkG6Lfa8NXrsdYLO8xsbFtM,2258
@@ -47,7 +47,7 @@ vex_ast/types/primitives.py,sha256=t_4kEVyPSmKRHEIRQcp-X5Yq46JbG1SxzlvHb0vL4IA,1
 vex_ast/types/type_checker.py,sha256=emzhmc6AlH71w0DrKLlZRMBlJNZAuENihvRXTenqg_Q,1083
 vex_ast/utils/README.md,sha256=Y9RJMQTqQbpjVkvYJmpeRihD1zfW9PhNL_LgoDJ84ak,1945
 vex_ast/utils/__init__.py,sha256=azzhhFoMykqOkVVm6X2V7dFW3jGBkOvEgnEP1JTCS3g,621
-vex_ast/utils/errors.py,sha256=uq_jyu-YasifXjKSpPbVxup-XFf0XSHmp7A0Zd87TUY,3850
+vex_ast/utils/errors.py,sha256=T1sOqWR3-0dYqmXAKDwO7VXIHCGZMnF_gEIGmFWy8Ps,3850
 vex_ast/utils/source_location.py,sha256=r857ypqUBqex2y_R0RzZo7pz5di9qtEeWFFQ4HwzYQE,1297
 vex_ast/utils/type_definitions.py,sha256=2rB85B1vZL1GLWItBuJJpyr-vmp3W346EMMX2TPZ99Q,313
 vex_ast/visitors/README.md,sha256=BKDj8LMuBlCrzgSSLmCkbET7WIqgxe-oCkqQbqXekrE,2725
@@ -56,8 +56,8 @@ vex_ast/visitors/analyzer.py,sha256=cBT5PRbtfQ_Dt1qiHN-wdO5KnC2yl-Ul5RasXg9geHY,
 vex_ast/visitors/base.py,sha256=Ph0taTIVMHV_QjXfDd0ipZnCXVJErkyRtPpfwfzo-kY,4859
 vex_ast/visitors/printer.py,sha256=CUY_73hxm7MoC-63JeIXaVYnZ8QqtfMqbek2R2HMEmE,5411
 vex_ast/visitors/transformer.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vex_ast-0.2.2.dist-info/licenses/LICENSE,sha256=IOSlfCuxGv4OAg421BRDKVi16RZ7-5kCMJ4B16r4kuc,69
-vex_ast-0.2.2.dist-info/METADATA,sha256=Rftzrxm-exQp3zpV5wcNVkErbkQn1ww0pz65tXOQhDw,5295
-vex_ast-0.2.2.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-vex_ast-0.2.2.dist-info/top_level.txt,sha256=MoZGrpKgNUDiqL9gWp4q3wMw3q93XPEEjmBNPJQcNAs,8
-vex_ast-0.2.2.dist-info/RECORD,,
+vex_ast-0.2.3.dist-info/licenses/LICENSE,sha256=IOSlfCuxGv4OAg421BRDKVi16RZ7-5kCMJ4B16r4kuc,69
+vex_ast-0.2.3.dist-info/METADATA,sha256=2TsjEVVuPN94XQ8QOW5bHItBbFh3IAvn1uuCoEJe8nQ,5295
+vex_ast-0.2.3.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+vex_ast-0.2.3.dist-info/top_level.txt,sha256=MoZGrpKgNUDiqL9gWp4q3wMw3q93XPEEjmBNPJQcNAs,8
+vex_ast-0.2.3.dist-info/RECORD,,