upgrade wakatime cli to v4.0.4

packages/wakatime/projects/base.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+"""
+    wakatime.projects.base
+    ~~~~~~~~~~~~~~~~~~~~~~
+
+    Base project for use when no other project can be found.
+
+    :copyright: (c) 2013 Alan Hamlett.
+    :license: BSD, see LICENSE for more details.
+"""
+
+import logging
+
+
+log = logging.getLogger('WakaTime')
+
+
+class BaseProject(object):
+    """ Parent project class only
+    used when no valid project can
+    be found for the current path.
+    """
+
+    def __init__(self, path, configs=None):
+        self.path = path
+        self._configs = configs
+
+    def project_type(self):
+        """ Returns None if this is the base class.
+        Returns the type of project if this is a
+        valid project.
+        """
+        project_type = self.__class__.__name__.lower()
+        if project_type == 'baseproject':
+            project_type = None
+        return project_type
+
+    def process(self):
+        """ Processes self.path into a project and
+        returns True if project is valid, otherwise
+        returns False.
+        """
+        return False
+
+    def name(self):
+        """ Returns the project's name.
+        """
+        return None
+
+    def branch(self):
+        """ Returns the current branch.
+        """
+        return None

packages/wakatime/projects/git.py

@@ -0,0 +1,59 @@
+# -*- coding: utf-8 -*-
+"""
+    wakatime.projects.git
+    ~~~~~~~~~~~~~~~~~~~~~
+
+    Information about the git project for a given file.
+
+    :copyright: (c) 2013 Alan Hamlett.
+    :license: BSD, see LICENSE for more details.
+"""
+
+import logging
+import os
+
+from .base import BaseProject
+from ..compat import u, open
+
+
+log = logging.getLogger('WakaTime')
+
+
+class Git(BaseProject):
+
+    def process(self):
+        self.configFile = self._find_git_config_file(self.path)
+        return self.configFile is not None
+
+    def name(self):
+        base = self._project_base()
+        if base:
+            return u(os.path.basename(base))
+        return None
+
+    def branch(self):
+        base = self._project_base()
+        if base:
+            head = os.path.join(self._project_base(), '.git', 'HEAD')
+            try:
+                with open(head, 'r', encoding='utf-8') as fh:
+                    return u(fh.readline().strip().rsplit('/', 1)[-1])
+            except IOError:
+                pass
+        return None
+
+    def _project_base(self):
+        if self.configFile:
+            return os.path.dirname(os.path.dirname(self.configFile))
+        return None
+
+    def _find_git_config_file(self, path):
+        path = os.path.realpath(path)
+        if os.path.isfile(path):
+            path = os.path.split(path)[0]
+        if os.path.isfile(os.path.join(path, '.git', 'config')):
+            return os.path.join(path, '.git', 'config')
+        split_path = os.path.split(path)
+        if split_path[1] == '':
+            return None
+        return self._find_git_config_file(split_path[0])

packages/wakatime/projects/mercurial.py

@@ -0,0 +1,52 @@
+# -*- coding: utf-8 -*-
+"""
+    wakatime.projects.mercurial
+    ~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+    Information about the mercurial project for a given file.
+
+    :copyright: (c) 2013 Alan Hamlett.
+    :license: BSD, see LICENSE for more details.
+"""
+
+import logging
+import os
+
+from .base import BaseProject
+from ..compat import u, open
+
+
+log = logging.getLogger('WakaTime')
+
+
+class Mercurial(BaseProject):
+
+    def process(self):
+        self.configDir = self._find_hg_config_dir(self.path)
+        return self.configDir is not None
+
+    def name(self):
+        if self.configDir:
+            return u(os.path.basename(os.path.dirname(self.configDir)))
+        return None
+
+    def branch(self):
+        if self.configDir:
+            branch_file = os.path.join(self.configDir, 'branch')
+            try:
+                with open(branch_file, 'r', encoding='utf-8') as fh:
+                    return u(fh.readline().strip().rsplit('/', 1)[-1])
+            except IOError:
+                pass
+        return u('default')
+
+    def _find_hg_config_dir(self, path):
+        path = os.path.realpath(path)
+        if os.path.isfile(path):
+            path = os.path.split(path)[0]
+        if os.path.isdir(os.path.join(path, '.hg')):
+            return os.path.join(path, '.hg')
+        split_path = os.path.split(path)
+        if split_path[1] == '':
+            return None
+        return self._find_hg_config_dir(split_path[0])

packages/wakatime/projects/projectmap.py

