dj url

url.py

from dataclasses import dataclass
import importlib
import re

from django.conf import settings
from django.core.management.base import BaseCommand
from django.urls.base import resolve
from django.urls.base import reverse


@dataclass
class Result:
    url: str
    name: str
    view: str
    template: str = None


def example_url(url):
    return re.sub(r"<int:\w+>", "1", url)


def get_template(full_view):
    parts = full_view.split(".")
    class_name = parts[-1]
    module_path = ".".join(parts[:-1])
    module = importlib.import_module(module_path)
    cls = getattr(module, class_name)
    return getattr(cls, "template_name")


class Command(BaseCommand):
    help = "Forward/reverse view name -> view."

    def add_arguments(self, parser):
        parser.add_argument("argument", nargs=1, help="URL, URL name or view")
        parser.add_argument("display", nargs="?", help="One of url, name, view or template")

    def handle(self, *args, **options):
        argument = options["argument"][0]
        self.display = options["display"]

        if argument.startswith("/"):
            self.handle_url(argument)
        elif ":" in argument:
            self.handle_view_name(argument)
        else:
            self.handle_view(argument)

    def handle_url(self, url):
        url = example_url(url)
        match = resolve(url)
        view_name = match._func_path.split(".")[-1]
        name = f"{match.namespaces[0]}:{match.url_name}"
        self.print_result(Result(url=url, name=name, view=view_name, template=get_template(match._func_path)))

    def handle_view_name(self, view_name):
        self.handle_url(reverse(view_name))

    def handle_view(self, view):
        urlconf = __import__(settings.ROOT_URLCONF, {}, {}, [""])
        self.search_urlconf(urlconf.urlpatterns, view)

    def search_urlconf(self, url_patterns, view, base="/"):
        for pattern in url_patterns:
            if url_patterns := getattr(pattern, "url_patterns", None):
                self.search_urlconf(url_patterns, view, base + str(pattern.pattern))
            elif callback := getattr(pattern, "callback", None):
                if view_class := getattr(callback, "view_class", None):
                    if view_class.__name__ == view:
                        url = base + str(pattern.pattern)
                        self.handle_url(url)

    def print_result(self, result: Result):

        if self.display:
            print(getattr(result, self.display))
            return

        print("")
        print(f"{result.url} aka {result.name}")
        print(f"  -> {result.view}")
        if result.template:
            print(f"    -> {result.template}")
        print("")