feat(push): #9 - add push and get_repo_ssh_url

Author: leorolland

gitflow_toolbox/get_repo_ssh_url.py

@@ -0,0 +1,20 @@
+import click
+
+from gitflow_toolbox.common.gitlab import CurrentGitlab, RemoteGitlab
+from gitflow_toolbox.common.is_main_call import is_main_call
+
+
+@click.command()
+@click.option("--remote/--current", default=False)
+@click.pass_context
+def get_project_ssh_url(ctx: click.Context, remote: bool):
+    """Get Gitlab project SSH URL (for cloning)
+
+    Args:
+        remote (bool): whether to check on the current gitlab or remote gitlab (True=remote)
+    """
+    project = RemoteGitlab().project if remote else CurrentGitlab().project
+    ssh_url = project.attributes.get("ssh_url_to_repo")
+    if is_main_call(ctx):
+        click.echo(ssh_url)
+    return ssh_url

gitflow_toolbox/push.py

@@ -0,0 +1,46 @@
+import os
+import shutil
+import uuid
+
+import click
+import git
+
+from gitflow_toolbox.get_repo_ssh_url import get_project_ssh_url
+
+
+@click.command()
+@click.option("--from-gitlab", type=click.Choice(["current", "remote"], case_sensitive=False))
+@click.argument("source_branch", type=str)
+@click.option("--to-gitlab", type=click.Choice(["current", "remote"], case_sensitive=False))
+@click.argument("target_branch", type=str)
+@click.option("--force", is_flag=True, type=bool)
+@click.pass_context
+def push(
+    ctx: click.Context,
+    from_gitlab: tuple[str],
+    source_branch: str,
+    to_gitlab: tuple[str],
+    target_branch: str,
+    force: bool,
+):
+
+    project_from_ssh_url = ctx.invoke(get_project_ssh_url, remote=(from_gitlab == "remote"))
+    project_from_clone_dir = os.path.join("/tmp", f"gf_{uuid.uuid4()}")
+
+    # Clone
+    click.echo(f"Cloning {project_from_ssh_url} into {project_from_clone_dir}")
+    repo_from = git.Repo.clone_from(project_from_ssh_url, project_from_clone_dir, branch=source_branch)
+
+    # Add remote
+    project_to_ssh_url = ctx.invoke(get_project_ssh_url, remote=(to_gitlab == "remote"))
+    click.echo(f"Adding remote to {project_to_ssh_url}")
+    repo_from.create_remote("target", project_to_ssh_url)
+
+    # Push
+    click.echo(f"Pushing {from_gitlab} {source_branch} into {to_gitlab} {target_branch} (force={force})")
+    repo_from.remotes.target.push(refspec=f"{source_branch}:{target_branch}", force=force)
+
+    # Clean
+    click.echo(f"Removing cache {project_from_clone_dir}")
+    shutil.rmtree(project_from_clone_dir)
+    click.echo(f"✨ Successfully pushed {from_gitlab} {source_branch} into {to_gitlab} {target_branch}")

main.py

@@ -5,6 +5,8 @@
 from gitflow_toolbox.check_mr_exists import check_mr_exists
 from gitflow_toolbox.ensure_branch import ensure_branch
 from gitflow_toolbox.ensure_mr import ensure_mr
+from gitflow_toolbox.get_repo_ssh_url import get_project_ssh_url
+from gitflow_toolbox.push import push
 
 load_dotenv()
 
@@ -18,6 +20,8 @@ def cli():
 cli.add_command(ensure_branch)
 cli.add_command(check_mr_exists)
 cli.add_command(ensure_mr)
+cli.add_command(get_project_ssh_url)
+cli.add_command(push)
 
 if __name__ == "__main__":
     cli()

poetry.lock

@@ -16,6 +16,7 @@ python = ">=3.9.0,<4"
 click = "^8.1.3"
 python-gitlab = "^3.6.0"
 python-dotenv = "^0.20.0"
+GitPython = "^3.1.27"
 
 [tool.poetry.dev-dependencies]
 bandit = "^1.7"