ansible/test/sanity/code-smell/use-compat-six.py
Matt Clay 3f2b766d10 Add future and metaclass boilerplate to test code.
Continue to ignore:

- test/integration/
- test/legacy/
- test/units/
2019-07-12 09:18:24 -07:00

35 lines
1.1 KiB
Python
Executable file

#!/usr/bin/env python
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type
import os
import re
import sys
def main():
skip = set([
'test/sanity/code-smell/%s' % os.path.basename(__file__),
# digital_ocean is checking for six because dopy doesn't specify the
# requirement on six so it needs to try importing six to give the correct error message
'lib/ansible/modules/cloud/digital_ocean/digital_ocean.py',
])
for path in sys.argv[1:] or sys.stdin.read().splitlines():
if path in skip:
continue
if path.startswith('test/sanity/pylint/plugins/'):
continue
with open(path, 'r') as path_fd:
for line, text in enumerate(path_fd.readlines()):
match = re.search(r'((^\s*import\s+six\b)|(^\s*from\s+six\b))', text)
if match:
print('%s:%d:%d: use `ansible.module_utils.six` instead of `six`' % (
path, line + 1, match.start(1) + 1))
if __name__ == '__main__':
main()