Improve default value for None in text_signature

PyO3 · Mar 26, 2023 · a4a58f5 · a4a58f5
1 parent ebedcfb
commit a4a58f5
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 18 deletions.
diff --git a/newsfragments/3066.changed.md b/newsfragments/3066.changed.md
@@ -0,0 +1 @@
+Improve default value for `None` in `text_signature`.
diff --git a/pyo3-macros-backend/src/pyfunction/signature.rs b/pyo3-macros-backend/src/pyfunction/signature.rs
@@ -588,19 +588,30 @@ impl<'a> FunctionSignature<'a> {
     fn default_value_for_parameter(&self, parameter: &str) -> String {
         let mut default = "...".to_string();
         if let Some(fn_arg) = self.arguments.iter().find(|arg| arg.name == parameter) {
-            if let Some(syn::Expr::Lit(syn::ExprLit { lit, .. })) = fn_arg.default.as_ref() {
-                match lit {
-                    syn::Lit::Str(s) => default = s.token().to_string(),
-                    syn::Lit::Char(c) => default = c.token().to_string(),
-                    syn::Lit::Int(i) => default = i.base10_digits().to_string(),
-                    syn::Lit::Float(f) => default = f.base10_digits().to_string(),
-                    syn::Lit::Bool(b) => {
-                        default = if b.value() {
-                            "True".to_string()
-                        } else {
-                            "False".to_string()
+            if let Some(arg_default) = fn_arg.default.as_ref() {
+                match arg_default {
+                    // literal values
+                    syn::Expr::Lit(syn::ExprLit { lit, .. }) => match lit {
+                        syn::Lit::Str(s) => default = s.token().to_string(),
+                        syn::Lit::Char(c) => default = c.token().to_string(),
+                        syn::Lit::Int(i) => default = i.base10_digits().to_string(),
+                        syn::Lit::Float(f) => default = f.base10_digits().to_string(),
+                        syn::Lit::Bool(b) => {
+                            default = if b.value() {
+                                "True".to_string()
+                            } else {
+                                "False".to_string()
+                            }
                         }
+                        _ => {}
+                    },
+                    // None
+                    syn::Expr::Path(syn::ExprPath {
+                        qself: None, path, ..
+                    }) if path.is_ident("None") => {
+                        default = "None".to_string();
                     }
+                    // others, unsupported yet so defaults to `...`
                     _ => {}
                 }
             }

diff --git a/tests/test_text_signature.rs b/tests/test_text_signature.rs
@@ -154,26 +154,38 @@ fn test_auto_test_signature_function() {
 
     Python::with_gil(|py| {
         let f = wrap_pyfunction!(my_function)(py).unwrap();
-        py_assert!(py, f, "f.__text_signature__ == '(a, b, c)'");
+        py_assert!(
+            py,
+            f,
+            "f.__text_signature__ == '(a, b, c)', f.__text_signature__"
+        );
 
         let f = wrap_pyfunction!(my_function_2)(py).unwrap();
-        py_assert!(py, f, "f.__text_signature__ == '($module, a, b, c)'");
+        py_assert!(
+            py,
+            f,
+            "f.__text_signature__ == '($module, a, b, c)', f.__text_signature__"
+        );
 
         let f = wrap_pyfunction!(my_function_3)(py).unwrap();
-        py_assert!(py, f, "f.__text_signature__ == '(a, /, b=..., *, c=5)'");
+        py_assert!(
+            py,
+            f,
+            "f.__text_signature__ == '(a, /, b=None, *, c=5)', f.__text_signature__"
+        );
 
         let f = wrap_pyfunction!(my_function_4)(py).unwrap();
         py_assert!(
             py,
             f,
-            "f.__text_signature__ == '(a, /, b=..., *args, c, d=5, **kwargs)'"
+            "f.__text_signature__ == '(a, /, b=None, *args, c, d=5, **kwargs)', f.__text_signature__"
         );
 
         let f = wrap_pyfunction!(my_function_5)(py).unwrap();
         py_assert!(
             py,
             f,
-            "f.__text_signature__ == '(a=1, /, b=..., c=1.5, d=5, e=\"pyo3\", f=\\'f\\', h=True)', f.__text_signature__"
+            "f.__text_signature__ == '(a=1, /, b=None, c=1.5, d=5, e=\"pyo3\", f=\\'f\\', h=True)', f.__text_signature__"
         );
     });
 }
@@ -228,12 +240,12 @@ fn test_auto_test_signature_method() {
         py_assert!(
             py,
             cls,
-            "cls.method_2.__text_signature__ == '($self, a, /, b=..., *, c=5)'"
+            "cls.method_2.__text_signature__ == '($self, a, /, b=None, *, c=5)'"
         );
         py_assert!(
             py,
             cls,
-            "cls.method_3.__text_signature__ == '($self, a, /, b=..., *args, c, d=5, **kwargs)'"
+            "cls.method_3.__text_signature__ == '($self, a, /, b=None, *args, c, d=5, **kwargs)'"
         );
         py_assert!(
             py,