feature: modifico formato de listado de ramas

This commit is contained in:
jt
2025-11-26 22:51:58 -03:00
parent 542c629489
commit 717708469a
2 changed files with 33 additions and 22 deletions
+32 -22
View File
@@ -1,6 +1,6 @@
from argparse import ArgumentParser, Namespace
from git_flow import GitFlowError
from git_flow.command.base import COLOR_GREEN, COLOR_RED, COLOR_RESET, Command
from git_flow.command.base import COLOR_GREEN, COLOR_RED, COLOR_RESET, COLOR_WHITE_BOLD, Command
from git_flow.git import Git
@@ -30,17 +30,23 @@ class BranchCommand(Command):
self.environments = self.flowconfig["flow.branches"].split(",")
if args.environment is None:
self.show_envs()
self.show_all_branches()
elif args.environment in self.environments:
self.show_envs()
self.show_env_branches(args.environment)
elif args.environment == "trash":
self.show_trash_branches()
elif args.environment in self.environments:
self.show_env_branches(args.environment)
else:
raise GitFlowError(f"'{args.environment}' no es un entorno válido.")
def show_trash_branches(self):
branches = self.get_trash_branches()
if not branches:
print("No hay ramas a mostrar.")
return
for branch in branches:
prefix = (COLOR_RED + "*") if branch == self.current_branch else " "
print(prefix + " " + branch + COLOR_RESET)
@@ -50,30 +56,34 @@ class BranchCommand(Command):
def show_all_branches(self):
for environment in self.environments:
self.show_env_branches(environment)
self.show_env_branches(environment, len(self.environments) > 1)
trash = len(self.get_trash_branches())
print("\nTrash: " + str(trash) + " rama" + ("" if trash == 1 else "s"))
def show_env_branches(self, environment: str):
if trash:
print()
print(COLOR_WHITE_BOLD + "Papelera: " + COLOR_RESET + str(trash) + " rama" + ("" if trash == 1 else "s"))
def show_envs(self):
print(COLOR_WHITE_BOLD + "Entornos" + COLOR_RESET)
for environment in self.environments:
prefix = (COLOR_GREEN + "*") if environment == self.current_branch else " "
print(prefix + " " + environment + COLOR_RESET)
print()
def show_env_branches(self, environment: str, show_env: bool = False):
is_first_env = environment == self.environments[0]
branches = (
Git(
"for-each-ref",
"refs/heads/",
exclude=["refs/heads/trash/", "refs/heads/release/"],
format=BRANCH_FORMAT,
).lines()
if is_first_env
else Git(
"for-each-ref",
"refs/heads/release/" + environment + "/",
format=BRANCH_FORMAT,
).lines()
)
pattern = "refs/heads/" if is_first_env else "refs/heads/release/" + environment + "/"
pattern += "*/*"
exclude = ["refs/heads/trash/", "refs/heads/release/"] if is_first_env else []
branches = Git("for-each-ref", pattern, exclude=exclude, format=BRANCH_FORMAT).lines()
print(COLOR_WHITE_BOLD + (environment if show_env else "Ramas") + COLOR_RESET)
if not branches:
print("No hay ramas a mostrar.")
return
for branch in branches:
prefix = (COLOR_GREEN + "*") if branch == self.current_branch else " "
print(prefix + " " + branch + COLOR_RESET)