Show More
@@ -110,65 +110,14 b' if sys.version_info[0] >= 3:' | |||
|
110 | 110 | except IndexError: |
|
111 | 111 | return False |
|
112 | 112 | |
|
113 | def _findargnofcall(n): | |
|
114 | """Find arg n of a call expression (start at 0) | |
|
115 | ||
|
116 | Returns index of the first token of that argument, or None if | |
|
117 | there is not that many arguments. | |
|
118 | ||
|
119 | Assumes that token[i + 1] is '('. | |
|
120 | ||
|
121 | """ | |
|
122 | nested = 0 | |
|
123 | for j in range(i + 2, len(tokens)): | |
|
124 | if _isop(j, ')', ']', '}'): | |
|
125 | # end of call, tuple, subscription or dict / set | |
|
126 | nested -= 1 | |
|
127 | if nested < 0: | |
|
128 | return None | |
|
129 | elif n == 0: | |
|
130 | # this is the starting position of arg | |
|
131 | return j | |
|
132 | elif _isop(j, '(', '[', '{'): | |
|
133 | nested += 1 | |
|
134 | elif _isop(j, ',') and nested == 0: | |
|
135 | n -= 1 | |
|
136 | ||
|
137 | return None | |
|
138 | ||
|
139 | def _ensureunicode(j): | |
|
140 | """Make sure the token at j is a unicode string | |
|
141 | ||
|
142 | This rewrites a string token to include the unicode literal prefix | |
|
143 | so the string transformer won't add the byte prefix. | |
|
144 | ||
|
145 | Ignores tokens that are not strings. Assumes bounds checking has | |
|
146 | already been done. | |
|
147 | ||
|
148 | """ | |
|
149 | st = tokens[j] | |
|
150 | if st.type == token.STRING and st.string.startswith(("'", '"')): | |
|
151 | tokens[j] = st._replace(string='u%s' % st.string) | |
|
152 | ||
|
153 | 113 | for i, t in enumerate(tokens): |
|
154 | 114 | # This looks like a function call. |
|
155 | 115 | if t.type == token.NAME and _isop(i + 1, '('): |
|
156 | 116 | fn = t.string |
|
157 | 117 | |
|
158 | # *attr() builtins don't accept byte strings to 2nd argument. | |
|
159 | if fn in ( | |
|
160 | 'getattr', | |
|
161 | 'setattr', | |
|
162 | 'hasattr', | |
|
163 | 'safehasattr', | |
|
164 | ) and not _isop(i - 1, '.'): | |
|
165 | arg1idx = _findargnofcall(1) | |
|
166 | if arg1idx is not None: | |
|
167 | _ensureunicode(arg1idx) | |
|
168 | ||
|
169 | 118 | # It changes iteritems/values to items/values as they are not |
|
170 | 119 | # present in Python 3 world. |
|
171 |
|
|
|
120 | if fn in ('iteritems', 'itervalues') and not ( | |
|
172 | 121 | tokens[i - 1].type == token.NAME |
|
173 | 122 | and tokens[i - 1].string == 'def' |
|
174 | 123 | ): |
@@ -182,7 +131,7 b' if sys.version_info[0] >= 3:' | |||
|
182 | 131 | # ``replacetoken`` or any mechanism that changes semantics of module |
|
183 | 132 | # loading is changed. Otherwise cached bytecode may get loaded without |
|
184 | 133 | # the new transformation mechanisms applied. |
|
185 |
BYTECODEHEADER = b'HG\x00\x1 |
|
|
134 | BYTECODEHEADER = b'HG\x00\x14' | |
|
186 | 135 | |
|
187 | 136 | class hgloader(importlib.machinery.SourceFileLoader): |
|
188 | 137 | """Custom module loader that transforms source code. |
@@ -255,7 +255,7 b' class KeepAliveHandler(object):' | |||
|
255 | 255 | # If not a persistent connection, don't try to reuse it. Look |
|
256 | 256 | # for this using getattr() since vcr doesn't define this |
|
257 | 257 | # attribute, and in that case always close the connection. |
|
258 |
if getattr(r, |
|
|
258 | if getattr(r, 'will_close', True): | |
|
259 | 259 | self._cm.remove(h) |
|
260 | 260 | |
|
261 | 261 | if DEBUG: |
@@ -70,7 +70,7 b' def _importfrom(pkgname, modname):' | |||
|
70 | 70 | except AttributeError: |
|
71 | 71 | raise ImportError(r'cannot import name %s' % modname) |
|
72 | 72 | # force import; fakelocals[modname] may be replaced with the real module |
|
73 |
getattr(mod, |
|
|
73 | getattr(mod, '__doc__', None) | |
|
74 | 74 | return fakelocals[modname] |
|
75 | 75 | |
|
76 | 76 | |
@@ -94,7 +94,7 b' def _importfrom(pkgname, modname):' | |||
|
94 | 94 | |
|
95 | 95 | def _checkmod(pkgname, modname, mod): |
|
96 | 96 | expected = _cextversions.get((pkgname, modname)) |
|
97 |
actual = getattr(mod, |
|
|
97 | actual = getattr(mod, 'version', None) | |
|
98 | 98 | if actual != expected: |
|
99 | 99 | raise ImportError( |
|
100 | 100 | r'cannot import module %s.%s ' |
@@ -270,7 +270,7 b' if ispy3:' | |||
|
270 | 270 | def getdoc(obj): |
|
271 | 271 | """Get docstring as bytes; may be None so gettext() won't confuse it |
|
272 | 272 | with _('')""" |
|
273 |
doc = getattr(obj, |
|
|
273 | doc = getattr(obj, '__doc__', None) | |
|
274 | 274 | if doc is None: |
|
275 | 275 | return doc |
|
276 | 276 | return sysbytes(doc) |
@@ -24,7 +24,7 b' from ..utils import storageutil' | |||
|
24 | 24 | |
|
25 | 25 | |
|
26 | 26 | class basetestcase(unittest.TestCase): |
|
27 |
if not getattr(unittest.TestCase, |
|
|
27 | if not getattr(unittest.TestCase, 'assertRaisesRegex', False): | |
|
28 | 28 | assertRaisesRegex = ( # camelcase-required |
|
29 | 29 | unittest.TestCase.assertRaisesRegexp |
|
30 | 30 | ) |
@@ -623,7 +623,7 b' class observedbufferedinputpipe(buffered' | |||
|
623 | 623 | def _fillbuffer(self): |
|
624 | 624 | res = super(observedbufferedinputpipe, self)._fillbuffer() |
|
625 | 625 | |
|
626 |
fn = getattr(self._input._observer, |
|
|
626 | fn = getattr(self._input._observer, 'osread', None) | |
|
627 | 627 | if fn: |
|
628 | 628 | fn(res, _chunksize) |
|
629 | 629 | |
@@ -634,7 +634,7 b' class observedbufferedinputpipe(buffered' | |||
|
634 | 634 | def read(self, size): |
|
635 | 635 | res = super(observedbufferedinputpipe, self).read(size) |
|
636 | 636 | |
|
637 |
fn = getattr(self._input._observer, |
|
|
637 | fn = getattr(self._input._observer, 'bufferedread', None) | |
|
638 | 638 | if fn: |
|
639 | 639 | fn(res, size) |
|
640 | 640 | |
@@ -643,7 +643,7 b' class observedbufferedinputpipe(buffered' | |||
|
643 | 643 | def readline(self, *args, **kwargs): |
|
644 | 644 | res = super(observedbufferedinputpipe, self).readline(*args, **kwargs) |
|
645 | 645 | |
|
646 |
fn = getattr(self._input._observer, |
|
|
646 | fn = getattr(self._input._observer, 'bufferedreadline', None) | |
|
647 | 647 | if fn: |
|
648 | 648 | fn(res) |
|
649 | 649 |
General Comments 0
You need to be logged in to leave comments.
Login now