|
|
@@ -0,0 +1,65 @@
|
|
|
+import sys
|
|
|
+from pathlib import Path
|
|
|
+
|
|
|
+import caseconverter
|
|
|
+import click
|
|
|
+from jinja2 import Environment, FileSystemLoader
|
|
|
+
|
|
|
+
|
|
|
+MAGIC_WORD = "please"
|
|
|
+
|
|
|
+
|
|
|
+@click.command()
|
|
|
+@click.argument("name")
|
|
|
+@click.argument("magic_word")
|
|
|
+@click.option("--debug", is_flag=True, default=False, help="Enable debug mode.")
|
|
|
+def main(name: str, magic_word: str, debug: bool):
|
|
|
+ """
|
|
|
+ This is a script to configure the project.
|
|
|
+
|
|
|
+ It's intended to be called by the `ato create` command.
|
|
|
+ If you're calling it manually, there's likely something
|
|
|
+ wrong and you should probably stop.
|
|
|
+
|
|
|
+ This script is intended to be run in the same environment
|
|
|
+ as the ato CLI, so it's expecting to have access to the
|
|
|
+ same packages and tools; Jinja, etc...
|
|
|
+ """
|
|
|
+ if magic_word != MAGIC_WORD:
|
|
|
+ raise click.BadArgumentUsage(
|
|
|
+ "This script is intended to be called by the ato CLI."
|
|
|
+ )
|
|
|
+
|
|
|
+ # Common variables
|
|
|
+ extended_globals = {
|
|
|
+ "name": name,
|
|
|
+ "caseconverter": caseconverter,
|
|
|
+ "cwd": Path.cwd(),
|
|
|
+ }
|
|
|
+
|
|
|
+ # Load templates
|
|
|
+ env = Environment(loader=FileSystemLoader("."))
|
|
|
+
|
|
|
+ for template_path in Path(".").glob("**/*.j2"):
|
|
|
+ # Figure out the target path and variables and what not
|
|
|
+ target_path = template_path.parent / template_path.name.replace(
|
|
|
+ ".j2", ""
|
|
|
+ ).replace("__name__", caseconverter.kebabcase(name))
|
|
|
+
|
|
|
+ extended_globals["rel_path"] = target_path
|
|
|
+ extended_globals["python_path"] = sys.executable
|
|
|
+
|
|
|
+ template = env.get_template(str(template_path), globals=extended_globals)
|
|
|
+
|
|
|
+ # Make the noise!
|
|
|
+ with target_path.open("w") as f:
|
|
|
+ for chunk in template.generate():
|
|
|
+ f.write(chunk)
|
|
|
+
|
|
|
+ # Remove the template
|
|
|
+ if not debug:
|
|
|
+ template_path.unlink()
|
|
|
+
|
|
|
+
|
|
|
+if __name__ == "__main__":
|
|
|
+ main() # pylint: disable=no-value-for-parameter
|
mawildoer