|
12 | 12 | # See the License for the specific language governing permissions and
|
13 | 13 | # limitations under the License.
|
14 | 14 |
|
15 |
| -from typing import Any, Callable, Mapping, Sequence |
| 15 | +import sys |
16 | 16 | import types
|
| 17 | +from typing import Any, Callable, Mapping, Sequence |
17 | 18 | from inspect import Parameter, Signature
|
18 | 19 |
|
19 | 20 |
|
@@ -44,21 +45,31 @@ def pass_locals():
|
44 | 45 | mod_co_filename = code.co_filename
|
45 | 46 | mod_co_firstlineno = code.co_firstlineno
|
46 | 47 |
|
47 |
| - modified_code = types.CodeType( |
48 |
| - mod_co_argcount, |
49 |
| - code.co_kwonlyargcount, |
50 |
| - mod_co_nlocals, |
51 |
| - code.co_stacksize, |
52 |
| - code.co_flags, |
53 |
| - code.co_code, |
54 |
| - code.co_consts, |
55 |
| - code.co_names, |
56 |
| - mod_co_varnames, |
57 |
| - mod_co_filename, |
58 |
| - mod_co_name, |
59 |
| - mod_co_firstlineno, |
60 |
| - code.co_lnotab |
61 |
| - ) |
| 48 | + if sys.version_info >= (3, 8): |
| 49 | + modified_code = code.replace( |
| 50 | + co_argcount=mod_co_argcount, |
| 51 | + co_nlocals=mod_co_nlocals, |
| 52 | + co_varnames=mod_co_varnames, |
| 53 | + co_filename=mod_co_filename, |
| 54 | + co_name=mod_co_name, |
| 55 | + co_firstlineno=mod_co_firstlineno, |
| 56 | + ) |
| 57 | + else: |
| 58 | + modified_code = types.CodeType( |
| 59 | + mod_co_argcount, |
| 60 | + code.co_kwonlyargcount, |
| 61 | + mod_co_nlocals, |
| 62 | + code.co_stacksize, |
| 63 | + code.co_flags, |
| 64 | + code.co_code, |
| 65 | + code.co_consts, |
| 66 | + code.co_names, |
| 67 | + mod_co_varnames, |
| 68 | + mod_co_filename, |
| 69 | + mod_co_name, |
| 70 | + mod_co_firstlineno, |
| 71 | + code.co_lnotab |
| 72 | + ) |
62 | 73 |
|
63 | 74 | default_arg_values = tuple( p.default for p in parameters if p.default != Parameter.empty ) #!argdefs "starts from the right"/"is right-aligned"
|
64 | 75 | modified_func = types.FunctionType(modified_code, {'dict_func': func, 'locals': locals}, name=func_name, argdefs=default_arg_values)
|
|
0 commit comments