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

functions.
This commit is contained in:
Fred Drake 2000-09-12 20:47:29 +00:00
parent df13b9ff9d
commit 0b25053ca0

View file

@ -10,7 +10,7 @@ import sys
import refcounts import refcounts
PREFIX = r"\begin{cfuncdesc}{PyObject*}{" PREFIX = r"\begin{cfuncdesc}{Py(Var|)Object*}{"
def main(): def main():
@ -48,14 +48,15 @@ def main():
except KeyError: except KeyError:
sys.stderr.write("No refcount data for %s\n" % s) sys.stderr.write("No refcount data for %s\n" % s)
else: else:
if info.result_type == "PyObject*": if info.result_type in ("PyObject*", "PyVarObject*"):
if info.result_refs is None: if info.result_refs is None:
rc = "Always \NULL{}" rc = "Always \NULL{}"
else: else:
rc = info.result_refs and "New" or "Borrowed" rc = info.result_refs and "New" or "Borrowed"
rc = rc + " reference" rc = rc + " reference"
line = r"\begin{cfuncdesc}[%s]{PyObject*}{" % rc \ line = (r"\begin{cfuncdesc}[%s]{%s}{"
+ line[prefix_len:] % (rc, info.result_type)) \
+ line[prefix_len:]
output.write(line) output.write(line)
if infile != "-": if infile != "-":
input.close() input.close()