Add logic+tests for handling display of type of renewable generation (#7)
This commit is contained in:
@ -994,5 +994,23 @@ class CaseStudy(models.Model):
|
||||
"""Gets the 11 character YouTube video ID from the video field."""
|
||||
return parse.parse_qs(parse.urlparse(self.video).query)["v"][0]
|
||||
|
||||
def get_renewable_generation_detail(self):
|
||||
"""Prepend appropriate descriptive text when accessing renewable generation type."""
|
||||
|
||||
if self.generation_technology:
|
||||
if self.generation_technology.endswith('WE'):
|
||||
return _('Wind energy') + " – " + self.get_generation_technology_display()
|
||||
elif self.generation_technology.endswith('PV'):
|
||||
return _('Photovoltaic electricity') + " – " + self.get_generation_technology_display()
|
||||
elif self.generation_technology.endswith('HYD'):
|
||||
return _('Hydroelectric') + " – " + self.get_generation_technology_display()
|
||||
elif self.generation_technology == 'OTHR':
|
||||
return self.generation_technology_other
|
||||
else:
|
||||
return self.get_generation_technology_display()
|
||||
|
||||
else:
|
||||
return ""
|
||||
|
||||
class Meta:
|
||||
verbose_name_plural = 'case studies'
|
||||
|
Reference in New Issue
Block a user