1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
"""Plugin management for coverage.py"""
import sys
from coverage.misc import CoverageException
class CoveragePlugin(object):
"""Base class for coverage.py plugins."""
def __init__(self, options):
self.options = options
def file_tracer(self, filename):
"""Return a FileTracer object for this file."""
return None
def file_reporter(self, filename):
"""Return the FileReporter class to use for filename.
This will only be invoked if `filename` returns non-None from
`file_tracer`. It's an error to return None.
"""
raise Exception("Plugin %r needs to implement file_reporter" % self.plugin_name)
class FileTracer(object):
"""Support needed for files during the tracing phase."""
def source_filename(self):
return "xyzzy"
def dynamic_source_file_name(self):
"""Returns a callable that can return a source name for a frame.
The callable should take a filename and a frame, and return either a
filename or None:
def dynamic_source_filename_func(filename, frame)
Can return None if dynamic filenames aren't needed.
"""
return None
def line_number_range(self, frame):
"""Given a call frame, return the range of source line numbers."""
lineno = frame.f_lineno
return lineno, lineno
class FileReporter(object):
"""Support needed for files during the reporting phase."""
def __init__(self, filename):
self.filename = filename
class Plugins(object):
"""The currently loaded collection of coverage.py plugins."""
def __init__(self):
self.order = []
self.names = {}
@classmethod
def load_plugins(cls, modules, config):
"""Load plugins from `modules`.
Returns a list of loaded and configured plugins.
"""
plugins = cls()
for module in modules:
__import__(module)
mod = sys.modules[module]
plugin_class = getattr(mod, "Plugin", None)
if plugin_class:
options = config.get_plugin_options(module)
plugin = plugin_class(options)
plugin.plugin_name = module
plugins.order.append(plugin)
plugins.names[module] = plugin
return plugins
def __iter__(self):
return iter(self.order)
def get(self, module):
return self.names[module]
def overrides(obj, method_name, base_class):
"""Does `obj` override the `method_name` it got from `base_class`?
Determine if `obj` implements the method called `method_name`, which it
inherited from `base_class`.
Returns a boolean.
"""
klass = obj.__class__
klass_func = getattr(klass, method_name)
base_func = getattr(base_class, method_name)
# Python 2/3 compatibility: Python 2 returns an instancemethod object, the
# function is the .im_func attribute. Python 3 returns a plain function
# object already.
if sys.version_info < (3, 0):
klass_func = klass_func.im_func
base_func = base_func.im_func
return klass_func is not base_func
def plugin_implements(obj, method_name):
"""Does the plugin `obj` implement `method_name`?"""
return overrides(obj, method_name, CoveragePlugin)
|