93 lines
2.5 KiB
Python
93 lines
2.5 KiB
Python
|
#!/usr/bin/env python3
|
||
|
from typing import Any, Literal, Optional
|
||
|
from tomllib import load
|
||
|
from pathlib import Path
|
||
|
from dataclasses import dataclass, field
|
||
|
import questionary
|
||
|
|
||
|
TEMPLATES_DIR = Path("./_templates")
|
||
|
|
||
|
@dataclass(frozen=True)
|
||
|
class TemplateParameter:
|
||
|
prompt: str
|
||
|
default: Optional[Any] = None
|
||
|
|
||
|
@dataclass
|
||
|
class ProjectTemplate:
|
||
|
name: str
|
||
|
path: Path
|
||
|
templates: list[str] = field(default_factory=list)
|
||
|
parameters: dict[str, TemplateParameter] = field(default_factory=dict)
|
||
|
|
||
|
|
||
|
|
||
|
templates: dict[str, ProjectTemplate] = {}
|
||
|
|
||
|
|
||
|
for v in TEMPLATES_DIR.rglob("template.toml"):
|
||
|
with v.open("rb") as fp:
|
||
|
data = load(fp)
|
||
|
tpl = ProjectTemplate(
|
||
|
data.get('name', str(v.parent.relative_to(TEMPLATES_DIR))),
|
||
|
v.parent,
|
||
|
data.get('templates', []),
|
||
|
{
|
||
|
k: TemplateParameter(**v)
|
||
|
for k, v in data.get('params', {}).items()
|
||
|
}
|
||
|
)
|
||
|
templates[tpl.name] = tpl
|
||
|
|
||
|
|
||
|
name: Optional[str] = questionary.text("Project name").ask()
|
||
|
if not name:
|
||
|
exit(1)
|
||
|
|
||
|
template: Optional[ProjectTemplate] = questionary.select("Template", [
|
||
|
questionary.Choice(tpl.name, tpl) for tpl in templates.values()
|
||
|
]).ask()
|
||
|
if not template:
|
||
|
exit(1)
|
||
|
|
||
|
parameters = questionary.prompt([
|
||
|
{
|
||
|
"type": "text",
|
||
|
"name": k,
|
||
|
"message": param.prompt,
|
||
|
"default": param.default
|
||
|
}
|
||
|
for k, param in template.parameters.items()
|
||
|
])
|
||
|
if not parameters:
|
||
|
exit(1)
|
||
|
|
||
|
project_dir = Path(name)
|
||
|
project_dir.mkdir(parents=True, exist_ok=True)
|
||
|
|
||
|
for base, dirs, files in template.path.walk():
|
||
|
for dirname in dirs:
|
||
|
(base / dirname).mkdir(exist_ok=True, parents=True)
|
||
|
|
||
|
for filename in files:
|
||
|
in_path = base / filename
|
||
|
name = str(in_path.relative_to(template.path))
|
||
|
out_path = project_dir / name
|
||
|
if name == "template.toml":
|
||
|
continue
|
||
|
elif name in template.templates:
|
||
|
print("TMPL", out_path)
|
||
|
with in_path.open("r") as fp_in:
|
||
|
with out_path.open("w") as fp_out:
|
||
|
for line in fp_in:
|
||
|
line = line.replace("__PROGNAME", name)
|
||
|
for param in template.parameters:
|
||
|
line = line.replace(f"__{param}", parameters[param])
|
||
|
fp_out.write(line)
|
||
|
else:
|
||
|
print("COPY", out_path)
|
||
|
with in_path.open("rb") as fp_in:
|
||
|
with out_path.open("wb") as fp_out:
|
||
|
while (chunk := fp_in.read(32768)):
|
||
|
fp_out.write(chunk)
|
||
|
|