Some more functionality to the git and gitlab submodules. (a8ceb1fe) · Commits · nimrod / rcfiles

Documents/bin/rcfiles/git.py

+71 −0

Original line number	Diff line number	Diff line
		"""Git repository related functions."""

		import configparser
		import os.path
		import pathlib
		from sh.contrib import git # pylint: disable=import-error
		from . import gitlab


		def is_repo(path):
		@@ -18,3 +21,71 @@ def in_repo():
		it's enough to just check if the .git directory exists where we are.
		"""
		return is_repo(".")


		def get_remotes():
		"""Return a dictionary of remotes and their URL.
		Also, deduce the remote type ("gitlab", "github" or None if couldn't figure
		it out) and get the namespace and repository name.

		If not in a Git repository, return None.
		"""
		if not in_repo:
		return None

		gitlab_http_url = gitlab.get_url()
		gitlab_ssh_url = (
		f'git@{gitlab_http_url.removeprefix("https://").removesuffix("/")}:'
		)
		github_http_url = "https://github.com/"
		github_ssh_url = "git@github.com:"

		config = configparser.ConfigParser()
		config.read(".git/config")

		remotes = {
		x.removeprefix('remote "').removesuffix('"'): {
		"url": config[x]["url"],
		"name": x.removeprefix('remote "').removesuffix('"'),
		}
		for x in config.sections()
		if x.startswith("remote ")
		}

		for name, remote in remotes.items():
		if remote["url"].startswith(gitlab_http_url) or remote[
		"url"
		].startswith(gitlab_ssh_url):
		remotes[name]["type"] = "gitlab"
		parts = (
		remote["url"]
		.removeprefix(gitlab_http_url)
		.removeprefix(gitlab_ssh_url)
		.removesuffix(".git")
		.split("/")
		)
		if len(parts) == 2:
		remotes[name]["namespace"] = parts[0]
		remotes[name]["name"] = parts[1]
		elif remote["url"].startswith(github_http_url) or remote[
		"url"
		].startswith(github_ssh_url):
		remotes[name]["type"] = "github"
		parts = (
		remote["url"]
		.removeprefix(github_http_url)
		.removeprefix(github_ssh_url)
		.removesuffix(".git")
		.split("/")
		)
		if len(parts) == 2:
		remotes[name]["name"] = parts[1]
		else:
		remotes[remote]["type"] = None

		return remotes


		def add_remote(name, url):
		"""Add a remote to the Git repository."""
		git.remote("add", name, url)

Documents/bin/rcfiles/gitlab.py

+6 −0

Original line number	Diff line number	Diff line
		@@ -77,3 +77,9 @@ def create_project(conn, name, group=None, description=None, visibility=None):
		if visibility is not None:
		data["visibility"] = visibility
		return conn.projects.create(data)


		def read_only_project(conn, group, name):
		"""Make a GitLab project read-only."""
		project = get_project(conn, group, name)
		project.archive()