Make this script handle PyVarObject* functions as well as PyObject*

freddrake · freddrake · commit 0b25053ca0c6 · 2000-09-12T20:47:29.000Z
functions.
diff --git a/Doc/tools/anno-api.py b/Doc/tools/anno-api.py
@@ -10,7 +10,7 @@
 import refcounts
 
 
-PREFIX = r"\begin{cfuncdesc}{PyObject*}{"
+PREFIX = r"\begin{cfuncdesc}{Py(Var|)Object*}{"
 
 
 def main():
@@ -48,14 +48,15 @@ def main():
                 except KeyError:
                     sys.stderr.write("No refcount data for %s\n" % s)
                 else:
-                    if info.result_type == "PyObject*":
+                    if info.result_type in ("PyObject*", "PyVarObject*"):
                         if info.result_refs is None:
                             rc = "Always \NULL{}"
                         else:
                             rc = info.result_refs and "New" or "Borrowed"
                             rc = rc + " reference"
-                        line = r"\begin{cfuncdesc}[%s]{PyObject*}{" % rc \
-                               + line[prefix_len:]
+                        line = (r"\begin{cfuncdesc}[%s]{%s}{"
+                                % (rc, info.result_type)) \
+                                + line[prefix_len:]
             output.write(line)
         if infile != "-":
             input.close()
