ansible/test/sanity/code-smell/no-dict-iterkeys.py

29 lines
716 B
Python
Raw Normal View History

#!/usr/bin/env python
import os
import re
import sys
def main():
skip = set([
'test/sanity/code-smell/%s' % os.path.basename(__file__),
'lib/ansible/module_utils/six/__init__.py',
])
2018-02-27 08:32:19 +01:00
for path in sys.argv[1:] or sys.stdin.read().splitlines():
if path in skip:
continue
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()