1 # -*- coding: iso-8859-1 -*-
2 """Get useful information from live Python objects.
4 This module encapsulates the interface provided by the internal special
5 attributes (func_*, co_*, im_*, tb_*, etc.) in a friendlier fashion.
6 It also provides some help for examining source code and class layout.
8 Here are some of the useful functions provided by this module:
10 ismodule(), isclass(), ismethod(), isfunction(), istraceback(),
11 isframe(), iscode(), isbuiltin(), isroutine() - check object types
12 getmembers() - get members of an object that satisfy a given condition
14 getfile(), getsourcefile(), getsource() - find an object's source code
15 getdoc(), getcomments() - get documentation on an object
16 getmodule() - determine the module that an object came from
17 getclasstree() - arrange classes so as to represent their hierarchy
19 getargspec(), getargvalues() - get info about function arguments
20 formatargspec(), formatargvalues() - format an argument spec
21 getouterframes(), getinnerframes() - get info about frames
22 currentframe() - get the current stack frame
23 stack(), trace() - get info about frames on the stack or in a traceback
26 # This module is in the public domain. No warranties.
28 __author__
= 'Ka-Ping Yee <ping@lfw.org>'
29 __date__
= '1 Jan 2001'
31 import sys
, os
, types
, string
, re
, dis
, imp
, tokenize
, linecache
32 from operator
import attrgetter
34 # ----------------------------------------------------------- type-checking
36 """Return true if the object is a module.
38 Module objects provide these attributes:
39 __doc__ documentation string
40 __file__ filename (missing for built-in modules)"""
41 return isinstance(object, types
.ModuleType
)
44 """Return true if the object is a class.
46 Class objects provide these attributes:
47 __doc__ documentation string
48 __module__ name of module in which this class was defined"""
49 return isinstance(object, types
.ClassType
) or hasattr(object, '__bases__')
52 """Return true if the object is an instance method.
54 Instance method objects provide these attributes:
55 __doc__ documentation string
56 __name__ name with which this method was defined
57 im_class class object in which this method belongs
58 im_func function object containing implementation of method
59 im_self instance to which this method is bound, or None"""
60 return isinstance(object, types
.MethodType
)
62 def ismethoddescriptor(object):
63 """Return true if the object is a method descriptor.
65 But not if ismethod() or isclass() or isfunction() are true.
67 This is new in Python 2.2, and, for example, is true of int.__add__.
68 An object passing this test has a __get__ attribute but not a __set__
69 attribute, but beyond that the set of attributes varies. __name__ is
70 usually sensible, and __doc__ often is.
72 Methods implemented via descriptors that also pass one of the other
73 tests return false from the ismethoddescriptor() test, simply because
74 the other tests promise more -- you can, e.g., count on having the
75 im_func attribute (etc) when an object passes ismethod()."""
76 return (hasattr(object, "__get__")
77 and not hasattr(object, "__set__") # else it's a data descriptor
78 and not ismethod(object) # mutual exclusion
79 and not isfunction(object)
80 and not isclass(object))
82 def isdatadescriptor(object):
83 """Return true if the object is a data descriptor.
85 Data descriptors have both a __get__ and a __set__ attribute. Examples are
86 properties (defined in Python) and getsets and members (defined in C).
87 Typically, data descriptors will also have __name__ and __doc__ attributes
88 (properties, getsets, and members have both of these attributes), but this
90 return (hasattr(object, "__set__") and hasattr(object, "__get__"))
92 def isfunction(object):
93 """Return true if the object is a user-defined function.
95 Function objects provide these attributes:
96 __doc__ documentation string
97 __name__ name with which this function was defined
98 func_code code object containing compiled function bytecode
99 func_defaults tuple of any default values for arguments
100 func_doc (same as __doc__)
101 func_globals global namespace in which this function was defined
102 func_name (same as __name__)"""
103 return isinstance(object, types
.FunctionType
)
105 def istraceback(object):
106 """Return true if the object is a traceback.
108 Traceback objects provide these attributes:
109 tb_frame frame object at this level
110 tb_lasti index of last attempted instruction in bytecode
111 tb_lineno current line number in Python source code
112 tb_next next inner traceback object (called by this level)"""
113 return isinstance(object, types
.TracebackType
)
116 """Return true if the object is a frame object.
118 Frame objects provide these attributes:
119 f_back next outer frame object (this frame's caller)
120 f_builtins built-in namespace seen by this frame
121 f_code code object being executed in this frame
122 f_exc_traceback traceback if raised in this frame, or None
123 f_exc_type exception type if raised in this frame, or None
124 f_exc_value exception value if raised in this frame, or None
125 f_globals global namespace seen by this frame
126 f_lasti index of last attempted instruction in bytecode
127 f_lineno current line number in Python source code
128 f_locals local namespace seen by this frame
129 f_restricted 0 or 1 if frame is in restricted execution mode
130 f_trace tracing function for this frame, or None"""
131 return isinstance(object, types
.FrameType
)
134 """Return true if the object is a code object.
136 Code objects provide these attributes:
137 co_argcount number of arguments (not including * or ** args)
138 co_code string of raw compiled bytecode
139 co_consts tuple of constants used in the bytecode
140 co_filename name of file in which this code object was created
141 co_firstlineno number of first line in Python source code
142 co_flags bitmap: 1=optimized | 2=newlocals | 4=*arg | 8=**arg
143 co_lnotab encoded mapping of line numbers to bytecode indices
144 co_name name with which this code object was defined
145 co_names tuple of names of local variables
146 co_nlocals number of local variables
147 co_stacksize virtual machine stack space required
148 co_varnames tuple of names of arguments and local variables"""
149 return isinstance(object, types
.CodeType
)
151 def isbuiltin(object):
152 """Return true if the object is a built-in function or method.
154 Built-in functions and methods provide these attributes:
155 __doc__ documentation string
156 __name__ original name of this function or method
157 __self__ instance to which a method is bound, or None"""
158 return isinstance(object, types
.BuiltinFunctionType
)
160 def isroutine(object):
161 """Return true if the object is any kind of function or method."""
162 return (isbuiltin(object)
163 or isfunction(object)
165 or ismethoddescriptor(object))
167 def getmembers(object, predicate
=None):
168 """Return all members of an object as (name, value) pairs sorted by name.
169 Optionally, only return members that satisfy a given predicate."""
171 for key
in dir(object):
172 value
= getattr(object, key
)
173 if not predicate
or predicate(value
):
174 results
.append((key
, value
))
178 def classify_class_attrs(cls
):
179 """Return list of attribute-descriptor tuples.
181 For each name in dir(cls), the return list contains a 4-tuple
184 0. The name (a string).
186 1. The kind of attribute this is, one of these strings:
187 'class method' created via classmethod()
188 'static method' created via staticmethod()
189 'property' created via property()
190 'method' any other flavor of method
193 2. The class which defined this attribute (a class).
195 3. The object as obtained directly from the defining class's
196 __dict__, not via getattr. This is especially important for
197 data attributes: C.data is just a data object, but
198 C.__dict__['data'] may be a data descriptor with additional
199 info, like a __doc__ string.
206 # Get the object associated with the name.
207 # Getting an obj from the __dict__ sometimes reveals more than
208 # using getattr. Static and class methods are dramatic examples.
209 if name
in cls
.__dict
__:
210 obj
= cls
.__dict
__[name
]
212 obj
= getattr(cls
, name
)
214 # Figure out where it was defined.
215 homecls
= getattr(obj
, "__objclass__", None)
219 if name
in base
.__dict
__:
223 # Get the object again, in order to get it from the defining
224 # __dict__ instead of via getattr (if possible).
225 if homecls
is not None and name
in homecls
.__dict
__:
226 obj
= homecls
.__dict
__[name
]
228 # Also get the object via getattr.
229 obj_via_getattr
= getattr(cls
, name
)
231 # Classify the object.
232 if isinstance(obj
, staticmethod):
233 kind
= "static method"
234 elif isinstance(obj
, classmethod):
235 kind
= "class method"
236 elif isinstance(obj
, property):
238 elif (ismethod(obj_via_getattr
) or
239 ismethoddescriptor(obj_via_getattr
)):
244 result
.append((name
, kind
, homecls
, obj
))
248 # ----------------------------------------------------------- class helpers
249 def _searchbases(cls
, accum
):
250 # Simulate the "classic class" search order.
254 for base
in cls
.__bases
__:
255 _searchbases(base
, accum
)
258 "Return tuple of base classes (including cls) in method resolution order."
259 if hasattr(cls
, "__mro__"):
263 _searchbases(cls
, result
)
266 # -------------------------------------------------- source code extraction
267 def indentsize(line
):
268 """Return the indent size, in spaces, at the start of a line of text."""
269 expline
= string
.expandtabs(line
)
270 return len(expline
) - len(string
.lstrip(expline
))
273 """Get the documentation string for an object.
275 All tabs are expanded to spaces. To clean up docstrings that are
276 indented to line up with blocks of code, any whitespace than can be
277 uniformly removed from the second line onwards is removed."""
280 except AttributeError:
282 if not isinstance(doc
, types
.StringTypes
):
285 lines
= string
.split(string
.expandtabs(doc
), '\n')
289 # Find minimum indentation of any non-blank lines after first line.
291 for line
in lines
[1:]:
292 content
= len(string
.lstrip(line
))
294 indent
= len(line
) - content
295 margin
= min(margin
, indent
)
296 # Remove indentation.
298 lines
[0] = lines
[0].lstrip()
299 if margin
< sys
.maxint
:
300 for i
in range(1, len(lines
)): lines
[i
] = lines
[i
][margin
:]
301 # Remove any trailing or leading blank lines.
302 while lines
and not lines
[-1]:
304 while lines
and not lines
[0]:
306 return string
.join(lines
, '\n')
309 """Work out which source or compiled file an object was defined in."""
311 if hasattr(object, '__file__'):
312 return object.__file
__
313 raise TypeError('arg is a built-in module')
315 object = sys
.modules
.get(object.__module
__)
316 if hasattr(object, '__file__'):
317 return object.__file
__
318 raise TypeError('arg is a built-in class')
320 object = object.im_func
321 if isfunction(object):
322 object = object.func_code
323 if istraceback(object):
324 object = object.tb_frame
326 object = object.f_code
328 return object.co_filename
329 raise TypeError('arg is not a module, class, method, '
330 'function, traceback, frame, or code object')
332 def getmoduleinfo(path
):
333 """Get the module name, suffix, mode, and module type for a given file."""
334 filename
= os
.path
.basename(path
)
335 suffixes
= map(lambda (suffix
, mode
, mtype
):
336 (-len(suffix
), suffix
, mode
, mtype
), imp
.get_suffixes())
337 suffixes
.sort() # try longest suffixes first, in case they overlap
338 for neglen
, suffix
, mode
, mtype
in suffixes
:
339 if filename
[neglen
:] == suffix
:
340 return filename
[:neglen
], suffix
, mode
, mtype
342 def getmodulename(path
):
343 """Return the module name for a given file, or None."""
344 info
= getmoduleinfo(path
)
345 if info
: return info
[0]
347 def getsourcefile(object):
348 """Return the Python source file an object was defined in, if it exists."""
349 filename
= getfile(object)
350 if string
.lower(filename
[-4:]) in ('.pyc', '.pyo'):
351 filename
= filename
[:-4] + '.py'
352 for suffix
, mode
, kind
in imp
.get_suffixes():
353 if 'b' in mode
and string
.lower(filename
[-len(suffix
):]) == suffix
:
354 # Looks like a binary file. We want to only return a text file.
356 if os
.path
.exists(filename
):
358 # Ugly but necessary - '<stdin>' and '<string>' mean that getmodule()
359 # would infinitely recurse, because they're not real files nor loadable
360 # Note that this means that writing a PEP 302 loader that uses '<'
361 # at the start of a filename is now not a good idea. :(
362 if filename
[:1]!='<' and hasattr(getmodule(object), '__loader__'):
365 def getabsfile(object):
366 """Return an absolute path to the source or compiled file for an object.
368 The idea is for each object to have a unique origin, so this routine
369 normalizes the result as much as possible."""
370 return os
.path
.normcase(
371 os
.path
.abspath(getsourcefile(object) or getfile(object)))
375 def getmodule(object):
376 """Return the module an object was defined in, or None if not found."""
379 if hasattr(object, '__module__'):
380 return sys
.modules
.get(object.__module
__)
382 file = getabsfile(object)
385 if file in modulesbyfile
:
386 return sys
.modules
.get(modulesbyfile
[file])
387 for module
in sys
.modules
.values():
388 if ismodule(module
) and hasattr(module
, '__file__'):
391 getabsfile(module
))] = module
.__name
__
392 if file in modulesbyfile
:
393 return sys
.modules
.get(modulesbyfile
[file])
394 main
= sys
.modules
['__main__']
395 if not hasattr(object, '__name__'):
397 if hasattr(main
, object.__name
__):
398 mainobject
= getattr(main
, object.__name
__)
399 if mainobject
is object:
401 builtin
= sys
.modules
['__builtin__']
402 if hasattr(builtin
, object.__name
__):
403 builtinobject
= getattr(builtin
, object.__name
__)
404 if builtinobject
is object:
407 def findsource(object):
408 """Return the entire source file and starting line number for an object.
410 The argument may be a module, class, method, function, traceback, frame,
411 or code object. The source code is returned as a list of all the lines
412 in the file and the line number indexes a line in that list. An IOError
413 is raised if the source code cannot be retrieved."""
414 file = getsourcefile(object) or getfile(object)
415 module
= getmodule(object)
417 lines
= linecache
.getlines(file, module
.__dict
__)
419 lines
= linecache
.getlines(file)
421 raise IOError('could not get source code')
427 name
= object.__name
__
428 pat
= re
.compile(r
'^\s*class\s*' + name
+ r
'\b')
429 for i
in range(len(lines
)):
430 if pat
.match(lines
[i
]): return lines
, i
432 raise IOError('could not find class definition')
435 object = object.im_func
436 if isfunction(object):
437 object = object.func_code
438 if istraceback(object):
439 object = object.tb_frame
441 object = object.f_code
443 if not hasattr(object, 'co_firstlineno'):
444 raise IOError('could not find function definition')
445 lnum
= object.co_firstlineno
- 1
446 pat
= re
.compile(r
'^(\s*def\s)|(.*(?<!\w)lambda(:|\s))|^(\s*@)')
448 if pat
.match(lines
[lnum
]): break
451 raise IOError('could not find code object')
453 def getcomments(object):
454 """Get lines of comments immediately preceding an object's source code.
456 Returns None when source can't be found.
459 lines
, lnum
= findsource(object)
460 except (IOError, TypeError):
464 # Look for a comment block at the top of the file.
466 if lines
and lines
[0][:2] == '#!': start
= 1
467 while start
< len(lines
) and string
.strip(lines
[start
]) in ('', '#'):
469 if start
< len(lines
) and lines
[start
][:1] == '#':
472 while end
< len(lines
) and lines
[end
][:1] == '#':
473 comments
.append(string
.expandtabs(lines
[end
]))
475 return string
.join(comments
, '')
477 # Look for a preceding block of comments at the same indentation.
479 indent
= indentsize(lines
[lnum
])
481 if end
>= 0 and string
.lstrip(lines
[end
])[:1] == '#' and \
482 indentsize(lines
[end
]) == indent
:
483 comments
= [string
.lstrip(string
.expandtabs(lines
[end
]))]
486 comment
= string
.lstrip(string
.expandtabs(lines
[end
]))
487 while comment
[:1] == '#' and indentsize(lines
[end
]) == indent
:
488 comments
[:0] = [comment
]
491 comment
= string
.lstrip(string
.expandtabs(lines
[end
]))
492 while comments
and string
.strip(comments
[0]) == '#':
494 while comments
and string
.strip(comments
[-1]) == '#':
496 return string
.join(comments
, '')
498 class EndOfBlock(Exception): pass
501 """Provide a tokeneater() method to detect the end of a code block."""
504 self
.islambda
= False
506 self
.passline
= False
509 def tokeneater(self
, type, token
, (srow
, scol
), (erow
, ecol
), line
):
511 # look for the first "def", "class" or "lambda"
512 if token
in ("def", "class", "lambda"):
513 if token
== "lambda":
516 self
.passline
= True # skip to the end of the line
517 elif type == tokenize
.NEWLINE
:
518 self
.passline
= False # stop skipping when a NEWLINE is seen
520 if self
.islambda
: # lambdas always end at the first NEWLINE
524 elif type == tokenize
.INDENT
:
525 self
.indent
= self
.indent
+ 1
527 elif type == tokenize
.DEDENT
:
528 self
.indent
= self
.indent
- 1
529 # the end of matching indent/dedent pairs end a block
530 # (note that this only works for "def"/"class" blocks,
531 # not e.g. for "if: else:" or "try: finally:" blocks)
534 elif self
.indent
== 0 and type not in (tokenize
.COMMENT
, tokenize
.NL
):
535 # any other token on the same indentation level end the previous
536 # block as well, except the pseudo-tokens COMMENT and NL.
540 """Extract the block of code at the top of the given list of lines."""
541 blockfinder
= BlockFinder()
543 tokenize
.tokenize(iter(lines
).next
, blockfinder
.tokeneater
)
544 except (EndOfBlock
, IndentationError):
546 return lines
[:blockfinder
.last
]
548 def getsourcelines(object):
549 """Return a list of source lines and starting line number for an object.
551 The argument may be a module, class, method, function, traceback, frame,
552 or code object. The source code is returned as a list of the lines
553 corresponding to the object and the line number indicates where in the
554 original source file the first line of code was found. An IOError is
555 raised if the source code cannot be retrieved."""
556 lines
, lnum
= findsource(object)
558 if ismodule(object): return lines
, 0
559 else: return getblock(lines
[lnum
:]), lnum
+ 1
561 def getsource(object):
562 """Return the text of the source code for an object.
564 The argument may be a module, class, method, function, traceback, frame,
565 or code object. The source code is returned as a single string. An
566 IOError is raised if the source code cannot be retrieved."""
567 lines
, lnum
= getsourcelines(object)
568 return string
.join(lines
, '')
570 # --------------------------------------------------- class tree extraction
571 def walktree(classes
, children
, parent
):
572 """Recursive helper function for getclasstree()."""
574 classes
.sort(key
=attrgetter('__module__', '__name__'))
576 results
.append((c
, c
.__bases
__))
578 results
.append(walktree(children
[c
], children
, c
))
581 def getclasstree(classes
, unique
=0):
582 """Arrange the given list of classes into a hierarchy of nested lists.
584 Where a nested list appears, it contains classes derived from the class
585 whose entry immediately precedes the list. Each entry is a 2-tuple
586 containing a class and a tuple of its base classes. If the 'unique'
587 argument is true, exactly one entry appears in the returned structure
588 for each class in the given list. Otherwise, classes using multiple
589 inheritance and their descendants will appear multiple times."""
594 for parent
in c
.__bases
__:
595 if not parent
in children
:
596 children
[parent
] = []
597 children
[parent
].append(c
)
598 if unique
and parent
in classes
: break
601 for parent
in children
:
602 if parent
not in classes
:
604 return walktree(roots
, children
, None)
606 # ------------------------------------------------ argument list extraction
607 # These constants are from Python's compile.h.
608 CO_OPTIMIZED
, CO_NEWLOCALS
, CO_VARARGS
, CO_VARKEYWORDS
= 1, 2, 4, 8
611 """Get information about the arguments accepted by a code object.
613 Three things are returned: (args, varargs, varkw), where 'args' is
614 a list of argument names (possibly containing nested lists), and
615 'varargs' and 'varkw' are the names of the * and ** arguments or None."""
618 raise TypeError('arg is not a code object')
621 nargs
= co
.co_argcount
622 names
= co
.co_varnames
623 args
= list(names
[:nargs
])
626 # The following acrobatics are for anonymous (tuple) arguments.
627 for i
in range(nargs
):
628 if args
[i
][:1] in ('', '.'):
629 stack
, remain
, count
= [], [], []
630 while step
< len(code
):
633 if op
>= dis
.HAVE_ARGUMENT
:
634 opname
= dis
.opname
[op
]
635 value
= ord(code
[step
]) + ord(code
[step
+1])*256
637 if opname
in ('UNPACK_TUPLE', 'UNPACK_SEQUENCE'):
640 elif opname
== 'STORE_FAST':
641 stack
.append(names
[value
])
643 # Special case for sublists of length 1: def foo((bar))
644 # doesn't generate the UNPACK_TUPLE bytecode, so if
645 # `remain` is empty here, we have such a sublist.
647 stack
[0] = [stack
[0]]
650 remain
[-1] = remain
[-1] - 1
651 while remain
[-1] == 0:
654 stack
[-size
:] = [stack
[-size
:]]
656 remain
[-1] = remain
[-1] - 1
661 if co
.co_flags
& CO_VARARGS
:
662 varargs
= co
.co_varnames
[nargs
]
665 if co
.co_flags
& CO_VARKEYWORDS
:
666 varkw
= co
.co_varnames
[nargs
]
667 return args
, varargs
, varkw
669 def getargspec(func
):
670 """Get the names and default values of a function's arguments.
672 A tuple of four things is returned: (args, varargs, varkw, defaults).
673 'args' is a list of the argument names (it may contain nested lists).
674 'varargs' and 'varkw' are the names of the * and ** arguments or None.
675 'defaults' is an n-tuple of the default values of the last n arguments.
680 if not isfunction(func
):
681 raise TypeError('arg is not a Python function')
682 args
, varargs
, varkw
= getargs(func
.func_code
)
683 return args
, varargs
, varkw
, func
.func_defaults
685 def getargvalues(frame
):
686 """Get information about arguments passed into a particular frame.
688 A tuple of four things is returned: (args, varargs, varkw, locals).
689 'args' is a list of the argument names (it may contain nested lists).
690 'varargs' and 'varkw' are the names of the * and ** arguments or None.
691 'locals' is the locals dictionary of the given frame."""
692 args
, varargs
, varkw
= getargs(frame
.f_code
)
693 return args
, varargs
, varkw
, frame
.f_locals
697 return '(' + seq
[0] + ',)'
699 return '(' + string
.join(seq
, ', ') + ')'
701 def strseq(object, convert
, join
=joinseq
):
702 """Recursively walk a sequence, stringifying each element."""
703 if type(object) in (list, tuple):
704 return join(map(lambda o
, c
=convert
, j
=join
: strseq(o
, c
, j
), object))
706 return convert(object)
708 def formatargspec(args
, varargs
=None, varkw
=None, defaults
=None,
710 formatvarargs
=lambda name
: '*' + name
,
711 formatvarkw
=lambda name
: '**' + name
,
712 formatvalue
=lambda value
: '=' + repr(value
),
714 """Format an argument spec from the 4 values returned by getargspec.
716 The first four arguments are (args, varargs, varkw, defaults). The
717 other four arguments are the corresponding optional formatting functions
718 that are called to turn names and values into strings. The ninth
719 argument is an optional function to format the sequence of arguments."""
722 firstdefault
= len(args
) - len(defaults
)
723 for i
in range(len(args
)):
724 spec
= strseq(args
[i
], formatarg
, join
)
725 if defaults
and i
>= firstdefault
:
726 spec
= spec
+ formatvalue(defaults
[i
- firstdefault
])
728 if varargs
is not None:
729 specs
.append(formatvarargs(varargs
))
730 if varkw
is not None:
731 specs
.append(formatvarkw(varkw
))
732 return '(' + string
.join(specs
, ', ') + ')'
734 def formatargvalues(args
, varargs
, varkw
, locals,
736 formatvarargs
=lambda name
: '*' + name
,
737 formatvarkw
=lambda name
: '**' + name
,
738 formatvalue
=lambda value
: '=' + repr(value
),
740 """Format an argument spec from the 4 values returned by getargvalues.
742 The first four arguments are (args, varargs, varkw, locals). The
743 next four arguments are the corresponding optional formatting functions
744 that are called to turn names and values into strings. The ninth
745 argument is an optional function to format the sequence of arguments."""
746 def convert(name
, locals=locals,
747 formatarg
=formatarg
, formatvalue
=formatvalue
):
748 return formatarg(name
) + formatvalue(locals[name
])
750 for i
in range(len(args
)):
751 specs
.append(strseq(args
[i
], convert
, join
))
753 specs
.append(formatvarargs(varargs
) + formatvalue(locals[varargs
]))
755 specs
.append(formatvarkw(varkw
) + formatvalue(locals[varkw
]))
756 return '(' + string
.join(specs
, ', ') + ')'
758 # -------------------------------------------------- stack frame extraction
759 def getframeinfo(frame
, context
=1):
760 """Get information about a frame or traceback object.
762 A tuple of five things is returned: the filename, the line number of
763 the current line, the function name, a list of lines of context from
764 the source code, and the index of the current line within that list.
765 The optional second argument specifies the number of lines of context
766 to return, which are centered around the current line."""
767 if istraceback(frame
):
768 lineno
= frame
.tb_lineno
769 frame
= frame
.tb_frame
771 lineno
= frame
.f_lineno
772 if not isframe(frame
):
773 raise TypeError('arg is not a frame or traceback object')
775 filename
= getsourcefile(frame
) or getfile(frame
)
777 start
= lineno
- 1 - context
//2
779 lines
, lnum
= findsource(frame
)
783 start
= max(start
, 1)
784 start
= max(0, min(start
, len(lines
) - context
))
785 lines
= lines
[start
:start
+context
]
786 index
= lineno
- 1 - start
790 return (filename
, lineno
, frame
.f_code
.co_name
, lines
, index
)
792 def getlineno(frame
):
793 """Get the line number from a frame object, allowing for optimization."""
794 # FrameType.f_lineno is now a descriptor that grovels co_lnotab
795 return frame
.f_lineno
797 def getouterframes(frame
, context
=1):
798 """Get a list of records for a frame and all higher (calling) frames.
800 Each record contains a frame object, filename, line number, function
801 name, a list of lines of context, and index within the context."""
804 framelist
.append((frame
,) + getframeinfo(frame
, context
))
808 def getinnerframes(tb
, context
=1):
809 """Get a list of records for a traceback's frame and all lower frames.
811 Each record contains a frame object, filename, line number, function
812 name, a list of lines of context, and index within the context."""
815 framelist
.append((tb
.tb_frame
,) + getframeinfo(tb
, context
))
819 currentframe
= sys
._getframe
821 def stack(context
=1):
822 """Return a list of records for the stack above the caller's frame."""
823 return getouterframes(sys
._getframe
(1), context
)
825 def trace(context
=1):
826 """Return a list of records for the stack below the current exception."""
827 return getinnerframes(sys
.exc_info()[2], context
)