django-components/django_components/management/commands/startcomponent.py
2023-10-09 22:28:11 +02:00

81 lines
2.5 KiB
Python

import os
from django.conf import settings
from django.core.management.base import BaseCommand, CommandError
class Command(BaseCommand):
help = "Creates a new component"
def add_arguments(self, parser):
parser.add_argument(
"name", type=str, help="The name of the component to create"
)
def handle(self, *args, **kwargs):
name = kwargs["name"]
base_dir = settings.BASE_DIR
components_path = os.path.join(base_dir, f"components/{name}")
if os.path.exists(
components_path
): # If component directory already exists
raise CommandError(
f'The component "{name}" already exists at {components_path}.'
)
os.makedirs(components_path)
with open(components_path + "/index.js", "w") as f:
script_content = f"""
window.addEventListener('load', (event) => {{
console.log("{name} component is fully loaded");
}});
"""
f.write(script_content.strip())
with open(components_path + "/style.css", "w") as f:
style_content = f"""
.component-{name} {{
background: red;
}}
"""
f.write(style_content.strip())
with open(components_path + "/template.html", "w") as f:
template_content = f"""
<div class="component-{name}">
Hello from {name} component!
</div>
"""
f.write(template_content.strip())
with open(components_path + f"/{name}.py", "w") as f:
py_content = f"""
# In a file called [project root]/components/{name}/{name}.py
from django_components import component
@component.register("{name}")
class {name.capitalize()}(component.Component):
# Templates inside `[your apps]/components` dir and `[project root]/components` dir will be automatically found. To customize which template to use based on context
# you can override def get_template_name() instead of specifying the below variable.
template_name = "{name}/template.html"
# This component takes one parameter, a date string to show in the template
def get_context_data(self, date):
return {{
"date": date,
}}
class Media:
css = "{name}/style.css"
js = "{name}/script.js"
"""
f.write(py_content.strip())
self.stdout.write(
self.style.SUCCESS(
f"Successfully created {name} component at {components_path}"
)
)