ansible/test/sanity/code-smell/future-import-boilerplate.py

#!/usr/bin/env python
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type

import sys


def main():
    # We classify files in these directories as ones we don't care about fixing (at least for now)
    prune = (
        'contrib/inventory/',
        'contrib/vault/',
        'docs/',
        'examples/',
        'test/integration/',
        'test/legacy/',
        'test/units/',
    )

    for path in sys.argv[1:] or sys.stdin.read().splitlines():
        if any(path.startswith(p) for p in prune):
            continue

        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

        for text in lines:
            if text in (b'from __future__ import (absolute_import, division, print_function)',
                        b'from __future__ import absolute_import, division, print_function'):
                missing = False
                break

        if missing:
            print('%s: missing: from __future__ import (absolute_import, division, print_function)' % path)


if __name__ == '__main__':
    main()