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

move agency function to be inside Federal Agency model

Browse source
This commit is contained in:
CocoByte 2024-06-18 11:56:19 -06:00
parent 42efc73722
commit 732cf2379b
No known key found for this signature in database
GPG key ID: BBFAA2526384C97F
3 changed files with 13 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -6,9 +6,9 @@ from registrar.models.federal_agency import FederalAgency
from .utility.time_stamped_model import TimeStampedModel
def get_default_federal_agency():
"""returns non-federal agency"""
return FederalAgency.objects.filter(agency="Non-Federal Agency").first()
# def get_default_federal_agency():
# """returns non-federal agency"""
# return FederalAgency.objects.filter(agency="Non-Federal Agency").first()
class Portfolio(TimeStampedModel):
@ -20,7 +20,7 @@ class Portfolio(TimeStampedModel):
# use the short names in Django admin
OrganizationChoices = DomainRequest.OrganizationChoices
StateTerritoryChoices = DomainRequest.StateTerritoryChoices
# Stores who created this model. If no creator is specified in DJA,
# then the creator will default to the current request user"""
creator = models.ForeignKey("registrar.User", on_delete=models.PROTECT, help_text="Associated user", unique=False)
@ -35,7 +35,7 @@ class Portfolio(TimeStampedModel):
on_delete=models.PROTECT,
help_text="Associated federal agency",
unique=False,
default=get_default_federal_agency,
default=FederalAgency.get_non_federal_agency,
)
organization_type = models.CharField(