Hazel/python-mommy-venv
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

cli script to recompile config

Browse Source
This commit is contained in:
Hazel Noack 2025-07-29 12:46:56 +02:00
parent 05f7003ea8
commit d968795628
2 changed files with 41 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pyproject.toml
View File

@ -14,6 +14,7 @@ version = "0.0.0"
license-files = ["LICENSE"] license-files = ["LICENSE"]
[project.scripts] [project.scripts]
mommify-venv-compile = "python_mommy_venv.__main__:cli_compile_config"
mommify-venv = "python_mommy_venv.__main__:mommify_venv" mommify-venv = "python_mommy_venv.__main__:mommify_venv"
[build-system] [build-system]

53
python_mommy_venv/__main__.py
View File

@ -85,11 +85,27 @@ PIP_HOOK = """# GENERATED BY MOMMY
sys.exit(code)""" sys.exit(code)"""
def assert_venv(): def assert_venv(only_warn: bool = False):
if not IS_VENV: if not IS_VENV:
mommy_logger.error("mommy doesn't run in a virtual environment~") mommy_logger.error("mommy doesn't run in a virtual environment~")
serious_logger.error("this has to run in a virtual environment") serious_logger.error("this should run in a virtual environment")
exit(1) if not only_warn:
exit(1)
def write_compile_config(local: bool):
assert_venv(only_warn=not local)
compiled_base_dir = VENV_DIRECTORY if local else CONFIG_DIRECTORY
compiled_config_file = compiled_base_dir / COMPILED_CONFIG_FILE_NAME
compiled = compile_config()
mommy_logger.info("mommy writes its moods in %s", compiled_config_file)
serious_logger.info("writing compiled config file to %s", compiled_config_file)
compiled_base_dir.mkdir(parents=True, exist_ok=True)
with compiled_config_file.open("w") as f:
json.dump(compiled, f, indent=4)
if not local:
(VENV_DIRECTORY / COMPILED_CONFIG_FILE_NAME).unlink(missing_ok=True)
def wrap_interpreter(path: Path): def wrap_interpreter(path: Path):
@ -138,6 +154,26 @@ def install_pip_hook(path: Path):
f.write(text) f.write(text)
def cli_compile_config():
parser = argparse.ArgumentParser(description="only recompile the config")
parser.add_argument(
"-v", "--verbose",
action="store_true",
help="enable verbose and serious output"
)
parser.add_argument(
"-l", "--local",
action="store_true",
help="compile the config only for the current virtual environment"
)
args = parser.parse_args()
write_compile_config(args.local)
def mommify_venv(): def mommify_venv():
parser = argparse.ArgumentParser(description="patch the virtual environment to use mommy") parser = argparse.ArgumentParser(description="patch the virtual environment to use mommy")
@ -158,16 +194,7 @@ def mommify_venv():
config_logging(args.verbose) config_logging(args.verbose)
assert_venv() assert_venv()
compiled_base_dir = VENV_DIRECTORY if args.local else CONFIG_DIRECTORY write_compile_config(args.local)
compiled_config_file = compiled_base_dir / COMPILED_CONFIG_FILE_NAME
compiled = compile_config()
mommy_logger.info("mommy writes its moods in %s", compiled_config_file)
serious_logger.info("writing compiled config file to %s", compiled_config_file)
compiled_base_dir.mkdir(parents=True, exist_ok=True)
with compiled_config_file.open("w") as f:
json.dump(compiled, f, indent=4)
if not args.local:
(VENV_DIRECTORY / COMPILED_CONFIG_FILE_NAME).unlink(missing_ok=True)
mommy_logger.info("") mommy_logger.info("")