OLD | NEW |
| (Empty) |
1 """Miscellaneous stuff for Coverage.""" | |
2 | |
3 import inspect | |
4 from coverage.backward import md5, sorted # pylint: disable=W0622 | |
5 from coverage.backward import string_class, to_bytes | |
6 | |
7 | |
8 def nice_pair(pair): | |
9 """Make a nice string representation of a pair of numbers. | |
10 | |
11 If the numbers are equal, just return the number, otherwise return the pair | |
12 with a dash between them, indicating the range. | |
13 | |
14 """ | |
15 start, end = pair | |
16 if start == end: | |
17 return "%d" % start | |
18 else: | |
19 return "%d-%d" % (start, end) | |
20 | |
21 | |
22 def format_lines(statements, lines): | |
23 """Nicely format a list of line numbers. | |
24 | |
25 Format a list of line numbers for printing by coalescing groups of lines as | |
26 long as the lines represent consecutive statements. This will coalesce | |
27 even if there are gaps between statements. | |
28 | |
29 For example, if `statements` is [1,2,3,4,5,10,11,12,13,14] and | |
30 `lines` is [1,2,5,10,11,13,14] then the result will be "1-2, 5-11, 13-14". | |
31 | |
32 """ | |
33 pairs = [] | |
34 i = 0 | |
35 j = 0 | |
36 start = None | |
37 while i < len(statements) and j < len(lines): | |
38 if statements[i] == lines[j]: | |
39 if start == None: | |
40 start = lines[j] | |
41 end = lines[j] | |
42 j += 1 | |
43 elif start: | |
44 pairs.append((start, end)) | |
45 start = None | |
46 i += 1 | |
47 if start: | |
48 pairs.append((start, end)) | |
49 ret = ', '.join(map(nice_pair, pairs)) | |
50 return ret | |
51 | |
52 | |
53 def expensive(fn): | |
54 """A decorator to cache the result of an expensive operation. | |
55 | |
56 Only applies to methods with no arguments. | |
57 | |
58 """ | |
59 attr = "_cache_" + fn.__name__ | |
60 def _wrapped(self): | |
61 """Inner fn that checks the cache.""" | |
62 if not hasattr(self, attr): | |
63 setattr(self, attr, fn(self)) | |
64 return getattr(self, attr) | |
65 return _wrapped | |
66 | |
67 | |
68 def bool_or_none(b): | |
69 """Return bool(b), but preserve None.""" | |
70 if b is None: | |
71 return None | |
72 else: | |
73 return bool(b) | |
74 | |
75 | |
76 def join_regex(regexes): | |
77 """Combine a list of regexes into one that matches any of them.""" | |
78 if len(regexes) > 1: | |
79 return "(" + ")|(".join(regexes) + ")" | |
80 elif regexes: | |
81 return regexes[0] | |
82 else: | |
83 return "" | |
84 | |
85 | |
86 class Hasher(object): | |
87 """Hashes Python data into md5.""" | |
88 def __init__(self): | |
89 self.md5 = md5() | |
90 | |
91 def update(self, v): | |
92 """Add `v` to the hash, recursively if needed.""" | |
93 self.md5.update(to_bytes(str(type(v)))) | |
94 if isinstance(v, string_class): | |
95 self.md5.update(to_bytes(v)) | |
96 elif isinstance(v, (int, float)): | |
97 self.update(str(v)) | |
98 elif isinstance(v, (tuple, list)): | |
99 for e in v: | |
100 self.update(e) | |
101 elif isinstance(v, dict): | |
102 keys = v.keys() | |
103 for k in sorted(keys): | |
104 self.update(k) | |
105 self.update(v[k]) | |
106 else: | |
107 for k in dir(v): | |
108 if k.startswith('__'): | |
109 continue | |
110 a = getattr(v, k) | |
111 if inspect.isroutine(a): | |
112 continue | |
113 self.update(k) | |
114 self.update(a) | |
115 | |
116 def digest(self): | |
117 """Retrieve the digest of the hash.""" | |
118 return self.md5.digest() | |
119 | |
120 | |
121 class CoverageException(Exception): | |
122 """An exception specific to Coverage.""" | |
123 pass | |
124 | |
125 class NoSource(CoverageException): | |
126 """We couldn't find the source for a module.""" | |
127 pass | |
128 | |
129 class NotPython(CoverageException): | |
130 """A source file turned out not to be parsable Python.""" | |
131 pass | |
132 | |
133 class ExceptionDuringRun(CoverageException): | |
134 """An exception happened while running customer code. | |
135 | |
136 Construct it with three arguments, the values from `sys.exc_info`. | |
137 | |
138 """ | |
139 pass | |
OLD | NEW |