Python-3.8 PEP570 positional only argument

pacman1199 · May 11, 2019 · 248128d · 248128d
1 parent 380db0d
commit 248128d
Showing 1 changed file with 37 additions and 17 deletions.
diff --git a/IPython/core/interactiveshell.py b/IPython/core/interactiveshell.py
@@ -138,23 +138,43 @@ def removed_co_newlocals(function:types.FunctionType) -> types.FunctionType:
     from types import CodeType, FunctionType
     CO_NEWLOCALS = 0x0002
     code = function.__code__
-    new_code = CodeType(
-        code.co_argcount, 
-        code.co_kwonlyargcount,
-        code.co_nlocals, 
-        code.co_stacksize, 
-        code.co_flags & ~CO_NEWLOCALS,
-        code.co_code, 
-        code.co_consts,
-        code.co_names, 
-        code.co_varnames, 
-        code.co_filename, 
-        code.co_name, 
-        code.co_firstlineno, 
-        code.co_lnotab, 
-        code.co_freevars, 
-        code.co_cellvars
-    )
+    if sys.version_info > (3,8):
+        new_code = CodeType(
+            code.co_argcount,
+            code.co_posonlyargcount, # Python-3.8 PEP570 positional only argument
+            code.co_kwonlyargcount,
+            code.co_nlocals, 
+            code.co_stacksize, 
+            code.co_flags & ~CO_NEWLOCALS,
+            code.co_code, 
+            code.co_consts,
+            code.co_names, 
+            code.co_varnames, 
+            code.co_filename, 
+            code.co_name, 
+            code.co_firstlineno, 
+            code.co_lnotab, 
+            code.co_freevars, 
+            code.co_cellvars
+        )
+    else:
+        new_code = CodeType(
+            code.co_argcount,
+            code.co_kwonlyargcount,
+            code.co_nlocals, 
+            code.co_stacksize, 
+            code.co_flags & ~CO_NEWLOCALS,
+            code.co_code, 
+            code.co_consts,
+            code.co_names, 
+            code.co_varnames, 
+            code.co_filename, 
+            code.co_name, 
+            code.co_firstlineno, 
+            code.co_lnotab, 
+            code.co_freevars, 
+            code.co_cellvars
+        )            
     return FunctionType(new_code, globals(), function.__name__, function.__defaults__)