VimPlug/parso
1
0
Fork 0
mirror of https://github.com/davidhalter/parso.git synced 2025-12-20 03:11:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Check for future imports in the beginning.

Browse Source
This commit is contained in:
Dave Halter
2017-07-20 09:21:28 +02:00
parent 03526cd84e
commit 7c9877cd2e
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
parso/python/normalizer.py
View File

@@ -14,6 +14,10 @@ class CompressNormalizer(Normalizer):
return leaf.prefix + leaf.value return leaf.prefix + leaf.value
def is_future_import(from_import):
from_names = from_import.get_from_names()
return [n.value for n in from_names] == ['__future__']
class Context(object): class Context(object):
def __init__(self, node, parent_context=None): def __init__(self, node, parent_context=None):
self.node = node self.node = node
@@ -80,6 +84,10 @@ class ErrorFinder(Normalizer):
yield yield
self._context = context self._context = context
return return
elif node.type == 'import_from' and node.level == 0 \
and is_future_import(node):
message = "from __future__ imports must occur at the beginning of the file"
self._add_syntax_error(message, node)
yield yield