lavatech
/
rana-cli
2
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

use C++ for header file language if directory contains C++ files

This commit is contained in:
Alan Hamlett 2015-08-10 18:17:50 -07:00
parent bd68a7767e
commit 2da75aa119
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
wakatime/stats.py
View File

@ -123,13 +123,15 @@ def get_language_from_extension(file_name):
"""Returns a matching language for the given file extension.
"""
extension = file_name.rsplit('.', 1)[-1] if len(file_name.rsplit('.', 1)) > 1 else ''
file_minus_extension = file_name.rsplit('.', 1)[0] if len(file_name.rsplit('.', 1)) > 1 else file_name
if extension.lower() == 'h':
if os.path.isfile(file_minus_extension + '.cpp'):
extension = os.path.splitext(file_name)[1].lower()
if extension == '.h':
directory = os.path.dirname(file_name)
available_files = os.listdir(directory)
available_extensions = zip(*map(os.path.splitext, available_files))[1]
available_extensions = [ext.lower() for ext in available_extensions]
if '.cpp' in available_extensions:
return 'C++'
if os.path.isfile(file_minus_extension + '.c'):
if '.c' in available_extensions:
return 'C'
return None