startproject.py

import click
import os
import shutil
import subprocess


@click.command()
@click.option("-d", "--delete", is_flag=True, help="Delete existing project files")
@click.option("-dj", "--django", is_flag=True, help="Use django mongodb template")
@click.option("-w", "--wagtail", is_flag=True, help="Use wagtail mongodb template")
@click.argument("project_name", required=False, default="backend")
def startproject(
    delete,
    django,
    wagtail,
    project_name,
):
    """Run `startproject` with custom templates."""
    if os.path.exists("manage.py"):
        click.echo("manage.py already exists")
        if not delete:
            click.echo("Use -d to delete existing project files")
            return
    if delete:
        items = {
            ".babelrc": os.path.isfile,
            ".dockerignore": os.path.isfile,
            ".browserslistrc": os.path.isfile,
            ".eslintrc": os.path.isfile,
            ".nvmrc": os.path.isfile,
            ".stylelintrc.json": os.path.isfile,
            "Dockerfile": os.path.isfile,
            "apps": os.path.isdir,
            "home": os.path.isdir,
            "backend": os.path.isdir,
            "db.sqlite3": os.path.isfile,
            "frontend": os.path.isdir,
            "mongo_migrations": os.path.isdir,
            "manage.py": os.path.isfile,
            "package-lock.json": os.path.isfile,
            "package.json": os.path.isfile,
            "postcss.config.js": os.path.isfile,
            "requirements.txt": os.path.isfile,
            "search": os.path.isdir,
        }
        for item, check_function in items.items():
            if check_function(item):
                if os.path.isdir(item):
                    shutil.rmtree(item)
                    click.echo(f"Removed directory: {item}")
                elif os.path.isfile(item):
                    os.remove(item)
                    click.echo(f"Removed file: {item}")
            else:
                click.echo(f"Skipping: {item} does not exist")
        return
    template = None
    django_admin = "django-admin"
    startproject = "startproject"
    startapp = "startapp"
    if wagtail:
        template = os.path.join(os.path.join("src", "wagtail-mongodb-project"))
        django_admin = "wagtail"
        startproject = "start"
    elif django:
        template = os.path.join(os.path.join("src", "django-mongodb-project"))
    if not template:
        template = os.path.join(
            os.path.join("src", "django-mongodb-templates", "project_template")
        )
    click.echo(f"Using template: {template}")
    subprocess.run(
        [
            django_admin,
            startproject,
            project_name,
            ".",
            "--template",
            template,
        ]
    )
    frontend_template = os.path.join(
        "src", "django-mongodb-templates", "frontend_template"
    )
    click.echo(f"Using template: {frontend_template}")
    subprocess.run(
        [
            django_admin,
            startproject,
            "frontend",
            ".",
            "--template",
            frontend_template,
        ]
    )
    if not wagtail:
        home_template = os.path.join("src", "django-mongodb-templates", "home_template")
        click.echo(f"Using template: {home_template}")
        subprocess.run(
            [
                django_admin,
                startapp,
                "home",
                "--template",
                home_template,
            ]
        )