fix: build version notification function (#3527)

fix: added try/except block to version notification function
This commit is contained in:
Ítalo Johnny 2024-08-23 09:15:54 -03:00 • committed by GitHub
commit 8e733ce580
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -318,9 +318,16 @@ def is_prerelease(version: str) -> bool:
def fetch_latest_version(package_name: str, include_prerelease: bool) -> Optional[str]: def fetch_latest_version(package_name: str, include_prerelease: bool) -> Optional[str]:
valid_versions = []
try:
response = httpx.get(f"https://pypi.org/pypi/{package_name}/json") response = httpx.get(f"https://pypi.org/pypi/{package_name}/json")
versions = response.json()["releases"].keys() versions = response.json()["releases"].keys()
valid_versions = [v for v in versions if include_prerelease or not is_prerelease(v)] valid_versions = [v for v in versions if include_prerelease or not is_prerelease(v)]
except Exception as e:
logger.exception(e)
finally:
if not valid_versions: if not valid_versions:
return None # Handle case where no valid versions are found return None # Handle case where no valid versions are found
return max(valid_versions, key=lambda v: pkg_version.parse(v)) return max(valid_versions, key=lambda v: pkg_version.parse(v))