118 lines
4.2 KiB
Python
118 lines
4.2 KiB
Python
from argparse import ArgumentParser, Namespace
|
|
from git_flow import COLOR_WHITE_BOLD, COLOR_GREEN, COLOR_RED, COLOR_RESET, TRASH_BRANCH_PREFIX, WIP_BRANCH_PREFIX, GitFlowError
|
|
from git_flow.command.base import Command
|
|
from git_flow.git import Git
|
|
|
|
|
|
BRANCH_FORMAT = "%(refname:short)"
|
|
|
|
|
|
class BranchCommand(Command):
|
|
def name(self) -> str:
|
|
return "branch"
|
|
|
|
def description(self) -> str:
|
|
return """Lista ramas del repositorio, agrupandolas por entorno objetivo"""
|
|
|
|
def setup_parser(self, parser: ArgumentParser) -> ArgumentParser:
|
|
parser.add_argument(
|
|
"environment",
|
|
nargs="?",
|
|
help="Entorno de las ramas a listar. Por defecto, es el entorno actual.",
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--trash",
|
|
action="store_true",
|
|
help=f"Listar ramas en la 'papelera de reciclaje' (ramas que empiezan con '{TRASH_BRANCH_PREFIX}')"
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--wip",
|
|
action="store_true",
|
|
help=f"Listar ramas 'en progreso' (ramas que empiezan con '{WIP_BRANCH_PREFIX}')"
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--all",
|
|
action="store_true",
|
|
help=f"Listar ramas de todos los entornos"
|
|
)
|
|
|
|
return parser
|
|
|
|
def run(self, args: Namespace = Namespace()):
|
|
self.ensure_initialized()
|
|
|
|
self.current_branch = Git.get_current_branch()
|
|
self.environments = self.flowconfig["flow.branches"].split(",")
|
|
|
|
if self.current_branch in self.environments:
|
|
current_environment = self.current_branch
|
|
else:
|
|
(current_environment, _) = self.get_branch_env_and_type(self.current_branch)
|
|
|
|
if args.trash:
|
|
self.show_branches(TRASH_BRANCH_PREFIX)
|
|
elif args.wip:
|
|
self.show_branches(WIP_BRANCH_PREFIX)
|
|
elif args.all:
|
|
self.show_envs()
|
|
self.show_all_branches()
|
|
elif args.environment is None:
|
|
self.show_envs()
|
|
self.show_env_branches(current_environment)
|
|
elif args.environment in self.environments:
|
|
self.show_envs()
|
|
self.show_env_branches(args.environment)
|
|
else:
|
|
raise GitFlowError(f"'{args.environment}' no es un entorno válido.")
|
|
|
|
def show_branches(self, prefix: str):
|
|
branches = self.get_branches(prefix)
|
|
|
|
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)
|
|
|
|
def get_branches(self, prefix: str):
|
|
return Git("for-each-ref", "refs/heads/" + prefix, format=BRANCH_FORMAT).lines()
|
|
|
|
def show_all_branches(self):
|
|
for environment in self.environments:
|
|
self.show_env_branches(environment, len(self.environments) > 1)
|
|
|
|
trash = len(self.get_branches(TRASH_BRANCH_PREFIX))
|
|
|
|
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]
|
|
pattern = "refs/heads/" if is_first_env else "refs/heads/release/" + environment + "/"
|
|
pattern += "*/*"
|
|
exclude = ["refs/heads/" + TRASH_BRANCH_PREFIX, "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)
|