40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
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)
|