@@ -0,0 +1,66 @@
+# -*- coding: utf-8 -*-
+"""
+    wakatime.projects.projectmap
+    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+    Use the ~/.wakatime.cfg file to set custom project names by
+    recursively matching folder paths.
+    Project maps go under the [projectmap] config section.
+
+    For example:
+
+        [projectmap]
+        /home/user/projects/foo = new project name
+        /home/user/projects/bar = project2
+
+    Will result in file `/home/user/projects/foo/src/main.c` to have
+    project name `new project name`.
+
+    :copyright: (c) 2013 Alan Hamlett.
+    :license: BSD, see LICENSE for more details.
+"""
+
+import logging
+import os
+
+from .base import BaseProject
+from ..compat import u
+
+
+log = logging.getLogger('WakaTime')
+
+
+class ProjectMap(BaseProject):
+
+    def process(self):
+        if not self._configs:
+            return False
+
+        self.project = self._find_project(self.path)
+
+        return self.project is not None
+
+    def _find_project(self, path):
+        path = os.path.realpath(path)
+        if os.path.isfile(path):
+            path = os.path.split(path)[0]
+
+        if self._configs.get(path.lower()):
+            return self._configs.get(path.lower())
+        if self._configs.get('%s/' % path.lower()):
+            return self._configs.get('%s/' % path.lower())
+        if self._configs.get('%s\\' % path.lower()):
+            return self._configs.get('%s\\' % path.lower())
+
+        split_path = os.path.split(path)
+        if split_path[1] == '':
+            return None
+        return self._find_project(split_path[0])
+
+    def branch(self):
+        return None
+
+    def name(self):
+        if self.project:
+            return u(self.project)
+        return None

packages/wakatime/projects/subversion.py

@@ -0,0 +1,96 @@
+# -*- coding: utf-8 -*-
+"""
+    wakatime.projects.subversion
+    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+    Information about the svn project for a given file.
+
+    :copyright: (c) 2013 Alan Hamlett.
+    :license: BSD, see LICENSE for more details.
+"""
+
+import logging
+import os
+import platform
+from subprocess import Popen, PIPE
+
+from .base import BaseProject
+from ..compat import u, open
+try:
+    from collections import OrderedDict
+except ImportError:
+    from ..packages.ordereddict import OrderedDict
+
+
+log = logging.getLogger('WakaTime')
+
+
+class Subversion(BaseProject):
+    binary_location = None
+
+    def process(self):
+        return self._find_project_base(self.path)
+
+    def name(self):
+        return u(self.info['Repository Root'].split('/')[-1])
+
+    def branch(self):
+        return u(self.info['URL'].split('/')[-1])
+
+    def _find_binary(self):
+        if self.binary_location:
+            return self.binary_location
+        locations = [
+            'svn',
+            '/usr/bin/svn',
+            '/usr/local/bin/svn',
+        ]
+        for location in locations:
+            with open(os.devnull, 'wb') as DEVNULL:
+                try:
+                    Popen([location, '--version'], stdout=DEVNULL, stderr=DEVNULL)
+                    self.binary_location = location
+                    return location
+                except:
+                    pass
+        self.binary_location = 'svn'
+        return 'svn'
+
+    def _get_info(self, path):
+        info = OrderedDict()
+        stdout = None
+        try:
+            os.environ['LANG'] = 'en_US'
+            stdout, stderr = Popen([
+                self._find_binary(), 'info', os.path.realpath(path)
+            ], stdout=PIPE, stderr=PIPE).communicate()
+        except OSError:
+            pass
+        else:
+            if stdout:
+                for line in stdout.splitlines():
+                    if isinstance(line, bytes):
+                        line = bytes.decode(line)
+                    line = line.split(': ', 1)
+                    if len(line) == 2:
+                        info[line[0]] = line[1]
+        return info
+
+    def _find_project_base(self, path, found=False):
+        if platform.system() == 'Windows':
+            return False
+        path = os.path.realpath(path)
+        if os.path.isfile(path):
+            path = os.path.split(path)[0]
+        info = self._get_info(path)
+        if len(info) > 0:
+            found = True
+            self.base = path
+            self.info = info
+        elif found:
+            return True
+        split_path = os.path.split(path)
+        if split_path[1] == '':
+            return found
+        return self._find_project_base(split_path[0], found)
+

packages/wakatime/projects/wakatime.py

@@ -0,0 +1,58 @@
+# -*- coding: utf-8 -*-
+"""
+    wakatime.projects.wakatime
+    ~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+    Information from a .wakatime-project file about the project for
+    a given file. First line of .wakatime-project sets the project
+    name. Second line sets the current branch name.
+
+    :copyright: (c) 2013 Alan Hamlett.
+    :license: BSD, see LICENSE for more details.
+"""
+
+import logging
+import os
+
+from .base import BaseProject
+from ..compat import u, open
+
+
+log = logging.getLogger('WakaTime')
+
+
+class WakaTime(BaseProject):
+
+    def process(self):
+        self.config = self._find_config(self.path)
+        self._project_name = None
+        self._project_branch = None
+
+        if self.config:
+
+            try:
+                with open(self.config, 'r', encoding='utf-8') as fh:
+                    self._project_name = u(fh.readline().strip())
+                    self._project_branch = u(fh.readline().strip())
+            except IOError:
+                log.exception("Exception:")
+
+            return True
+        return False
+
+    def name(self):
+        return self._project_name
+
+    def branch(self):
+        return self._project_branch
+
+    def _find_config(self, path):
+        path = os.path.realpath(path)
+        if os.path.isfile(path):
+            path = os.path.split(path)[0]
+        if os.path.isfile(os.path.join(path, '.wakatime-project')):
+            return os.path.join(path, '.wakatime-project')
+        split_path = os.path.split(path)
+        if split_path[1] == '':
+            return None
+        return self._find_config(split_path[0])