add gentemplate logs

adiletto64 · Aug 21, 2023 · 17033ff · 17033ff
1 parent ce50a58
commit 17033ff
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 67 deletions.
diff --git a/hogwarts/magic_templates/gen_templates.py b/hogwarts/magic_templates/gen_templates.py
@@ -3,6 +3,7 @@
 from enum import Enum
 from typing import Optional, Type, Union
 
+from rich.console import Console
 from jinja2 import Environment
 from django.views.generic import DetailView, CreateView
 from django.conf import settings
@@ -19,6 +20,7 @@
 TEMPLATES_FOLDER = settings.TEMPLATES[0]["DIRS"][0]
 env = Environment("[#", "#]", "[[", "]]")
 
+console = Console()
 
 class ViewType(Enum):
     CREATE = "CreateView"
@@ -38,6 +40,7 @@ class Endpoint:
 
 def gen_templates(app_name: str):
     endpoints = get_endpoints(app_name)
+    console.print(f"generating in templates folder [bold]{TEMPLATES_FOLDER}")
 
     for endpoint in endpoints:
         fields = [field.name for field in endpoint.model._meta.fields]
@@ -49,12 +52,12 @@ def gen_templates(app_name: str):
         if endpoint.view_type == ViewType.CREATE:
             result = render_template({"model": model_name}, "create")
             write_template(endpoint.template_name, result)
-            print("created template:", endpoint.template_name)
+            console.print("created template:", endpoint.template_name, style="bright_black")
 
         elif endpoint.view_type == ViewType.UPDATE:
             result = render_template({"model": model_name}, "update")
             write_template(endpoint.template_name, result)
-            print("created template:", endpoint.template_name)
+            console.print("created template:", endpoint.template_name, style="bright_black")
 
         elif endpoint.view_type == ViewType.LIST:
             name = endpoint.view.context_object_name
@@ -71,7 +74,7 @@ def gen_templates(app_name: str):
 
             result = render_template(context_data, "list")
             write_template(endpoint.template_name, result)
-            print("created template:", endpoint.template_name)
+            console.print("created template:", endpoint.template_name, style="bright_black")
 
         elif endpoint.view_type == ViewType.DETAIL:
             name = endpoint.view.context_object_name
@@ -86,7 +89,7 @@ def gen_templates(app_name: str):
 
             result = render_template(context_data, "detail")
             write_template(endpoint.template_name, result)
-            print("created template:", endpoint.template_name)
+            console.print("created template:", endpoint.template_name, style="bright_black")
 
 
 def find_path_name(endpoints: list[Endpoint], model, view_type: ViewType):

diff --git a/hogwarts/management/commands/gentemplates.py b/hogwarts/management/commands/gentemplates.py
@@ -1,7 +1,7 @@
 from django.core.management.base import BaseCommand
 
-from ...magic_templates.gen_templates import gen_templates
 
+from ...magic_templates.gen_templates import gen_templates
 
 class Command(BaseCommand):
     help = "Code generation command"
@@ -11,7 +11,4 @@ def add_arguments(self, parser):
 
     def handle(self, *args, **options):
         app_name: str = options["app"]
-
         gen_templates(app_name)
-
-        print("Successfully created")
diff --git a/templates/posts/post_list.html b/templates/posts/post_list.html
diff --git a/templates/posts/post_update.html b/templates/posts/post_update.html