mirror of https://github.com/ansible/ansible.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
3 years ago
|
from __future__ import annotations
|
||
3 years ago
|
|
||
|
import ast
|
||
|
import sys
|
||
|
|
||
|
|
||
|
def main():
|
||
|
for path in sys.argv[1:] or sys.stdin.read().splitlines():
|
||
|
with open(path, 'rb') as path_fd:
|
||
|
lines = path_fd.read().splitlines()
|
||
|
|
||
|
missing = True
|
||
|
if not lines:
|
||
|
# Files are allowed to be empty of everything including boilerplate
|
||
|
missing = False
|
||
|
|
||
3 years ago
|
invalid_future = []
|
||
|
|
||
3 years ago
|
for text in lines:
|
||
|
if text == b'from __future__ import annotations':
|
||
|
missing = False
|
||
|
break
|
||
|
|
||
1 year ago
|
if text.strip().startswith(b'from __future__ ') or text.strip().startswith(b'__metaclass__ '):
|
||
3 years ago
|
invalid_future.append(text.decode())
|
||
|
|
||
3 years ago
|
if missing:
|
||
|
with open(path) as file:
|
||
|
contents = file.read()
|
||
|
|
||
|
# noinspection PyBroadException
|
||
|
try:
|
||
|
node = ast.parse(contents)
|
||
|
|
||
|
# files consisting of only assignments have no need for future import boilerplate
|
||
|
# the most likely case is that of a documentation only python file
|
||
|
if all(isinstance(statement, ast.Assign) for statement in node.body):
|
||
|
missing = False
|
||
|
except Exception: # pylint: disable=broad-except
|
||
|
pass # the compile sanity test will report this error
|
||
|
|
||
|
if missing:
|
||
|
print('%s: missing: from __future__ import annotations' % path)
|
||
|
|
||
3 years ago
|
for text in invalid_future:
|
||
|
print('%s: invalid: %s' % (path, text))
|
||
|
|
||
3 years ago
|
|
||
|
if __name__ == '__main__':
|
||
|
main()
|