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
No related branches found
No related tags found
2 merge requests!216Release,!215Volby
...@@ -21,6 +21,7 @@ from wagtail.images.edit_handlers import ImageChooserPanel ...@@ -21,6 +21,7 @@ from wagtail.images.edit_handlers import ImageChooserPanel
from wagtailmetadata.models import MetadataPageMixin from wagtailmetadata.models import MetadataPageMixin
from shared.models import SubpageMixin from shared.models import SubpageMixin
from shared.utils import get_subpage_url
from tuning import help from tuning import help
from .forms import DonateForm from .forms import DonateForm
...@@ -73,13 +74,6 @@ class DonateFormAmountsMixin(models.Model): ...@@ -73,13 +74,6 @@ class DonateFormAmountsMixin(models.Model):
abstract = True 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): class DonateHomePage(DonateFormMixin, DonateFormAmountsMixin, Page, MetadataPageMixin):
### FIELDS ### FIELDS
...@@ -198,15 +192,15 @@ class DonateHomePage(DonateFormMixin, DonateFormAmountsMixin, Page, MetadataPage ...@@ -198,15 +192,15 @@ class DonateHomePage(DonateFormMixin, DonateFormAmountsMixin, Page, MetadataPage
@property @property
def info_page_url(self): def info_page_url(self):
return get_url(self, DonateInfoPage) return get_subpage_url(self, DonateInfoPage)
@property @property
def projects_page_url(self): def projects_page_url(self):
return get_url(self, DonateProjectIndexPage) return get_subpage_url(self, DonateProjectIndexPage)
@property @property
def regions_page_url(self): def regions_page_url(self):
return get_url(self, DonateRegionIndexPage) return get_subpage_url(self, DonateRegionIndexPage)
@property @property
def has_projects(self): def has_projects(self):
...@@ -332,7 +326,7 @@ class DonateRegionPage( ...@@ -332,7 +326,7 @@ class DonateRegionPage(
@property @property
def targeted_donations_page_url(self): def targeted_donations_page_url(self):
return get_url(self, DonateTargetedDonationsPage) return get_subpage_url(self, DonateTargetedDonationsPage)
@property @property
def has_targeted_donations(self): 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.
Finish editing this message first!
Please register or to comment