support for Python3

wakatime/__init__.py

@@ -27,10 +27,17 @@ import re
 import sys
 import time
 import traceback
-import urllib2
+try:
+    from urllib2 import HTTPError, Request, urlopen
+except ImportError:
+    from urllib.error import HTTPError
+    from urllib.request import Request, urlopen
 
 from .log import setup_logging
 from .project import find_project
+try:
+    import argparse
+except ImportError:
     from .packages import argparse
 
 
@@ -130,15 +137,16 @@ def send_action(project=None, tags=None, key=None, targetFile=None,
     if tags:
         data['tags'] = list(set(tags))
     log.debug(data)
-    request = urllib2.Request(url=url, data=json.dumps(data))
+    request = Request(url=url, data=str.encode(json.dumps(data)))
     user_agent = get_user_agent(plugin)
     request.add_header('User-Agent', user_agent)
     request.add_header('Content-Type', 'application/json')
-    request.add_header('Authorization', 'Basic %s' % base64.b64encode(key))
+    auth = 'Basic %s' % bytes.decode(base64.b64encode(str.encode(key)))
+    request.add_header('Authorization', auth)
     response = None
     try:
-        response = urllib2.urlopen(request)
-    except urllib2.HTTPError as exc:
+        response = urlopen(request)
+    except HTTPError as exc:
         data = {
             'response_code': exc.getcode(),
             'response_content': exc.read(),

wakatime/log.py

@@ -22,6 +22,9 @@ except ImportError:
 class CustomEncoder(json.JSONEncoder):
 
     def default(self, obj):
+        if isinstance(obj, bytes):
+            obj = bytes.decode(obj)
+            return json.dumps(obj)
         return super(CustomEncoder, self).default(obj)
 
 

wakatime/projects/git.py

@@ -14,6 +14,9 @@ import os
 from subprocess import Popen, PIPE
 
 from .base import BaseProject
+try:
+    from collections import OrderedDict
+except ImportError:
     from ..packages.ordereddict import OrderedDict
 
 
@@ -67,6 +70,8 @@ class Git(BaseProject):
             pass
         if stdout:
             for line in stdout.splitlines():
+                if isinstance(line, bytes):
+                    line = bytes.decode(line)
                 line = line.split(' ', 1)
                 if line[0] == '*':
                     return line[1]

wakatime/projects/subversion.py

@@ -14,6 +14,9 @@ import os
 from subprocess import Popen, PIPE
 
 from .base import BaseProject
+try:
+    from collections import OrderedDict
+except ImportError:
     from ..packages.ordereddict import OrderedDict
 
 
@@ -51,6 +54,8 @@ class Subversion(BaseProject):
                     'URL',
                 ]
                 for line in stdout.splitlines():
+                    if isinstance(line, bytes):
+                        line = bytes.decode(line)
                     line = line.split(': ', 1)
                     if line[0] in interesting:
                         info[line[0]] = line[1]