68 lines
2.1 KiB
Python
Executable File
68 lines
2.1 KiB
Python
Executable File
#!/usr/bin/python2
|
|
# python-deps - find the dependencies of a given python script.
|
|
|
|
import copy
|
|
import os, sys
|
|
from modulefinder import ModuleFinder
|
|
# pylint: disable=wildcard-import
|
|
from distutils.sysconfig import *
|
|
|
|
sitedir = get_python_lib()
|
|
libdir = get_config_var('LIBDEST')
|
|
|
|
alsoNeeded = { libdir+"/urllib.py": libdir+"/urllib2.py" }
|
|
|
|
# A couple helper functions...
|
|
def moduledir(pyfile):
|
|
'''Given a python file, return the module dir it belongs to, or None.'''
|
|
for topdir in sitedir, libdir:
|
|
relpath = os.path.relpath(pyfile, topdir)
|
|
if '/' not in relpath: continue
|
|
modname = relpath.split('/')[0]
|
|
if modname not in ('..', 'site-packages'):
|
|
return os.path.join(topdir, modname)
|
|
|
|
# pylint: disable=redefined-outer-name
|
|
def pyfiles(moddir):
|
|
'''basically, "find $moddir -type f -name "*.py"'''
|
|
for curdir, _dirs, files in os.walk(moddir):
|
|
for f in files:
|
|
if f.endswith(".py"):
|
|
yield os.path.join(curdir, f)
|
|
|
|
# OK. Use modulefinder to find all the modules etc. this script uses!
|
|
mods = []
|
|
deps = []
|
|
|
|
scripts = copy.copy(sys.argv[1:])
|
|
|
|
while scripts:
|
|
script = scripts.pop()
|
|
|
|
finder = ModuleFinder()
|
|
finder.run_script(script) # parse the script
|
|
for mod in finder.modules.itervalues():
|
|
if not mod.__file__: # this module is builtin, so we can skip it
|
|
continue
|
|
|
|
if mod.__file__ not in deps: # grab the file itself
|
|
deps.append(mod.__file__)
|
|
|
|
moddir = moduledir(mod.__file__) # if it's part of a module...
|
|
if moddir and moddir not in mods: #
|
|
deps += list(pyfiles(moddir)) # ...get the whole module
|
|
mods.append(moddir)
|
|
|
|
if mod.__file__ in alsoNeeded and alsoNeeded[mod.__file__] not in deps:
|
|
scripts.append(alsoNeeded[mod.__file__])
|
|
|
|
# Include some bits that the python install itself needs
|
|
print(get_makefile_filename())
|
|
print(get_config_h_filename())
|
|
print(os.path.join(libdir,'site.py'))
|
|
print(os.path.join(libdir,'sysconfig.py'))
|
|
|
|
# And print the list of deps.
|
|
for d in deps:
|
|
print(d)
|