diff --git a/tracerouteparser.py b/tracerouteparser.py index d7afff5..d406fb5 100644 --- a/tracerouteparser.py +++ b/tracerouteparser.py @@ -36,7 +36,11 @@ # IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE # POSSIBILITY OF SUCH DAMAGE. -import cStringIO +try: + from StringIO import StringIO +except ImportError: + from io import StringIO + import re class Probe(object): @@ -109,7 +113,7 @@ def __str__(self): def parse_data(self, data): """Parser entry point, given string of the whole traceroute output.""" - self.parse_hdl(cStringIO.StringIO(data)) + self.parse_hdl(StringIO(data)) def parse_hdl(self, hdl): """Parser entry point, given readable file handle.""" @@ -206,7 +210,7 @@ def demo(): # Built-up data structures as string. Should look effectively # identical to the above input string. - print trp + print(trp) if __name__ == '__main__': demo()