# -*- 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 import sys 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 # pragma: nocover 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 self._get_branch_from_head_file(fh.readline()) except UnicodeDecodeError: # pragma: nocover try: with open(head, 'r', encoding=sys.getfilesystemencoding()) as fh: return self._get_branch_from_head_file(fh.readline()) except: log.traceback(logging.WARNING) except IOError: # pragma: nocover log.traceback(logging.WARNING) return u('master') def _project_base(self): if self.configFile: return os.path.dirname(os.path.dirname(self.configFile)) return None # pragma: nocover 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]) def _get_branch_from_head_file(self, line): if u(line.strip()).startswith('ref: '): return u(line.strip().rsplit('/', 1)[-1]) return None