From d664279297e66045b3845c765b185021056debfd Mon Sep 17 00:00:00 2001 From: Alan Hamlett Date: Fri, 25 Oct 2013 21:32:30 -0700 Subject: [PATCH] upgrade wakatime package --- .../wakatime/wakatime/projects/subversion.py | 3 +++ plugin/packages/wakatime/wakatime/stats.py | 19 +++++++++++++++++++ 2 files changed, 22 insertions(+) diff --git a/plugin/packages/wakatime/wakatime/projects/subversion.py b/plugin/packages/wakatime/wakatime/projects/subversion.py index 6a7aed5..403f344 100644 --- a/plugin/packages/wakatime/wakatime/projects/subversion.py +++ b/plugin/packages/wakatime/wakatime/projects/subversion.py @@ -11,6 +11,7 @@ import logging import os +import platform from subprocess import Popen, PIPE from .base import BaseProject @@ -62,6 +63,8 @@ class Subversion(BaseProject): 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] diff --git a/plugin/packages/wakatime/wakatime/stats.py b/plugin/packages/wakatime/wakatime/stats.py index ac29996..9aaed1e 100644 --- a/plugin/packages/wakatime/wakatime/stats.py +++ b/plugin/packages/wakatime/wakatime/stats.py @@ -23,7 +23,17 @@ from pygments.lexers import guess_lexer_for_filename log = logging.getLogger(__name__) +# force file name extensions to be recognized as a certain language +EXTENSIONS = { + 'md': 'Markdown', +} + + def guess_language(file_name): + if file_name: + language = guess_language_from_extension(file_name.rsplit('.', 1)[-1]) + if language: + return language lexer = None try: with open(file_name) as f: @@ -36,6 +46,15 @@ def guess_language(file_name): return None +def guess_language_from_extension(extension): + if extension: + if extension in EXTENSIONS: + return EXTENSIONS[extension] + if extension.lower() in EXTENSIONS: + return mapping[EXTENSIONS.lower()] + return None + + def number_lines_in_file(file_name): lines = 0 try: