zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-07-12 22:18:25 +02:00
Code Issues Projects Releases Packages Wiki Activity

Update portfolio.py

Browse source
This commit is contained in:
zandercymatics 2024-08-14 13:33:02 -06:00
parent 5afad6b91d
commit 273cf91f76
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
1 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

18
src/registrar/models/portfolio.py
View file

@ -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)