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

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/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/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.dist-info → vex_ast-0.2.2.dist-info}/METADATA

@@ -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.dist-info → vex_ast-0.2.2.dist-info}/RECORD

@@ -1,6 +1,6 @@
 vex_ast/README.md,sha256=tRaq6n2Ab4IahHSo3utW2imVuTrJu_7zaelgF-lGBYo,2007
 vex_ast/READMEAPI.md,sha256=hGn4IMT9NnI_LW0kd36BdKKXdIhYFLbQcgbVj1EWJps,9107
-vex_ast/__init__.py,sha256=ctZTmMc0ispErWMja0pTwdyst6ELgcfBJ-nwejYg-sc,1604
+vex_ast/__init__.py,sha256=0xxA3UL3UMPB1Vlf1jL99bNUnvm41yJEoG5sMeylptY,1757
 vex_ast/ast/README.md,sha256=7IQDHEXmKyJ8zfJNwa3pMGTDZKVPMFD0YeHT06ATRyM,3205
 vex_ast/ast/__init__.py,sha256=_KT8R_WG9_DLvPlDhVCNWz-yw0OtdoKxxWZYbtloCzU,2434
 vex_ast/ast/core.py,sha256=q-15q5VfGIVPAmFsSiWeFAZ55MPsXfCRa21rrt2hmlM,2670
@@ -10,8 +10,8 @@ vex_ast/ast/literals.py,sha256=PXbOH5Y2fxEngWk_FOiFsx3PC2SEqcx0bcfOGJ3SdbI,2618
 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=lhEm1dt-nzMiaUan_AWEOm0NiwXN7wYNkLCl1c3drqc,4455
-vex_ast/ast/vex_nodes.py,sha256=kVil-ApFIeZ_BoeqYppfVRPSTg1KFUOcvt1vCE6QRgs,10148
+vex_ast/ast/validators.py,sha256=ySpi5H0XRdGXXMFAfmuVEyc3Q9mMXhubZAkP4N8NL3c,4717
+vex_ast/ast/vex_nodes.py,sha256=HmUPfiogz0rnObyhg1XEsxPIDdM5xbNxl_kW9NND8pQ,11345
 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
@@ -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.1.dist-info/licenses/LICENSE,sha256=IOSlfCuxGv4OAg421BRDKVi16RZ7-5kCMJ4B16r4kuc,69
-vex_ast-0.2.1.dist-info/METADATA,sha256=Cm7vpEPf4qNlTPKTYcO4_HejvWBeyH-kcPpIq6dRe5I,5295
-vex_ast-0.2.1.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-vex_ast-0.2.1.dist-info/top_level.txt,sha256=MoZGrpKgNUDiqL9gWp4q3wMw3q93XPEEjmBNPJQcNAs,8
-vex_ast-0.2.1.dist-info/RECORD,,
+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,,