diff options
author | Richard Ipsum <richard.ipsum@codethink.co.uk> | 2014-10-09 18:27:58 +0100 |
---|---|---|
committer | Richard Ipsum <richard.ipsum@codethink.co.uk> | 2014-11-04 12:40:55 +0000 |
commit | 516b7ae1a334c4b7175bb59eaab80f2b66d749eb (patch) | |
tree | 6e245a2c3930d8c4f3a641fa867459ad578ca299 | |
parent | 3b8a9bade82df0d81c632e0a020e57fb042b81b4 (diff) | |
download | import-516b7ae1a334c4b7175bb59eaab80f2b66d749eb.tar.gz |
Only run our stuff if we're called as main
-rwxr-xr-x | import/pip_lorry.py (renamed from import/pip.to_lorry) | 18 |
1 files changed, 10 insertions, 8 deletions
diff --git a/import/pip.to_lorry b/import/pip_lorry.py index f5c9335..580e94a 100755 --- a/import/pip.to_lorry +++ b/import/pip_lorry.py @@ -159,14 +159,16 @@ def str_repo_lorry(package_name, repo_type, url): return json.dumps({package_name.lower(): {'type': repo_type, 'url': url}}, indent=4, sort_keys=True) -if len(sys.argv) != 2: - error('usage: %s python_package' % sys.argv[0]) +if __name__ == '__main__': + if len(sys.argv) != 2: + error('usage: %s python_package' % sys.argv[0]) -package_name = sys.argv[1] -metadata = fetch_package_metadata(package_name) -info = metadata['info'] + package_name = sys.argv[1] + metadata = fetch_package_metadata(package_name) + info = metadata['info'] -repo_type = find_repo_type(info['home_page']) if 'home_page' in info else None + repo_type = (find_repo_type(info['home_page']) + if 'home_page' in info else None) -print(str_repo_lorry(package_name, repo_type, info['home_page']) - if repo_type else generate_tarball_lorry(package_name)) + print(str_repo_lorry(package_name, repo_type, info['home_page']) + if repo_type else generate_tarball_lorry(package_name)) |