diff --git a/wakatime/languages/python.py b/wakatime/languages/python.py new file mode 100644 index 0000000..1b1a087 --- /dev/null +++ b/wakatime/languages/python.py @@ -0,0 +1,116 @@ +# -*- coding: utf-8 -*- +""" + wakatime.languages.python + ~~~~~~~~~~~~~~~~~~~~~~~~~ + + Parse dependencies from Python code. + + :copyright: (c) 2013 Alan Hamlett. + :license: BSD, see LICENSE for more details. +""" + +from . import TokenParser +from ..compat import u + + +class PythonParser(TokenParser): + state = None + parens = 0 + nonpackage = False + + def parse(self, tokens=[]): + if not tokens and not self.tokens: + self.tokens = self._extract_tokens() + for index, token, content in self.tokens: + self._process_token(token, content) + return self.dependencies + + def _process_token(self, token, content): + if u(token).split('.')[-1] == 'Namespace': + self._process_namespace(token, content) + elif u(token).split('.')[-1] == 'Name': + self._process_name(token, content) + elif u(token).split('.')[-1] == 'Word': + self._process_word(token, content) + elif u(token).split('.')[-1] == 'Operator': + self._process_operator(token, content) + elif u(token).split('.')[-1] == 'Punctuation': + self._process_punctuation(token, content) + elif u(token).split('.')[-1] == 'Text': + self._process_text(token, content) + else: + self._process_other(token, content) + + def _process_namespace(self, token, content): + if self.state is None: + self.state = content + else: + self._process_import(token, content) + + def _process_name(self, token, content): + if self.state is not None: + if self.nonpackage: + self.nonpackage = False + else: + if self.state == 'from': + self.append(content) + if self.state == 'from-2' and content != 'import': + self.append(content) + elif self.state == 'import': + self.append(content) + elif self.state == 'import-2': + self.append(content) + else: + self.state = None + + def _process_word(self, token, content): + if self.state is not None: + if self.nonpackage: + self.nonpackage = False + else: + if self.state == 'from': + self.append(content) + if self.state == 'from-2' and content != 'import': + self.append(content) + elif self.state == 'import': + self.append(content) + elif self.state == 'import-2': + self.append(content) + else: + self.state = None + + def _process_operator(self, token, content): + if self.state is not None: + if content == '.': + self.nonpackage = True + + def _process_punctuation(self, token, content): + if content == '(': + self.parens += 1 + elif content == ')': + self.parens -= 1 + self.nonpackage = False + + def _process_text(self, token, content): + if self.state is not None: + if content == "\n" and self.parens == 0: + self.state = None + self.nonpackage = False + + def _process_other(self, token, content): + pass + + def _process_import(self, token, content): + self.nonpackage = False + if self.state == 'from': + self.append(content) + self.state = 'from-2' + elif self.state == 'from-2' and content != 'import': + self.append(content) + elif self.state == 'import': + self.append(content) + self.state = 'import-2' + elif self.state == 'import-2': + self.append(content) + else: + self.state = None