229 lines
7.6 KiB
Python
229 lines
7.6 KiB
Python
import json
|
|
import os
|
|
import subprocess
|
|
import textwrap
|
|
|
|
from cpl_cli.configuration.venv_helper_service import VenvHelper
|
|
from cpl_cli.migrations.base.migration_service_abc import MigrationServiceABC
|
|
from cpl_core.configuration.configuration_abc import ConfigurationABC
|
|
from cpl_core.console.console import Console
|
|
from cpl_core.console.foreground_color_enum import ForegroundColorEnum
|
|
from cpl_core.environment.application_environment_abc import ApplicationEnvironmentABC
|
|
from cpl_core.utils.pip import Pip
|
|
from cpl_cli.cli_settings import CLISettings
|
|
from cpl_cli.command_abc import CommandABC
|
|
from cpl_cli.configuration import BuildSettings
|
|
from cpl_cli.configuration.project_settings import ProjectSettings
|
|
from cpl_cli.configuration.settings_helper import SettingsHelper
|
|
|
|
|
|
class UpdateService(CommandABC):
|
|
def __init__(
|
|
self,
|
|
config: ConfigurationABC,
|
|
env: ApplicationEnvironmentABC,
|
|
build_settings: BuildSettings,
|
|
project_settings: ProjectSettings,
|
|
cli_settings: CLISettings,
|
|
migrations: MigrationServiceABC,
|
|
):
|
|
"""
|
|
Service for the CLI command update
|
|
:param config:
|
|
:param env:
|
|
:param build_settings:
|
|
:param project_settings:
|
|
:param cli_settings:
|
|
"""
|
|
CommandABC.__init__(self)
|
|
|
|
self._config = config
|
|
self._env = env
|
|
self._build_settings = build_settings
|
|
self._project_settings = project_settings
|
|
self._cli_settings = cli_settings
|
|
self._migrations = migrations
|
|
self._is_simulation = False
|
|
|
|
self._project_file = f"{self._project_settings.name}.json"
|
|
|
|
@property
|
|
def help_message(self) -> str:
|
|
return textwrap.dedent(
|
|
"""\
|
|
Updates the CPL and project dependencies.
|
|
Usage: cpl update
|
|
"""
|
|
)
|
|
|
|
def _collect_project_dependencies(self) -> list[tuple]:
|
|
"""
|
|
Collects project dependencies
|
|
:return:
|
|
"""
|
|
dependencies = []
|
|
for package in [*self._project_settings.dependencies, *self._project_settings.dev_dependencies]:
|
|
name = package
|
|
if "==" in package:
|
|
name = package.split("==")[0]
|
|
elif ">=" in package:
|
|
name = package.split(">=")[0]
|
|
elif "<=" in package:
|
|
name = package.split("<=")[0]
|
|
|
|
dependencies.append((package, name))
|
|
|
|
return dependencies
|
|
|
|
def _update_project_dependencies(self, dependencies):
|
|
"""
|
|
Updates project dependencies
|
|
:return:
|
|
"""
|
|
for package, name in dependencies:
|
|
Pip.install(
|
|
name,
|
|
"--upgrade",
|
|
"--upgrade-strategy",
|
|
"eager",
|
|
source=self._cli_settings.pip_path,
|
|
stdout=subprocess.DEVNULL,
|
|
stderr=subprocess.DEVNULL,
|
|
)
|
|
|
|
new_package = Pip.get_package(name)
|
|
if new_package is None:
|
|
Console.error(f"Update for package {package} failed")
|
|
continue
|
|
|
|
self._project_json_update_dependency(package, new_package)
|
|
|
|
def _check_project_dependencies(self):
|
|
"""
|
|
Checks project dependencies for updates
|
|
:return:
|
|
"""
|
|
dependencies = Console.spinner(
|
|
"Collecting installed dependencies",
|
|
self._collect_project_dependencies,
|
|
text_foreground_color=ForegroundColorEnum.green,
|
|
spinner_foreground_color=ForegroundColorEnum.cyan,
|
|
)
|
|
|
|
Console.spinner(
|
|
"Updating installed dependencies",
|
|
self._update_project_dependencies,
|
|
dependencies,
|
|
text_foreground_color=ForegroundColorEnum.green,
|
|
spinner_foreground_color=ForegroundColorEnum.cyan,
|
|
)
|
|
|
|
if "cpl-cli" in [y for x, y in dependencies]:
|
|
import cpl_cli
|
|
|
|
Console.spinner(
|
|
"Running migrations",
|
|
self._migrations.migrate_from,
|
|
cpl_cli.__version__,
|
|
text_foreground_color=ForegroundColorEnum.green,
|
|
spinner_foreground_color=ForegroundColorEnum.cyan,
|
|
)
|
|
|
|
Console.write_line(f"Found {len(self._project_settings.dependencies)} dependencies.")
|
|
|
|
@staticmethod
|
|
def _check_outdated():
|
|
"""
|
|
Checks for outdated packages in project
|
|
:return:
|
|
"""
|
|
table_str: bytes = Console.spinner(
|
|
"Analyzing for available package updates",
|
|
Pip.get_outdated,
|
|
text_foreground_color=ForegroundColorEnum.green,
|
|
spinner_foreground_color=ForegroundColorEnum.cyan,
|
|
)
|
|
|
|
table = str(table_str, "utf-8").split("\n")
|
|
if len(table) > 1 and table[0] != "":
|
|
Console.write_line("\tAvailable updates for packages:")
|
|
for row in table:
|
|
Console.write_line(f"\t{row}")
|
|
|
|
Console.set_foreground_color(ForegroundColorEnum.yellow)
|
|
Console.write_line(f"\tUpdate with {Pip.get_executable()} -m pip install --upgrade <package>")
|
|
Console.set_foreground_color(ForegroundColorEnum.default)
|
|
|
|
def _save_formatted_package_name_to_deps_collection(self, deps: [str], old_package: str, new_package: str):
|
|
if old_package not in deps:
|
|
return
|
|
|
|
initial_package = new_package
|
|
|
|
if "/" in new_package:
|
|
new_package = new_package.split("/")[0]
|
|
|
|
if "\r" in new_package:
|
|
new_package = new_package.replace("\r", "")
|
|
|
|
if new_package == initial_package:
|
|
return
|
|
|
|
index = self._project_settings.dependencies.index(old_package)
|
|
deps[index] = new_package
|
|
|
|
def _project_json_update_dependency(self, old_package: str, new_package: str):
|
|
"""
|
|
Writes new package version to project.json
|
|
:param old_package:
|
|
:param new_package:
|
|
:return:
|
|
"""
|
|
if self._is_simulation:
|
|
return
|
|
|
|
self._save_formatted_package_name_to_deps_collection(
|
|
self._project_settings.dependencies, old_package, new_package
|
|
)
|
|
self._save_formatted_package_name_to_deps_collection(
|
|
self._project_settings.dev_dependencies, old_package, new_package
|
|
)
|
|
|
|
config = {
|
|
ProjectSettings.__name__: SettingsHelper.get_project_settings_dict(self._project_settings),
|
|
BuildSettings.__name__: SettingsHelper.get_build_settings_dict(self._build_settings),
|
|
}
|
|
|
|
with open(os.path.join(self._env.working_directory, self._project_file), "w") as project:
|
|
project.write(json.dumps(config, indent=2))
|
|
project.close()
|
|
|
|
def execute(self, args: list[str]):
|
|
"""
|
|
Entry point of command
|
|
:param args:
|
|
:return:
|
|
"""
|
|
if "simulate" in args:
|
|
args.remove("simulate")
|
|
Console.write_line("Running in simulation mode:")
|
|
self._is_simulation = True
|
|
|
|
if "cpl-prod" in args:
|
|
args.remove("cpl-prod")
|
|
self._cli_settings.from_dict({"PipPath": "https://pip.sh-edraft.de"})
|
|
|
|
if "cpl-exp" in args:
|
|
args.remove("cpl-exp")
|
|
self._cli_settings.from_dict({"PipPath": "https://pip-exp.sh-edraft.de"})
|
|
|
|
if "cpl-dev" in args:
|
|
args.remove("cpl-dev")
|
|
self._cli_settings.from_dict({"PipPath": "https://pip-dev.sh-edraft.de"})
|
|
|
|
VenvHelper.init_venv(False, self._env, self._project_settings.python_executable)
|
|
|
|
self._check_project_dependencies()
|
|
self._check_outdated()
|
|
Pip.reset_executable()
|