add custom scaffold support 📂

README.md

@@ -52,3 +52,7 @@ Arguments
 ``` 
 python manage.py gentemplates <your-app-name>
 ```
+
+Want to create own scaffolding templates? 
+1. create folder, copy and customize from [this templates](https://github.com/adiletto64/django-hogwarts/tree/master/hogwarts/scaffold)
+2. add that folder to setting `HOGWARTS_SCAFFOLD_FOLDER = "<your-folder>"`

hogwarts/magic_templates/gen_templates.py

@@ -16,12 +16,15 @@
 
 GENERIC_VIEWS = ("CreateView", "UpdateView", "ListView", "DetailView")
 
-SCAFFOLD_FOLDER = os.path.join(apps.get_app_config("hogwarts").path, "scaffold")
+CUSTOM_SCAFFOLD_FOLDER = getattr(settings, "HOGWARTS_SCAFFOLD_FOLDER", False)
+SCAFFOLD_FOLDER = CUSTOM_SCAFFOLD_FOLDER or os.path.join(apps.get_app_config("hogwarts").path, "scaffold")
+
 TEMPLATES_FOLDER = settings.TEMPLATES[0]["DIRS"][0]
 env = Environment("[#", "#]", "[[", "]]")
 
 console = Console()
 
+
 class ViewType(Enum):
     CREATE = "CreateView"
     UPDATE = "UpdateView"
@@ -41,6 +44,10 @@ class Endpoint:
 def gen_templates(app_name: str):
     endpoints = get_endpoints(app_name)
     console.print(f"generating in templates folder [bold]{TEMPLATES_FOLDER}")
+    if CUSTOM_SCAFFOLD_FOLDER:
+        console.print(f"using custom scaffold {CUSTOM_SCAFFOLD_FOLDER}", style="yellow")
+    else:
+        console.print(f"using default scaffold folder", style="bright_black")
 
     for endpoint in endpoints:
         fields = [field.name for field in endpoint.model._meta.fields]