symbex 0.3__py3-none-any.whl → 0.3.2__py3-none-any.whl

symbex/lib.py

@@ -1,5 +1,5 @@
 import fnmatch
-from ast import literal_eval, parse, AST, FunctionDef, ClassDef
+from ast import literal_eval, parse, AST, AsyncFunctionDef, FunctionDef, ClassDef
 from itertools import zip_longest
 import textwrap
 from typing import Iterable, List, Optional, Tuple
@@ -13,7 +13,7 @@ def find_symbol_nodes(
     matches = []
     module = parse(code)
     for node in module.body:
-        if not isinstance(node, (ClassDef, FunctionDef)):
+        if not isinstance(node, (ClassDef, FunctionDef, AsyncFunctionDef)):
             continue
         name = getattr(node, "name", None)
         if match(name, symbols):
@@ -21,7 +21,7 @@ def find_symbol_nodes(
         # If it's a class search its methods too
         if isinstance(node, ClassDef):
             for child in node.body:
-                if isinstance(child, FunctionDef):
+                if isinstance(child, (FunctionDef, AsyncFunctionDef)):
                     qualified_name = f"{name}.{child.name}"
                     if match(qualified_name, symbols):
                         matches.append((child, name))
@@ -37,7 +37,7 @@ def code_for_node(
     start = None
     end = None
     if signatures:
-        if isinstance(node, FunctionDef):
+        if isinstance(node, (FunctionDef, AsyncFunctionDef)):
             definition, lineno = function_definition(node), node.lineno
             if class_name:
                 definition = "    " + definition
@@ -109,7 +109,14 @@ def function_definition(function_node: AST):
     # if defaults has 2 and args has 3 then those
     # defaults correspond to the last two args
     defaults = [None] * (len(all_args) - len(function_node.args.defaults))
-    defaults.extend(literal_eval(default) for default in function_node.args.defaults)
+    for default in function_node.args.defaults:
+        try:
+            value = literal_eval(default)
+            if isinstance(value, str):
+                value = f'"{value}"'
+        except ValueError:
+            value = getattr(default, "id", "...")
+        defaults.append(value)
 
     arguments = []
 
@@ -144,7 +151,11 @@ def function_definition(function_node: AST):
             # None shows as returns.value is None
             return_annotation = " -> None"
 
-    return f"def {function_name}({arguments_str}){return_annotation}"
+    def_ = "def "
+    if isinstance(function_node, AsyncFunctionDef):
+        def_ = "async def "
+
+    return f"{def_}{function_name}({arguments_str}){return_annotation}"
 
 
 def class_definition(class_def):

{symbex-0.3.dist-info → symbex-0.3.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: symbex
-Version: 0.3
+Version: 0.3.2
 Summary: Find the Python code for specified symbols
 Home-page: https://github.com/simonw/symbex
 Author: Simon Willison
@@ -136,7 +136,7 @@ cog.out(
 # File: symbex/cli.py Line: 37
 def cli(symbols, files, directories, signatures, silent)
 
-# File: symbex/lib.py Line: 150
+# File: symbex/lib.py Line: 161
 def class_definition(class_def)
 
 # File: symbex/lib.py Line: 32

symbex-0.3.2.dist-info/RECORD

@@ -0,0 +1,10 @@
+symbex/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+symbex/__main__.py,sha256=8hDtWlaFZK24KhfNq_ZKgtXqYHsDQDetukOCMlsbW0Q,59
+symbex/cli.py,sha256=tWzT49q-fPN34HQH2sEFHBV3MXFKy9J5eh2EzOo5_CA,2857
+symbex/lib.py,sha256=_IwiAfr9rCAe4ptdNt6o6MMibLHnwBmqpmgYRdRmEa4,6499
+symbex-0.3.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+symbex-0.3.2.dist-info/METADATA,sha256=xlQ1koJSuZ6luJeC1WtNrU-D5g-dz82Br1NPYZ3yUXc,5695
+symbex-0.3.2.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
+symbex-0.3.2.dist-info/entry_points.txt,sha256=YgMSEfEGqNMHM9RysFObH8lkQKVZKyymKLnXbVue_Uk,42
+symbex-0.3.2.dist-info/top_level.txt,sha256=qwle8HjAaYgpdMIHlJcTcN4gaG4wmDqUvkt54beTBTs,7
+symbex-0.3.2.dist-info/RECORD,,

symbex-0.3.dist-info/RECORD

@@ -1,10 +0,0 @@
-symbex/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-symbex/__main__.py,sha256=8hDtWlaFZK24KhfNq_ZKgtXqYHsDQDetukOCMlsbW0Q,59
-symbex/cli.py,sha256=tWzT49q-fPN34HQH2sEFHBV3MXFKy9J5eh2EzOo5_CA,2857
-symbex/lib.py,sha256=q8WhjbPLiXuGnbD3scwCvJrqVvs-RHAMbBEXIYyFqeI,6121
-symbex-0.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-symbex-0.3.dist-info/METADATA,sha256=Yn6vp7j1nWfTn_GYWXmeWhD7NFXHw2OxI3YnpMRVvew,5693
-symbex-0.3.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-symbex-0.3.dist-info/entry_points.txt,sha256=YgMSEfEGqNMHM9RysFObH8lkQKVZKyymKLnXbVue_Uk,42
-symbex-0.3.dist-info/top_level.txt,sha256=qwle8HjAaYgpdMIHlJcTcN4gaG4wmDqUvkt54beTBTs,7
-symbex-0.3.dist-info/RECORD,,