sublime-wakatime/packages/wakatime/dependencies/python.py

86 lines
2.6 KiB
Python
Raw Normal View History

2014-12-23 14:39:07 +03:00
# -*- coding: utf-8 -*-
"""
wakatime.languages.python
~~~~~~~~~~~~~~~~~~~~~~~~~
Parse dependencies from Python code.
2014-12-25 10:04:39 +03:00
:copyright: (c) 2014 Alan Hamlett.
2014-12-23 14:39:07 +03:00
:license: BSD, see LICENSE for more details.
"""
from . import TokenParser
class PythonParser(TokenParser):
state = None
parens = 0
nonpackage = False
2015-09-29 13:11:25 +03:00
exclude = [
r'^os$',
r'^sys\.',
]
2014-12-23 14:39:07 +03:00
2015-09-29 13:11:25 +03:00
def parse(self):
2014-12-23 14:39:07 +03:00
for index, token, content in self.tokens:
self._process_token(token, content)
return self.dependencies
def _process_token(self, token, content):
2015-09-29 13:11:25 +03:00
if self.partial(token) == 'Namespace':
2014-12-23 14:39:07 +03:00
self._process_namespace(token, content)
2015-09-29 13:11:25 +03:00
elif self.partial(token) == 'Operator':
2014-12-23 14:39:07 +03:00
self._process_operator(token, content)
2015-09-29 13:11:25 +03:00
elif self.partial(token) == 'Punctuation':
2014-12-23 14:39:07 +03:00
self._process_punctuation(token, content)
2015-09-29 13:11:25 +03:00
elif self.partial(token) == 'Text':
2014-12-23 14:39:07 +03:00
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:
2014-12-25 10:04:39 +03:00
if content == 'as':
self.nonpackage = True
else:
self._process_import(token, content)
2014-12-23 14:39:07 +03:00
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):
2014-12-25 10:04:39 +03:00
if not self.nonpackage:
if self.state == 'from':
2015-09-29 13:11:25 +03:00
self.append(content, truncate=True, truncate_to=1)
2014-12-25 10:04:39 +03:00
self.state = 'from-2'
elif self.state == 'from-2' and content != 'import':
2015-09-29 13:11:25 +03:00
self.append(content, truncate=True, truncate_to=1)
2014-12-25 10:04:39 +03:00
elif self.state == 'import':
2015-09-29 13:11:25 +03:00
self.append(content, truncate=True, truncate_to=1)
2014-12-25 10:04:39 +03:00
self.state = 'import-2'
elif self.state == 'import-2':
2015-09-29 13:11:25 +03:00
self.append(content, truncate=True, truncate_to=1)
2014-12-25 10:04:39 +03:00
else:
self.state = None
2014-12-23 14:39:07 +03:00
self.nonpackage = False