You've already forked python-uncompyle6
mirror of
https://github.com/rocky/python-uncompyle6.git
synced 2025-08-02 08:34:46 +08:00
119 lines
3.7 KiB
Python
Executable File
119 lines
3.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# emacs-mode: -*-python-*-
|
|
"""
|
|
test_pythonlib -- uncompyle and verify Python libraries
|
|
|
|
Usage-Examples:
|
|
|
|
test_pythonlib --all # decompile all tests (suite + libs)
|
|
test_pythonlib --all --verify # decomyile all tests and verify results
|
|
test_pythonlib --test # decompile only the testsuite
|
|
test_pythonlib --2.2 --verify # decompile and verify python lib 2.2
|
|
|
|
Adding own test-trees:
|
|
|
|
Step 1) Edit this file and add a new entry to 'test_options', eg.
|
|
test_options['mylib'] = ('/usr/lib/mylib', PYOC, 'mylib')
|
|
Step 2: Run the test:
|
|
test_pythonlib --mylib # decompile 'mylib'
|
|
test_pythonlib --mylib --verify # decompile verify 'mylib'
|
|
"""
|
|
|
|
from uncompyle import main, verify
|
|
import os, time, shutil
|
|
from fnmatch import fnmatch
|
|
|
|
#----- configure this for your needs
|
|
|
|
target_base = '/tmp/py-dis/'
|
|
lib_prefix = '/usr/lib'
|
|
|
|
PYC = ('*.pyc', )
|
|
PYO = ('*.pyo', )
|
|
PYOC = ('*.pyc', '*.pyo')
|
|
|
|
test_options = {
|
|
# name: (src_basedir, pattern, output_base_suffix)
|
|
'test': ('./test', PYOC, 'test'),
|
|
'1.5': (os.path.join(lib_prefix, 'python1.5'), PYC, 'python-lib1.5'),
|
|
'1.6': (os.path.join(lib_prefix, 'python1.6'), PYC, 'python-lib1.6'),
|
|
'2.0': (os.path.join(lib_prefix, 'python2.0'), PYC, 'python-lib2.0'),
|
|
'2.1': (os.path.join(lib_prefix, 'python2.1'), PYC, 'python-lib2.1'),
|
|
'2.2': (os.path.join(lib_prefix, 'python2.2'), PYC, 'python-lib2.2'),
|
|
'2.5': (os.path.join(lib_prefix, 'python2.5'), PYC, 'python-lib2.5'),
|
|
'2.6': (os.path.join(lib_prefix, 'python2.6'), PYC, 'python-lib2.6'),
|
|
'2.7': (os.path.join(lib_prefix, 'python2.7'), PYC, 'python-lib2.7')
|
|
}
|
|
|
|
#-----
|
|
|
|
def do_tests(src_dir, patterns, target_dir, start_with=None, do_verify=0):
|
|
|
|
def visitor(files, dirname, names):
|
|
files.extend(
|
|
[os.path.normpath(os.path.join(dirname, n))
|
|
for n in names
|
|
for pat in patterns
|
|
if fnmatch(n, pat)])
|
|
|
|
files = []
|
|
cwd = os.getcwd()
|
|
os.chdir(src_dir)
|
|
os.path.walk(os.curdir, visitor, files)
|
|
os.chdir(cwd)
|
|
files.sort()
|
|
|
|
if start_with:
|
|
try:
|
|
start_with = files.index(start_with)
|
|
files = files[start_with:]
|
|
print '>>> starting with file', files[0]
|
|
except ValueError:
|
|
pass
|
|
|
|
print time.ctime()
|
|
main(src_dir, target_dir, files, [], do_verify=do_verify)
|
|
print time.ctime()
|
|
|
|
if __name__ == '__main__':
|
|
import getopt, sys
|
|
|
|
do_verify = 0
|
|
test_dirs = []
|
|
start_with = None
|
|
|
|
test_options_keys = test_options.keys(); test_options_keys.sort()
|
|
opts, args = getopt.getopt(sys.argv[1:], '',
|
|
['start-with=', 'verify', 'all', ] \
|
|
+ test_options_keys )
|
|
for opt, val in opts:
|
|
if opt == '--verify':
|
|
do_verify = 1
|
|
elif opt == '--start-with':
|
|
start_with = val
|
|
elif opt[2:] in test_options_keys:
|
|
test_dirs.append(test_options[opt[2:]])
|
|
elif opt == '--all':
|
|
for val in test_options_keys:
|
|
test_dirs.append(test_options[val])
|
|
|
|
for src_dir, pattern, target_dir in test_dirs:
|
|
if os.path.exists(src_dir):
|
|
target_dir = os.path.join(target_base, target_dir)
|
|
if os.path.exists(target_dir):
|
|
shutil.rmtree(target_dir, ignore_errors=1)
|
|
do_tests(src_dir, pattern, target_dir, start_with, do_verify)
|
|
else:
|
|
print '### skipping', src_dir
|
|
|
|
# python 1.5:
|
|
|
|
# test/re_tests memory error
|
|
# test/test_b1 memory error
|
|
|
|
# Verification notes:
|
|
# - xdrlib fails verification due the same lambda used twice
|
|
# (verification is successfull when using original .pyo as
|
|
# input)
|
|
#
|