* Fix HTTP error handling

This commit is contained in:
j3rome 2021-06-28 12:39:03 -04:00
parent f94c4a2d7c
commit 4253e368fc
1 changed files with 6 additions and 7 deletions

13
main.py
View File

@ -38,13 +38,12 @@ def get_pypi_history(package_name, ignore_release_candidat=True):
"""
try:
resp = urlopen(f"https://pypi.org/pypi/{package_name}/json")
except:
print("[ERROR] Internet access is required to fetch package history from Pypi")
exit(1)
if resp.code != 200:
print(f"[INFO] Couldn't find package '{package_name} on Pypi. Ignoring")
return None
except Exception as e:
if hasattr(e, 'getcode') and e.getcode() == 404:
return None
else:
print("[ERROR] Internet access is required to fetch package history from Pypi")
exit(1)
resp = json.loads(resp.read())