update to python 3.8+

Author: a16bitsysop

pypytools/pypylog/model.py

@@ -1,7 +1,10 @@
-import itertools
 from collections import defaultdict
 import attr
 import numpy as np
+from pypytools.util import PY3
+
+if not PY3:
+    import itertools
 
 @attr.s
 class Event(object):
@@ -46,15 +49,15 @@ def add_event(self, ev):
 
     def print_summary(self):
         fmt = '%-28s %6s %8s'
-        print fmt % ('section', 'n', 'delta')
-        print '-'*44
+        print(fmt % ('section', 'n', 'delta'))
+        print('-'*44)
         for name, events in sorted(self.sections.iteritems()):
             total = 0
             for ev in events:
                 delta = ev.end - ev.start
                 assert delta >= 0
                 total += delta
-            print fmt % (name, len(events), format(delta, '.4f'))
+            print(fmt % (name, len(events), format(delta, '.4f')))
 
 class Series(object):
 
@@ -79,8 +82,12 @@ def __len__(self):
         return len(self.X)
 
     def __iter__(self):
-        for x, y in itertools.izip(self.X, self.Y):
-            yield x, y
+        if PY3:
+            for x, y in zip(self.X, self.Y):
+                yield x, y
+        else:
+            for x, y in itertools.izip(self.X, self.Y):
+                yield x, y
 
     def __getitem__(self, i):
         return self.X[i], self.Y[i]

pypytools/pypylog/parse.py

@@ -1,6 +1,7 @@
 import re
 import attr
 from pypytools.pypylog import model
+from pypytools.util import PY3
 
 # stolen from rpython/tool/logparse.py
 _color = "(?:\x1b.*?m)?"
@@ -35,6 +36,8 @@ def parse_file(f):
         #
         if log is None:
             log = model.PyPyLog()
+        if PY3:
+            basestring = str
         if isinstance(fname, basestring):
             with open(fname) as f:
                 return parse_file(f)

pypytools/pypylog/testing/test_parse.py

@@ -1,6 +1,13 @@
 import pytest
 import textwrap
-from cStringIO import StringIO
+
+from pypytools.util import PY3
+
+if PY3:
+    from io import StringIO
+else:
+    from cStringIO import StringIO
+
 from pypytools.pypylog import parse
 from pypytools.pypylog import model
 from pypytools.pypylog.model import Event, GcMinor, GcCollectStep
@@ -33,7 +40,7 @@ def test_mismatch(self):
         [456] foo}
         [0ab] bar}
         """
-        pytest.raises(parse.ParseError, "self.parse(log)")
+        pytest.raises(parse.ParseError(self.parse(log)))
 
     def test_nested(self):
         log = self.parse("""
@@ -124,4 +131,4 @@ def test_parse_frequency():
     assert pf('40 KHz') == 40e3
     assert pf('40 MHz') == 40e6
     assert pf('40 GHz') == 40e9
-    pytest.raises(ValueError, "pf('')")
+    pytest.raises(ValueError(pf('')))

pypytools/util.py

@@ -2,6 +2,7 @@
 from sys import version_info
 
 PY3 = version_info.major == 3
+PY3M = version_info.minor
 
 def clonefunc(f):
     """Deep clone the given function to create a new one.
@@ -22,6 +23,8 @@ def clonefunc(f):
             co.co_firstlineno, co.co_lnotab, co.co_freevars, co.co_cellvars]
     if PY3:
         args.insert(1, co.co_kwonlyargcount)
+    if PY3 and PY3M >= 8:
+        args.insert(1, co.co_posonlyargcount)
     co2 = types.CodeType(*args)
     #
     # then, we clone the function itself, using the new co2