diff options
Diffstat (limited to 'baserockimport/exts/python.find_deps')
-rwxr-xr-x | baserockimport/exts/python.find_deps | 45 |
1 files changed, 25 insertions, 20 deletions
diff --git a/baserockimport/exts/python.find_deps b/baserockimport/exts/python.find_deps index b9791ef..a73c457 100755 --- a/baserockimport/exts/python.find_deps +++ b/baserockimport/exts/python.find_deps @@ -189,7 +189,7 @@ def resolve_versions(specsets): logging.debug("Treating %s as %s" % (proj_name, new_proj_name)) proj_name = new_proj_name - releases = client.package_releases(proj_name) + releases = get_releases(client, proj_name) logging.debug('Found %d releases of %s: %s' % (len(releases), proj_name, releases)) @@ -327,31 +327,36 @@ def find_runtime_deps(source, name, version=None, use_requirements_file=False): return runtime_deps -def main(): - if len(sys.argv) not in [3, 4]: - print('usage: %s PACKAGE_SOURCE_DIR NAME [VERSION]' % sys.argv[0]) - sys.exit(1) +class PythonFindDepsExtension(ImportExtension): - logging.debug('%s: sys.argv[1:]: %s' % (sys.argv[0], sys.argv[1:])) - source, name = sys.argv[1:3] - version = sys.argv[3] if len(sys.argv) == 4 else None + def __init__(self): + super(PythonFindDepsExtension, self).__init__() - client = xmlrpclib.ServerProxy(PYPI_URL) - new_name = name_or_closest(client, name) + def run(self): + if len(sys.argv) not in [3, 4]: + print('usage: %s PACKAGE_SOURCE_DIR NAME [VERSION]' % sys.argv[0]) + sys.exit(1) - if new_name == None: - error("Couldn't find any project with name '%s'" % name) + logging.debug('%s: sys.argv[1:]: %s' % (sys.argv[0], sys.argv[1:])) + source, name = sys.argv[1:3] + version = sys.argv[3] if len(sys.argv) == 4 else None - logging.debug('Treating %s as %s' % (name, new_name)) - name = new_name + client = xmlrpclib.ServerProxy(PYPI_URL) + new_name = name_or_closest(client, name) + + if new_name == None: + error("Couldn't find any project with name '%s'" % name) + + logging.debug('Treating %s as %s' % (name, new_name)) + name = new_name - deps = {} - deps['build-dependencies'] = find_build_deps(source, name, version) - deps['runtime-dependencies'] = find_runtime_deps(source, name, version) + deps = {} + deps['build-dependencies'] = find_build_deps(source, name, version) + deps['runtime-dependencies'] = find_runtime_deps(source, name, version) - root = {'python': deps} + root = {'python': deps} - print(json.dumps(root)) + print(json.dumps(root)) if __name__ == '__main__': - PythonExtension().run() + PythonFindDepsExtension().run() |