test

djpro/menu/admin.py

@@ -0,0 +1,7 @@
+from django.contrib import admin
+from .models import MenuItem
+
+class MenuItemAdmin(admin.ModelAdmin):
+    list_display = ('title', 'parent', 'url')
+
+admin.site.register(MenuItem, MenuItemAdmin)

djpro/menu/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class MenuConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'menu'

djpro/menu/models.py

@@ -0,0 +1,10 @@
+from django.db import models
+
+
+class MenuItem(models.Model):
+    title = models.CharField(max_length=100)
+    parent = models.ForeignKey('self', on_delete=models.CASCADE, null=True, blank=True, related_name='children')
+    url = models.CharField(max_length=100, null=True, blank=True)
+
+    def __str__(self):
+        return self.title

djpro/menu/templatetags/menu_tags.py

@@ -0,0 +1,12 @@
+from django import template
+from menu.models import MenuItem
+
+register = template.Library()
+
+@register.simple_tag(takes_context=True)
+def draw_menu(context, menu_name):
+    menu_items = MenuItem.objects.filter(title=menu_name).select_related('parent').prefetch_related('children')
+    return {
+        'menu_items': menu_items,
+        'current_url': context['request'].path,
+    }

djpro/menu/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

djpro/menu/urls.py

@@ -0,0 +1,6 @@
+from django.urls import path
+from . import views
+
+urlpatterns = [
+    path('', views.index, name='index'),
+]

djpro/menu/views.py

@@ -0,0 +1,9 @@
+from django.shortcuts import render
+from .models import MenuItem
+
+def index(request):
+    menu_items = MenuItem.objects.all()
+    context = {
+        'menu_items': menu_items,
+    }
+    return render(request, 'base.html', context)