Mercurial Repositories > eric / file diff

diff -r f415080106ba -r affc66a603c7 DataViews/CodeMetrics.py
--- a/DataViews/CodeMetrics.py	Sat Apr 20 19:52:20 2013 +0200
+++ b/DataViews/CodeMetrics.py	Sat Apr 20 20:29:37 2013 +0200
@@ -18,6 +18,7 @@
 
 import os
 import io
+import sys
 import keyword
 import token
 import tokenize
@@ -67,7 +68,10 @@
         
         source = io.BytesIO(text.encode("utf-8"))
         try:
-            gen = tokenize.tokenize(source.readline)
+            if sys.version_info[0] == 2:
+                gen = tokenize.generate_tokens(source.readline)
+            else:
+                gen = tokenize.tokenize(source.readline)
             for toktype, toktext, start, end, line in gen:
                 (srow, scol) = start
                 (erow, ecol) = end