mirror of
https://github.com/valitydev/salt.git
synced 2024-11-09 01:36:48 +00:00
002aa88a97
Without allow_unicode=True, unicode characters are processed through the str representer and on Python 2 are dumped as a Unicode code point (i.e. a literal \u0414). This commit makes allow_unicode=True the default in our salt.utils.yamlloader.safe_dump() helper. It also adds a new salt.utils.yamlloader.dump() helper which wraps yaml.dump() and also makes allow_unicode=True the default. To make importing and using our custom yaml loader/dumper easier, a convenience module called salt.utils.yaml has been added, which does a wildcard import from both salt.utils.yamldumper and salt.utils.yamlloader. Refs to yaml.load/dump and yaml.safe_load/safe_dump have been updated to salt.utils.yaml, to ensure that unicode is handled properly.
109 lines
2.7 KiB
Python
109 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
#!/usr/bin/env python2
|
|
|
|
from __future__ import absolute_import, print_function
|
|
try:
|
|
import argparse # pylint: disable=minimum-python-version
|
|
HAS_ARGPARSE = True
|
|
except ImportError:
|
|
HAS_ARGPARSE = False
|
|
import os
|
|
import sys
|
|
import os
|
|
import modulefinder
|
|
import pprint
|
|
import salt.utils.json
|
|
import salt.utils.yaml
|
|
|
|
|
|
def parse():
|
|
'''
|
|
Parse the arguments
|
|
'''
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
'-r',
|
|
'--root',
|
|
dest='root',
|
|
default='.',
|
|
help='The base code directory to look in')
|
|
parser.add_argument(
|
|
'-i',
|
|
'--bif',
|
|
dest='bif',
|
|
default='site-packages')
|
|
parser.add_argument(
|
|
'-f',
|
|
'--format',
|
|
dest='format',
|
|
choices=('pprint', 'yaml'),
|
|
default='pprint')
|
|
out = parser.parse_args()
|
|
return out.__dict__
|
|
|
|
|
|
def mod_data(opts, full):
|
|
'''
|
|
Grab the module's data
|
|
'''
|
|
ret = {}
|
|
finder = modulefinder.ModuleFinder()
|
|
try:
|
|
finder.load_file(full)
|
|
except ImportError as exc:
|
|
print('ImportError - {0} (Reason: {1})'.format(full, exc), file=sys.stderr)
|
|
return ret
|
|
for name, mod in finder.modules.items():
|
|
basemod = name.split('.')[0]
|
|
if basemod in ret:
|
|
continue
|
|
if basemod.startswith('_'):
|
|
continue
|
|
if not mod.__file__:
|
|
continue
|
|
if opts['bif'] not in mod.__file__:
|
|
# Bif - skip
|
|
continue
|
|
if name == os.path.basename(mod.__file__)[:-3]:
|
|
continue
|
|
ret[basemod] = mod.__file__
|
|
for name, err in finder.badmodules.items():
|
|
basemod = name.split('.')[0]
|
|
if basemod in ret:
|
|
continue
|
|
if basemod.startswith('_'):
|
|
continue
|
|
ret[basemod] = err
|
|
return ret
|
|
|
|
|
|
def scan(opts):
|
|
'''
|
|
Scan the provided root for python source files
|
|
'''
|
|
ret = {}
|
|
for root, dirs, files in os.walk(opts['root']):
|
|
for fn_ in files:
|
|
full = os.path.join(root, fn_)
|
|
if full.endswith('.py'):
|
|
ret.update(mod_data(opts, full))
|
|
return ret
|
|
|
|
|
|
if __name__ == '__main__':
|
|
if not HAS_ARGPARSE:
|
|
print('The argparse python module is required')
|
|
opts = parse()
|
|
try:
|
|
scand = scan(opts)
|
|
if opts['format'] == 'yaml':
|
|
print(salt.utils.yaml.safe_dump(scand))
|
|
if opts['format'] == 'json':
|
|
print(salt.utils.json.dumps(scand))
|
|
else:
|
|
pprint.pprint(scand)
|
|
exit(0)
|
|
except KeyboardInterrupt:
|
|
print('\nInterrupted on user request', file=sys.stderr)
|
|
exit(1)
|