feature: agrego comando branch para listar ramas
This commit is contained in:
@@ -0,0 +1,79 @@
|
||||
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.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. Puede ser vacio para listar ramas de todos los entornos, un entorno válido, o 'trash' para listar las ramas eliminadas.",
|
||||
)
|
||||
|
||||
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 args.environment is None:
|
||||
self.show_all_branches()
|
||||
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()
|
||||
|
||||
for branch in branches:
|
||||
prefix = (COLOR_RED + "*") if branch == self.current_branch else " "
|
||||
print(prefix + " " + branch + COLOR_RESET)
|
||||
|
||||
def get_trash_branches(self):
|
||||
return Git("for-each-ref", "refs/heads/trash/", format=BRANCH_FORMAT).lines()
|
||||
|
||||
def show_all_branches(self):
|
||||
for environment in self.environments:
|
||||
self.show_env_branches(environment)
|
||||
|
||||
trash = len(self.get_trash_branches())
|
||||
print("\nTrash: " + str(trash) + " rama" + ("" if trash == 1 else "s"))
|
||||
|
||||
def show_env_branches(self, environment: str):
|
||||
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()
|
||||
)
|
||||
|
||||
for branch in branches:
|
||||
prefix = (COLOR_GREEN + "*") if branch == self.current_branch else " "
|
||||
print(prefix + " " + branch + COLOR_RESET)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user