python/bin/githelp.py at patch-1 · ioload/python · GitHub

Repository navigation

from contextlib import contextmanager

from enum import Enum

from pathlib import Path

import shutil

import subprocess

from typing import Iterator, Union

GITHUB_EXERCISM = f"https://github.com/exercism"

class Repo(Enum):

ProblemSpecifications = f"{GITHUB_EXERCISM}/problem-specifications.git"

def clone(repo: Union[str, Repo], directory: Union[str, Path, None] = None) -> bool:

if isinstance(repo, Repo):

repo = repo.value

if directory is None:

directory = repo.split("/")[-1].split(".")[0]

directory = Path(directory)

if not directory.is_dir():

try:

subprocess.run(["git", "clone", repo, str(directory)], check=True)

return True

except subprocess.CalledProcessError:

pass

return False

@contextmanager

def clone_if_missing(repo: Union[str, Repo], directory: Union[str, Path, None] = None) -> Iterator[None]:

temp_clone = clone(repo, directory)

try:

yield directory

finally:

if temp_clone:

shutil.rmtree(directory)