fix: Add python3 compatible imports

ganesh-git2014 · Sep 21, 2016 · f51fc2f · f51fc2f
1 parent f855d45
commit f51fc2f
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/catt/http_server.py b/catt/http_server.py
@@ -1,8 +1,14 @@
 import os
 import time
 import traceback
-import SocketServer
-from BaseHTTPServer import BaseHTTPRequestHandler
+try:
+    import SocketServer
+except ImportError:
+    import socketserver as SocketServer
+try:
+    from BaseHTTPServer import BaseHTTPRequestHandler
+except ImportError:
+    from http.server import BaseHTTPRequestHandler
 
 
 def serve_file(filename, address="", port=45114):