mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
28 lines
859 B
Python
28 lines
859 B
Python
|
import requests as request
|
||
|
|
|
||
|
|
import ldm.invoke._version as version
|
||
|
|
|
||
|
|
local_version = str(version.__version__)
|
||
|
|
|
||
|
|
|
||
|
|
def get_pypi_versions(package_name="InvokeAI") -> list[str]:
|
||
|
|
"""Get the versions of the package from PyPI"""
|
||
|
|
url = f"https://pypi.org/pypi/{package_name}/json"
|
||
|
|
response = request.get(url).json()
|
||
|
|
versions: list[str] = list(response["releases"].keys())
|
||
|
|
return versions
|
||
|
|
|
||
|
|
|
||
|
|
def local_on_pypi(package_name="InvokeAI", local_version=local_version) -> bool:
|
||
|
|
"""Compare the versions of the package from PyPI and the local package"""
|
||
|
|
pypi_versions = get_pypi_versions(package_name)
|
||
|
|
return local_version in pypi_versions
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
package_name = "InvokeAI"
|
||
|
|
if local_on_pypi():
|
||
|
|
print(f"Package {package_name} is up to date")
|
||
|
|
else:
|
||
|
|
print(f"Package {package_name} is not up to date")
|