diff --git a/src/kup/__main__.py b/src/kup/__main__.py index ddbfb26..2866994 100644 --- a/src/kup/__main__.py +++ b/src/kup/__main__.py @@ -516,7 +516,15 @@ def install_package( ) verb = 'updated' if package_name.base in installed_packages else 'installed' - display_version = f' ({package_version})' if package_version else ' (master)' + + if package_version is not None: + display_version = package_version + elif package.branch is not None: + display_version = package.branch + else: + display_version = None + display_version = f' ({display_version})' if display_version is not None else '' + rich.print( f" ✅ Successfully {verb} '[green]{package_name.base}[/]' version [blue]{package.uri}{display_version}[/]." ) diff --git a/src/kup/package.py b/src/kup/package.py index 7222a77..1783886 100644 --- a/src/kup/package.py +++ b/src/kup/package.py @@ -246,12 +246,13 @@ def __init__( access_token: Optional[str] = None, substituters: Optional[list[str]] = None, public_keys: Optional[list[str]] = None, + update_branch: Optional[str] = None, ): self.commit = commit self.status = status self.index = index self.tag = tag - super().__init__(org, repo, package, None, ssh_git, access_token, substituters, public_keys) + super().__init__(org, repo, package, update_branch, ssh_git, access_token, substituters, public_keys) @property def concrete_repo_path_with_access(self) -> Tuple[str, List[str]]: @@ -321,6 +322,7 @@ def parse(url: str, package: GithubPackage, idx: Union[int, str], load_versions: tag, idx, package.ssh_git, + update_branch=package.branch, )