From 273cf91f76d3a731d53734ac101d3d8b4403706b Mon Sep 17 00:00:00 2001 From: zandercymatics <141044360+zandercymatics@users.noreply.github.com> Date: Wed, 14 Aug 2024 13:33:02 -0600 Subject: [PATCH] Update portfolio.py --- src/registrar/models/portfolio.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/src/registrar/models/portfolio.py b/src/registrar/models/portfolio.py index fd5c3870d..bd1c091f8 100644 --- a/src/registrar/models/portfolio.py +++ b/src/registrar/models/portfolio.py @@ -124,6 +124,15 @@ class Portfolio(TimeStampedModel): def __str__(self) -> str: return str(self.organization_name) + def save(self, *args, **kwargs): + """Save override for custom properties""" + + # The urbanization field is only intended for the state_territory puerto rico + if self.state_territory != self.StateTerritoryChoices.PUERTO_RICO and self.urbanization: + self.urbanization = None + + super().save(*args, **kwargs) + @property def portfolio_type(self): """ @@ -148,12 +157,3 @@ class Portfolio(TimeStampedModel): def get_suborganizations(self): """Returns all suborganizations associated with this portfolio""" return self.portfolio_suborganizations.all() - - def save(self, *args, **kwargs): - """Save override for custom properties""" - - # The urbanization field is only intended for the state_territory puerto rico - if self.state_territory != self.StateTerritoryChoices.PUERTO_RICO and self.urbanization: - self.urbanization = None - - super().save(*args, **kwargs)