Python 3 fix

2 files changed, 10 insertions, 2 deletions

diff --git a/gerber/excellon.py b/gerber/excellon.py
index ba8573d..7333a98 100755
--- a/gerber/excellon.py
+++ b/gerber/excellon.py
@@ -25,7 +25,11 @@ This module provides Excellon file classes and parsing utilities
 

 import math

 import operator

-from cStringIO import StringIO

+

+try:

+    from cStringIO import StringIO

+except(ImportError):

+    from io import StringIO

 

 from .excellon_statements import *

 from .cam import CamFile, FileSettings

diff --git a/gerber/rs274x.py b/gerber/rs274x.py
index 000f7a1..210b590 100644
--- a/gerber/rs274x.py
+++ b/gerber/rs274x.py
@@ -21,8 +21,12 @@
 import copy
 import json
 import re
-from cStringIO import StringIO
 
+try:
+    from cStringIO import StringIO
+except(ImportError):
+    from io import StringIO
+    
 from .gerber_statements import *
 from .primitives import *
 from .cam import CamFile, FileSettings