Add a function to return just the line number of a code object.

diff --git a/Lib/codehack.py b/Lib/codehack.py
index 8a4f611696ded8280e5a189ac5b96fba43b89cc9..d00d2bfec9ce2417bd9e42279c4e0ec830f88663 100644 (file)
--- a/Lib/codehack.py
+++ b/Lib/codehack.py
@@ -52,3 +52,12 @@ def getcodename(co):
 
 def getfuncname(func):
        return getcodename(func.func_code)
+
+# A part of the above code to extract just the line number from a code object.
+
+def getlineno(co):
+       code = co.co_code
+       if ord(code[0]) == SET_LINENO:
+               return ord(code[1]) | ord(code[2]) << 8
+       else:
+               return -1

diff --git a/Lib/lib-old/codehack.py b/Lib/lib-old/codehack.py
index 8a4f611696ded8280e5a189ac5b96fba43b89cc9..d00d2bfec9ce2417bd9e42279c4e0ec830f88663 100644 (file)
--- a/Lib/lib-old/codehack.py
+++ b/Lib/lib-old/codehack.py
@@ -52,3 +52,12 @@ def getcodename(co):
 
 def getfuncname(func):
        return getcodename(func.func_code)
+
+# A part of the above code to extract just the line number from a code object.
+
+def getlineno(co):
+       code = co.co_code
+       if ord(code[0]) == SET_LINENO:
+               return ord(code[1]) | ord(code[2]) << 8
+       else:
+               return -1