Fix branch for installed packages (#133)

Author: juliankuners

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}[/]."
     )

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,
         )