# -*- 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 subprocess import Popen, PIPE from .base import BaseProject from ..packages.ordereddict import OrderedDict log = logging.getLogger(__name__) class Git(BaseProject): def process(self): self.config = self._find_config(self.path) if self.config: return True return False def name(self): base = self._project_base() if base: return os.path.basename(base) return None def tags(self): tags = [] if self.config: proj_name = self.name() if proj_name: tags.append(proj_name) sections = self._parse_config() for section in sections: if section.split(' ', 1)[0] == 'remote' and 'url' in sections[section]: remote = sections[section]['url'].rsplit(':', 1)[-1].rsplit('/', 1)[-1].split('.git', 1)[0] if remote: tags.append(remote) branch = self._current_branch() if branch is not None: tags.append(branch) return tags def _project_base(self): if self.config: return os.path.dirname(os.path.dirname(self.config)) return None def _current_branch(self): stdout = None try: stdout, stderr = Popen([ 'git', 'branch', '--no-color', '--list' ], stdout=PIPE, stderr=PIPE, cwd=self._project_base() ).communicate() except OSError: pass if stdout: for line in stdout.splitlines(): line = line.split(' ', 1) if line[0] == '*': return line[1] return None 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, '.git', 'config')): return os.path.join(path, '.git', 'config') split_path = os.path.split(path) if split_path[1] == '': return None return self._find_config(split_path[0]) def _parse_config(self): sections = {} try: f = open(self.config, 'r') except IOError as e: log.exception("Exception:") else: with f: section = None for line in f.readlines(): line = line.lstrip() if len(line) > 0 and line[0] == '[': section = line[1:].split(']', 1)[0] temp = section.split(' ', 1) section = temp[0].lower() if len(temp) > 1: section = ' '.join([section, temp[1]]) sections[section] = {} else: try: (setting, value) = line.split('=', 1) except ValueError: setting = line.split('#', 1)[0].split(';', 1)[0] value = 'true' setting = setting.strip().lower() value = value.split('#', 1)[0].split(';', 1)[0].strip() sections[section][setting] = value f.close() return sections