vex-ast 0.2.1__tar.gz → 0.2.2__tar.gz

{vex_ast-0.2.1/vex_ast.egg-info → vex_ast-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vex_ast
-Version: 0.2.1
+Version: 0.2.2
 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.1 → vex_ast-0.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "vex_ast"
-version = "0.2.1"
+version = "0.2.2"
 description = "A Python package for generating Abstract Syntax Trees for VEX V5 code."
 readme = "README.md"
 authors = [

{vex_ast-0.2.1 → vex_ast-0.2.2}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='vex_ast',
-    version='0.2.1',
+    version='0.2.2',
     description='A Python package for generating Abstract Syntax Trees for VEX V5 code.',
     long_description=open('README.md').read(),
     long_description_content_type='text/markdown',

{vex_ast-0.2.1 → vex_ast-0.2.2}/vex_ast/__init__.py

@@ -30,7 +30,11 @@ from .serialization.schema import generate_ast_schema, export_schema_to_file
 __version__ = "0.2.0"
 
 # Initialize the registry with default functions
-initialize()
+try:
+    initialize()
+    print("VEX function registry initialized successfully")
+except Exception as e:
+    print(f"Error initializing VEX function registry: {e}")
 
 __all__ = [
     # Core functionality
@@ -65,4 +69,4 @@ __all__ = [
     "deserialize_ast_from_json",
     "generate_ast_schema",
     "export_schema_to_file"
-]
+]

{vex_ast-0.2.1 → vex_ast-0.2.2}/vex_ast/ast/validators.py

@@ -81,11 +81,17 @@ class VexFunctionValidator(AstVisitor[List[Tuple[VexAPICall, str]]]):
                 # Check if this is a method call on a known object type
                 if '.' in function_name:
                     obj_name, method_name = function_name.split('.', 1)
-                    # For method calls, we need to check if the method exists for any object type
-                    # since we don't know the actual type of the object at validation time
-                    # Just check if the method name exists in the registry API
+                    
+                    # First check if the method exists in the registry
                     if registry_api.get_function(method_name):
                         is_vex_function = True
+                    else:
+                        # Try to check if it's a method on any known object type
+                        from ..types.objects import MOTOR, TIMER, BRAIN, CONTROLLER
+                        for obj_type in [MOTOR, TIMER, BRAIN, CONTROLLER]:
+                            if registry_api.get_method(obj_type, method_name):
+                                is_vex_function = True
+                                break
                 # Or check if it's a direct function
                 elif registry_api.get_function(function_name):
                     is_vex_function = True

{vex_ast-0.2.1 → vex_ast-0.2.2}/vex_ast/ast/vex_nodes.py

@@ -59,7 +59,20 @@ class VexAPICall(FunctionCall):
         if '.' in function_name:
             # For method calls like "motor1.spin", extract the method name
             obj_name, method_name = function_name.split('.', 1)
+            
+            # First try to get the method signature directly
             self._signature = registry_api.get_function(method_name)
+            
+            # If that fails, try to get it as a method of a specific object type
+            # This is a fallback since we don't know the actual type at parse time
+            if not self._signature:
+                # Try common object types
+                from ..types.objects import MOTOR, TIMER, BRAIN, CONTROLLER
+                for obj_type in [MOTOR, TIMER, BRAIN, CONTROLLER]:
+                    method_sig = registry_api.get_method(obj_type, method_name)
+                    if method_sig:
+                        self._signature = method_sig
+                        break
         else:
             # For direct function calls
             self._signature = registry_api.get_function(function_name)
@@ -205,7 +218,19 @@ def create_vex_api_call(function: IExpression, args: List[IExpression],
     if '.' in function_name:
         # For method calls like "motor1.spin", extract the method name
         obj_name, method_name = function_name.split('.', 1)
+        
+        # First try to get the method signature directly
         signature = registry_api.get_function(method_name)
+        
+        # If that fails, try to get it as a method of a specific object type
+        if not signature:
+            # Try common object types
+            from ..types.objects import MOTOR, TIMER, BRAIN, CONTROLLER
+            for obj_type in [MOTOR, TIMER, BRAIN, CONTROLLER]:
+                method_sig = registry_api.get_method(obj_type, method_name)
+                if method_sig:
+                    signature = method_sig
+                    break
     else:
         # For direct function calls
         signature = registry_api.get_function(function_name)

{vex_ast-0.2.1 → vex_ast-0.2.2/vex_ast.egg-info}/PKG-INFO

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