diff --git a/core/models.py b/core/models.py index 2b4b6b0..072fbea 100644 --- a/core/models.py +++ b/core/models.py @@ -1,6 +1,10 @@ from django.db import models from admin_ordering.models import OrderableModel from filebrowser.fields import FileBrowseField +from django.urls.base import reverse_lazy + +from db.calendar.models import Calendar +from db.membership.models import Membership # Create your models here. @@ -13,6 +17,18 @@ class Button(OrderableModel): def __str__(self): return self.title or '-' + @property + def link(self): + h = self.href + c = None + + if h == reverse_lazy('calendar'): + c = Calendar.get_current() + elif h in reverse_lazy('membership'): + c = Membership.get_current() + + return h + (('#' + str(c.year)) if c is not None else '') + class Meta(OrderableModel.Meta): verbose_name = 'Przycisk nawigacji (zielony)' verbose_name_plural = 'Przyciski nawigacji (zielone)' diff --git a/core/templates/base.html b/core/templates/base.html index 6f252b2..7e1cb75 100644 --- a/core/templates/base.html +++ b/core/templates/base.html @@ -55,7 +55,7 @@