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

handle unknown language when parsing dependencies

Browse source
This commit is contained in:
Alan Hamlett 2014-12-23 05:30:40 -06:00
parent 0f76334fc3
commit 78640aebf2
1 changed files with 12 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
wakatime/languages/__init__.py
View file

@ -42,8 +42,10 @@ class TokenParser(object):
self._save_dependency(dep, truncate=truncate) self._save_dependency(dep, truncate=truncate)
def _extract_tokens(self): def _extract_tokens(self):
with open(self.source_file, 'r', encoding='utf-8') as fh: if self.lexer:
return self.lexer.get_tokens_unprocessed(fh.read(512000)) with open(self.source_file, 'r', encoding='utf-8') as fh:
return self.lexer.get_tokens_unprocessed(fh.read(512000))
return []
def _save_dependency(self, dep, truncate=True): def _save_dependency(self, dep, truncate=True):
dep = dep.strip().split('.')[0].strip() if truncate else dep.strip() dep = dep.strip().split('.')[0].strip() if truncate else dep.strip()
@ -60,13 +62,14 @@ class DependencyParser(object):
self.source_file = source_file self.source_file = source_file
self.lexer = lexer self.lexer = lexer
try: if self.lexer:
module_name = self.lexer.__module__.split('.')[-1] try:
class_name = self.lexer.__class__.__name__.replace('Lexer', 'Parser', 1) module_name = self.lexer.__module__.split('.')[-1]
module = import_module('.%s' % module_name, package=__package__) class_name = self.lexer.__class__.__name__.replace('Lexer', 'Parser', 1)
self.parser = getattr(module, class_name) module = import_module('.%s' % module_name, package=__package__)
except ImportError as ex: self.parser = getattr(module, class_name)
log.debug(ex) except ImportError as ex:
log.debug(ex)
def parse(self): def parse(self):
if self.parser: if self.parser: