Source code for djpro.base

# This file is part of djpro.
#
# Copyright (c) 2024, Franco Gidaszewski <gidaszewskifranco@gmail.com>
#
# For the full copyright and license information, please view
# the LICENSE.txt file that was distributed with this source code.

import argparse
import os
from importlib import import_module
from .utils.logging import logging, log_with_color, Fore


class BaseCommand:
    def __init__(self):
        self.parser = argparse.ArgumentParser(
            description="CLI tool to create scalable and modular Django projects."
        )
        self.subparsers = self.parser.add_subparsers(dest="command", help="Commands")

    def execute(self):
        """Parse and run user command"""
        args = self.parser.parse_args()
        args.base_dir = os.path.join(os.getcwd(), "src")  # sources folder

        if not args.command:
            log_with_color(
                logging.INFO,
                "\nCommand not found.",
                Fore.YELLOW,
                delay=1,
            )
            help_module = import_module("djpro.commands.help")
            help_module.help()
            return

        if args.command == "help":
            try:
                help_module = import_module("djpro.commands.help")
                help_module.help()
            except ModuleNotFoundError:
                log_with_color(
                    logging.INFO,
                    "\nHelp command module not found.",
                    Fore.YELLOW,
                    delay=1,
                )
            return

        try:
            command_module = import_module(f"djpro.commands.{args.command}")
            log_with_color(logging.INFO, "Running commands...", Fore.BLUE, delay=2)
            command_module.run(args)

        except ModuleNotFoundError:
            log_with_color(
                logging.INFO,
                "\nCommand not found.",
                Fore.YELLOW,
                delay=1,
            )
            help_module = import_module("djpro.commands.help")
            help_module.help()

    def add_command(self, name, help_message, arguments=None):
        """Register a new subcommand"""
        command_parser = self.subparsers.add_parser(name, help=help_message)

        if arguments:
            for arg_name, arg_options in arguments.items():
                command_parser.add_argument(arg_name, **arg_options)


def main():
    cli = BaseCommand()

    # Help
    cli.add_command(
        "help",
        "Show help for a command.",
    )

    # Project
    cli.add_command(
        "project",
        "Create a new project.",
        arguments={
            "project_name": {
                "help": "Create a Django project with a scalable and modular architecture.",
                "type": str,
            },
            "--api": {
                "help": "Include API configuration",
                "action": "store_true",
            },
            "--docker": {
                "help": "Include Docker configuration for easy deployment",
                "action": "store_true",
            },
            "--unfold": {
                "help": "Include Unfold custom theme to Django admin site",
                "action": "store_true",
            },
        },
    )

    """ # Add service (not implemented)
    cli.add_command(
        "service",
        "Add a new service to your current project.",
        arguments={
            "--unfold": {
                "help": "Add Unfold configuration to your current project.",
                "action": "store_true",
            },
            "--docker": {
                "help": "Add Docker configuration to your current project.",
                "action": "store_true",
            },
        },
    )"""

    # Module
    cli.add_command(
        "module",
        "Create a new module.",
        arguments={
            "module_name": {
                "help": "Create a new module in your current project.",
                "type": str,
            },
            "-models": {
                "help": "Include models.py file.",
                "action": "store_true",
            },
            "-views": {
                "help": "Include views.py file.",
                "action": "store_true",
            },
            "-serializers": {
                "help": "Include serializers.py file.",
                "action": "store_true",
            },
            "-urls": {
                "help": "Include urls.py file.",
                "action": "store_true",
            },
            "-forms": {
                "help": "Include forms.py file.",
                "action": "store_true",
            },
            "-admin": {
                "help": "Include admin.py file.",
                "action": "store_true",
            },
            "-apps": {
                "help": "Include apps.py file.",
                "action": "store_true",
            },
        },
    )

    # Simplify manage.py commands
    cli.add_command(
        "man",
        "Run manage.py commands in a shorter form.",
        arguments={
            "man_command": {
                "help": "The manage.py command you want to run.",
                "type": str,
            },
            "help": {
                "help": "View list of available manage.py commands.",
                "action": "store_true",
            },
        },
    )

    cli.execute()