summaryrefslogtreecommitdiff
path: root/test/lib/ansible_test/_util/controller/sanity/code-smell/no-dict-iterkeys.py
blob: 237ee5b1c15752e66f8fcfe8c83d966092b8f807 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#!/usr/bin/env python
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type

import re
import sys


def main():
    for path in sys.argv[1:] or sys.stdin.read().splitlines():
        with open(path, 'r') as path_fd:
            for line, text in enumerate(path_fd.readlines()):
                match = re.search(r'\.(iterkeys)', text)

                if match:
                    print('%s:%d:%d: use `dict.keys` or `for key in dict:` instead of `dict.iterkeys`' % (
                        path, line + 1, match.start(1) + 1))


if __name__ == '__main__':
    main()