Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix ruff errors

Browse Source
This commit is contained in:
Millun Atluri 2023-12-19 09:58:10 +11:00
parent 16b7246412
commit cd3111c324
1 changed files with 7 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
invokeai/frontend/install/invokeai_update.py
View File

@ -4,11 +4,11 @@ pip install <path_to_git_source>.
""" """
import os import os
import platform import platform
from distutils.version import LooseVersion
import pkg_resources import pkg_resources
import psutil import psutil
import requests import requests
from distutils.version import LooseVersion
from rich import box, print from rich import box, print
from rich.console import Console, group from rich.console import Console, group
from rich.panel import Panel from rich.panel import Panel
@ -48,20 +48,19 @@ def invokeai_is_running() -> bool:
def get_pypi_versions(): def get_pypi_versions():
url = f"https://pypi.org/pypi/invokeai/json" url = "https://pypi.org/pypi/invokeai/json"
try: try:
data = requests.get(url).json() data = requests.get(url).json()
except: except Exception:
raise Exception("Unable to fetch version information from PyPi") raise Exception("Unable to fetch version information from PyPi")
versions = list(data["releases"].keys()) versions = list(data["releases"].keys())
versions.sort(key=LooseVersion, reverse=True) versions.sort(key=LooseVersion, reverse=True)
latest_version = [v for v in versions if 'rc' not in v][0] latest_version = [v for v in versions if "rc" not in v][0]
latest_release_candidate = [v for v in versions if 'rc' in v][0] latest_release_candidate = [v for v in versions if "rc" in v][0]
return latest_version, latest_release_candidate return latest_version, latest_release_candidate
def welcome(latest_release: str, latest_prerelease: str): def welcome(latest_release: str, latest_prerelease: str):
@group() @group()
def text(): def text():
@ -102,11 +101,7 @@ def get_extras():
return extras return extras
def main(): def main():
if invokeai_is_running(): if invokeai_is_running():
print(":exclamation: [bold red]Please terminate all running instances of InvokeAI before updating.[/red bold]") print(":exclamation: [bold red]Please terminate all running instances of InvokeAI before updating.[/red bold]")
input("Press any key to continue...") input("Press any key to continue...")
@ -116,7 +111,6 @@ def main():
welcome(latest_release, latest_prerelease) welcome(latest_release, latest_prerelease)
release = latest_release release = latest_release
choice = Prompt.ask("Choice:", choices=["1", "2", "3"], default="1") choice = Prompt.ask("Choice:", choices=["1", "2", "3"], default="1")
@ -125,14 +119,13 @@ def main():
elif choice == "2": elif choice == "2":
release = latest_prerelease release = latest_prerelease
elif choice == "3": elif choice == "3":
release = Prompt.ask("Enter an InvokeAI version name") release = Prompt.ask("Enter an InvokeAI version name")
extras = get_extras() extras = get_extras()
print(f":crossed_fingers: Upgrading to [yellow]{release}[/yellow]") print(f":crossed_fingers: Upgrading to [yellow]{release}[/yellow]")
cmd = f'pip install "invokeai{extras}=={release}" --use-pep517 --upgrade' cmd = f'pip install "invokeai{extras}=={release}" --use-pep517 --upgrade'
print("") print("")
print("") print("")
if os.system(cmd) == 0: if os.system(cmd) == 0: