network: Add missing freeaddrinfo on error path.

[collectd.git] / contrib / collectd_network.py

diff --git a/contrib/collectd_network.py b/contrib/collectd_network.py
index dc6cdac..cb328f2 100644 (file)
--- a/contrib/collectd_network.py
+++ b/contrib/collectd_network.py
@@ -16,11 +16,16 @@
 Collectd network protocol implementation.
 """
 
-import socket,struct,sys
-try:
-  from io import StringIO
-except ImportError:
-  from cStringIO import StringIO
+import socket,struct
+import platform
+if platform.python_version() < '2.8.0':
+    # Python 2.7 and below io.StringIO does not like unicode
+    from StringIO import StringIO
+else:
+    try:
+      from io import StringIO
+    except ImportError:
+      from cStringIO import StringIO
 
 from datetime import datetime
 from copy import deepcopy