ansible/test/sanity/code-smell/no-main-display.py

#!/usr/bin/env python
from __future__ import print_function

import os
import re
import sys

MAIN_DISPLAY_IMPORT = 'from __main__ import display'


def main():
    skip = set()

    for path in sys.argv[1:] or sys.stdin.read().splitlines():
        if path in skip:
            continue

        with open(path, 'r') as f:
            for i, line in enumerate(f.readlines()):
                if MAIN_DISPLAY_IMPORT in line:
                    lineno = i + 1
                    colno = line.index(MAIN_DISPLAY_IMPORT) + 1
                    print('%s:%d:%d: Display is a singleton, just import and instantiate' % (path, lineno, colno))


if __name__ == '__main__':
    main()