upgrade wakatime package

This commit is contained in:
Alan Hamlett 2013-10-25 21:33:31 -07:00
parent 791a969a10
commit 744116079a

View File

@ -23,7 +23,17 @@ from pygments.lexers import guess_lexer_for_filename
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
# force file name extensions to be recognized as a certain language
EXTENSIONS = {
'md': 'Markdown',
}
def guess_language(file_name): def guess_language(file_name):
if file_name:
language = guess_language_from_extension(file_name.rsplit('.', 1)[-1])
if language:
return language
lexer = None lexer = None
try: try:
with open(file_name) as f: with open(file_name) as f:
@ -36,6 +46,15 @@ def guess_language(file_name):
return None 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): def number_lines_in_file(file_name):
lines = 0 lines = 0
try: try: