mirror of
https://github.com/TECHNOFAB11/bumpver.git
synced 2025-12-12 14:30:09 +01:00
65 lines
1.6 KiB
Python
65 lines
1.6 KiB
Python
|
import shlex
|
||
|
|
import shutil
|
||
|
|
import tempfile
|
||
|
|
import pathlib2 as pl
|
||
|
|
import subprocess as sp
|
||
|
|
|
||
|
|
|
||
|
|
class Shell:
|
||
|
|
def __init__(self, cwd):
|
||
|
|
self.cwd = cwd
|
||
|
|
|
||
|
|
def __call__(self, cmd):
|
||
|
|
if isinstance(cmd, str):
|
||
|
|
cmd_parts = shlex.split(cmd)
|
||
|
|
else:
|
||
|
|
cmd_parts = cmd
|
||
|
|
output = sp.check_output(cmd_parts, cwd=self.cwd)
|
||
|
|
return output.decode("utf-8")
|
||
|
|
|
||
|
|
|
||
|
|
_MODULE_PATH = pl.Path(__file__)
|
||
|
|
FIXTURES_DIR = _MODULE_PATH.parent / "fixtures"
|
||
|
|
|
||
|
|
|
||
|
|
class Project:
|
||
|
|
def __init__(self, project_prefix="a"):
|
||
|
|
if not project_prefix.endswith("_"):
|
||
|
|
project_prefix += "_"
|
||
|
|
|
||
|
|
tmpdir = pl.Path(tempfile.mkdtemp(prefix="pytest_"))
|
||
|
|
self.tmpdir = tmpdir
|
||
|
|
|
||
|
|
_project_dir = tmpdir / "pycalver_project"
|
||
|
|
_project_dir.mkdir()
|
||
|
|
|
||
|
|
for fpath in FIXTURES_DIR.glob(project_prefix + "*"):
|
||
|
|
fname = fpath.name[len(project_prefix) :]
|
||
|
|
shutil.copy(fpath, _project_dir / fname)
|
||
|
|
|
||
|
|
self.dir = _project_dir
|
||
|
|
|
||
|
|
def __enter__(self):
|
||
|
|
return self
|
||
|
|
|
||
|
|
def __exit__(self, *exc):
|
||
|
|
shutil.rmtree(str(self.tmpdir))
|
||
|
|
return False
|
||
|
|
|
||
|
|
def sh(self, cmd):
|
||
|
|
shell = Shell(str(self.dir))
|
||
|
|
return shell(cmd)
|
||
|
|
|
||
|
|
def git_init(self):
|
||
|
|
self.sh("git init")
|
||
|
|
self.sh("git add pycalver.toml")
|
||
|
|
self.sh("git add README.md")
|
||
|
|
self.sh("git commit -m 'initial commit'")
|
||
|
|
|
||
|
|
def hg_init(self):
|
||
|
|
self.sh = Shell(str(self.dir))
|
||
|
|
self.sh("hg init")
|
||
|
|
self.sh("hg add pycalver.toml")
|
||
|
|
self.sh("hg add README.md")
|
||
|
|
self.sh("hg commit -m 'initial commit'")
|