Skip to content
Snippets Groups Projects
Commit dc78acba authored by jan.bednarik's avatar jan.bednarik
Browse files

donate: Get subpage URL refactoring

parent afdd0b64
Branches
No related tags found
2 merge requests!216Release,!215Volby
......@@ -21,6 +21,7 @@ from wagtail.images.edit_handlers import ImageChooserPanel
from wagtailmetadata.models import MetadataPageMixin
from shared.models import SubpageMixin
from shared.utils import get_subpage_url
from tuning import help
from .forms import DonateForm
......@@ -73,13 +74,6 @@ class DonateFormAmountsMixin(models.Model):
abstract = True
def get_url(page, dest_page_type):
try:
return page.get_children().type(dest_page_type).live().first().get_url()
except (Page.DoesNotExist, AttributeError):
return "#"
class DonateHomePage(DonateFormMixin, DonateFormAmountsMixin, Page, MetadataPageMixin):
### FIELDS
......@@ -198,15 +192,15 @@ class DonateHomePage(DonateFormMixin, DonateFormAmountsMixin, Page, MetadataPage
@property
def info_page_url(self):
return get_url(self, DonateInfoPage)
return get_subpage_url(self, DonateInfoPage)
@property
def projects_page_url(self):
return get_url(self, DonateProjectIndexPage)
return get_subpage_url(self, DonateProjectIndexPage)
@property
def regions_page_url(self):
return get_url(self, DonateRegionIndexPage)
return get_subpage_url(self, DonateRegionIndexPage)
@property
def has_projects(self):
......@@ -332,7 +326,7 @@ class DonateRegionPage(
@property
def targeted_donations_page_url(self):
return get_url(self, DonateTargetedDonationsPage)
return get_subpage_url(self, DonateTargetedDonationsPage)
@property
def has_targeted_donations(self):
......
from wagtail.core.models import Page
def get_subpage_url(page, dest_page_type):
try:
return page.get_descendants().type(dest_page_type).live().first().get_url()
except (Page.DoesNotExist, AttributeError):
return "#"
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment