symbex 0.3.1__tar.gz → 0.3.2__tar.gz

{symbex-0.3.1/symbex.egg-info → symbex-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: symbex
-Version: 0.3.1
+Version: 0.3.2
 Summary: Find the Python code for specified symbols
 Home-page: https://github.com/simonw/symbex
 Author: Simon Willison
@@ -132,7 +132,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.1 → symbex-0.3.2}/README.md

@@ -117,7 +117,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.1 → symbex-0.3.2}/setup.py

@@ -1,7 +1,7 @@
 from setuptools import setup
 import os
 
-VERSION = "0.3.1"
+VERSION = "0.3.2"
 
 
 def get_long_description():

{symbex-0.3.1 → symbex-0.3.2}/symbex/lib.py

@@ -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 = []
 

{symbex-0.3.1 → symbex-0.3.2/symbex.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: symbex
-Version: 0.3.1
+Version: 0.3.2
 Summary: Find the Python code for specified symbols
 Home-page: https://github.com/simonw/symbex
 Author: Simon Willison
@@ -132,7 +132,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.1 → symbex-0.3.2}/tests/test_symbex.py

@@ -11,7 +11,7 @@ def directory_full_of_code(tmpdir):
     for path, content in (
         ("foo.py", "def foo1():\n    pass\n\n@decorated\ndef foo2():\n    pass\n\n"),
         ("bar.py", "class BarClass:\n    pass\n\n"),
-        ("nested/baz.py", "def baz():\n    pass\n\n"),
+        ("nested/baz.py", 'def baz(delimiter=", ", type=str):\n    pass\n\n'),
         ("nested/error.py", "def baz_error()" + "bug:\n    pass\n\n"),
         (
             "methods.py",
@@ -60,7 +60,7 @@ def directory_full_of_code(tmpdir):
         ),
         (
             ["baz", "--silent"],
-            "# File: nested/baz.py Line: 1\ndef baz():\n    pass\n\n",
+            '# File: nested/baz.py Line: 1\ndef baz(delimiter=", ", type=str):\n    pass\n\n',
         ),
         (
             ["async_func", "--silent"],
@@ -69,12 +69,12 @@ def directory_full_of_code(tmpdir):
         # The -f option
         (
             ["baz", "-f", "nested/baz.py", "--silent"],
-            "# File: nested/baz.py Line: 1\ndef baz():\n    pass\n\n",
+            '# File: nested/baz.py Line: 1\ndef baz(delimiter=", ", type=str):\n    pass\n\n',
         ),
         # The -d option
         (
             ["baz", "-d", "nested", "--silent"],
-            "# File: nested/baz.py Line: 1\ndef baz():\n    pass\n\n",
+            '# File: nested/baz.py Line: 1\ndef baz(delimiter=", ", type=str):\n    pass\n\n',
         ),
         # Classes
         (
@@ -144,7 +144,10 @@ def test_fixture(directory_full_of_code, monkeypatch, args, expected):
             "def foo2()",
         ),
         (["BarClass", "--silent"], "# File: bar.py Line: 1\n" "class BarClass"),
-        (["baz", "--silent"], ("# File: nested/baz.py Line: 1\n" "def baz()")),
+        (
+            ["baz", "--silent"],
+            ("# File: nested/baz.py Line: 1\n" 'def baz(delimiter=", ", type=str)'),
+        ),
     ),
 )
 def test_symbex_symbols(directory_full_of_code, monkeypatch, args, expected):
@@ -162,8 +165,11 @@ def test_errors(directory_full_of_code, monkeypatch):
     monkeypatch.chdir(directory_full_of_code)
     result = runner.invoke(cli, ["baz"], catch_exceptions=False)
     assert result.exit_code == 0
-    assert result.stdout == (
-        "# File: nested/baz.py Line: 1\n" "def baz():\n" "    pass\n\n"
+    expected = (
+        "# File: nested/baz.py Line: 1\n"
+        'def baz(delimiter=", ", type=str):\n'
+        "    pass\n\n"
     )
+    assert result.stdout == expected
     # This differs between different Python versions
     assert result.stderr.startswith("# Syntax error in nested/error.py:")