def get_queryset(self):
# retrieve post request payload
json_request = self.request.query_params
# retrieve fiscal_year & awarding_agency_id from request
fiscal_year = json_request.get("fiscal_year", None)
awarding_agency_id = json_request.get("awarding_agency_id", None)
# required query parameters were not provided
if not (fiscal_year and awarding_agency_id):
raise InvalidParameterException(
"Missing one or more required query parameters: fiscal_year, awarding_agency_id"
)
if not check_valid_toptier_agency(awarding_agency_id):
raise InvalidParameterException("Awarding Agency ID provided must correspond to a toptier agency")
# change user provided PK (awarding_agency_id) to toptier_agency_id,
# filter and include all subtier_agency_id(s).
top_tier_agency_id = Agency.objects.filter(id=awarding_agency_id).first().toptier_agency_id
queryset = TransactionNormalized.objects.all()
# Filter based on fiscal year and agency id
queryset = queryset.filter(
fiscal_year=fiscal_year, awarding_agency__toptier_agency=top_tier_agency_id
).annotate(award_category=F("award__category"))
# Sum obligations for each Award Category type
queryset = (
queryset.values("award_category")
.annotate(obligated_amount=Sum("federal_action_obligation"))
.order_by("-obligated_amount")
)
return queryset
def get_queryset(self):
# retrieve post request payload
json_request = self.request.query_params
# Retrieve fiscal_year & awarding_agency_id from Request
fiscal_year = json_request.get("fiscal_year")
awarding_agency_id = json_request.get("awarding_agency_id")
# Optional Award Category
award_category = json_request.get("award_category")
# Required Query Parameters were not Provided
if not (fiscal_year and awarding_agency_id):
raise InvalidParameterException(
"Missing one or more required query parameters: fiscal_year, awarding_agency_id"
)
if not check_valid_toptier_agency(awarding_agency_id):
raise InvalidParameterException("Awarding Agency ID provided must correspond to a toptier agency")
toptier_agency = Agency.objects.filter(id=awarding_agency_id).first().toptier_agency
queryset = TransactionNormalized.objects.filter(federal_action_obligation__isnull=False)
# DS-1655: if the AID is "097" (DOD), Include the branches of the military in the queryset
if toptier_agency.toptier_code == DOD_CGAC:
tta_list = DOD_ARMED_FORCES_CGAC
queryset = queryset.filter(
# Filter based on fiscal_year and awarding_category_id
fiscal_year=fiscal_year,
awarding_agency__toptier_agency__toptier_code__in=tta_list,
)
else:
queryset = queryset.filter(
# Filter based on fiscal_year and awarding_category_id
fiscal_year=fiscal_year,
awarding_agency__toptier_agency__toptier_code=toptier_agency.toptier_code,
)
queryset = queryset.annotate(
award_category=F("award__category"),
recipient_id=F("recipient__legal_entity_id"),
recipient_name=F("recipient__recipient_name"),
)
if award_category is not None:
# Filter based on award_category
if award_category != "other":
queryset = queryset.filter(award_category=award_category)
else:
queryset = queryset.filter(Q(award_category="insurance") | Q(award_category="other")).annotate(
award_category=Case(When(award_category="insurance", then=Value("other")), output_field=CharField())
)
# Sum Obligations for each Recipient
queryset = (
queryset.values("award_category", "recipient_id", "recipient_name")
.annotate(obligated_amount=Sum("federal_action_obligation"))
.order_by("-obligated_amount")
)
return queryset
def test_check_valid_toptier_agency_invalid(monkeypatch):
agencies = MockSet()
monkeypatch.setattr('usaspending_api.references.models.Agency.objects', agencies)
agencies.add(
MockModel(mock_name='subtier agency', id=54321, toptier_flag=False)
)
assert not check_valid_toptier_agency(54321)
def test_check_valid_toptier_agency_valid(monkeypatch):
agencies = MockSet()
monkeypatch.setattr('usaspending_api.references.models.Agency.objects', agencies)
agencies.add(
MockModel(mock_name='toptier agency', id=12345, toptier_flag=True)
)
assert check_valid_toptier_agency(12345)
def test_check_valid_toptier_agency_invalid(monkeypatch):
agencies = MockSet()
monkeypatch.setattr("usaspending_api.references.models.Agency.objects",
agencies)
agencies.add(
MockModel(mock_name="subtier agency", id=54321, toptier_flag=False))
assert not check_valid_toptier_agency(54321)
def test_check_valid_toptier_agency_valid(monkeypatch):
agencies = MockSet()
monkeypatch.setattr("usaspending_api.references.models.Agency.objects",
agencies)
agencies.add(
MockModel(mock_name="toptier agency", id=12345, toptier_flag=True))
assert check_valid_toptier_agency(12345)
def get_queryset(self):
# retrieve post request payload
json_request = self.request.query_params
# Retrieve fiscal_year & awarding_agency_id from Request
fiscal_year = json_request.get("fiscal_year")
awarding_agency_id = json_request.get("awarding_agency_id")
# Optional Award Category
award_category = json_request.get("award_category")
# Required Query Parameters were not Provided
if not (fiscal_year and awarding_agency_id):
raise InvalidParameterException(
"Missing one or more required query parameters: fiscal_year, awarding_agency_id"
)
if not check_valid_toptier_agency(awarding_agency_id):
raise InvalidParameterException(
"Awarding Agency ID provided must correspond to a toptier agency"
)
queryset = TransactionNormalized.objects.filter(
federal_action_obligation__isnull=False,
fiscal_year=fiscal_year,
awarding_agency_id=awarding_agency_id
).annotate(
award_category=F("award__category"),
recipient_name=Coalesce(
F("award__latest_transaction__assistance_data__awardee_or_recipient_legal"
),
F("award__latest_transaction__contract_data__awardee_or_recipient_legal"
),
),
)
if award_category is not None:
# Filter based on award_category
if award_category != "other":
queryset = queryset.filter(award_category=award_category)
else:
queryset = queryset.filter(
Q(award_category="insurance")
| Q(award_category="other")).annotate(award_category=Case(
When(award_category="insurance", then=Value("other")),
output_field=CharField()))
# Sum Obligations for each Recipient
queryset = (queryset.values(
"award_category", "recipient_name").annotate(obligated_amount=Sum(
"federal_action_obligation")).order_by("-obligated_amount"))
return queryset
def test_check_valid_toptier_agency_invalid():
mommy.make("references.Agency", id=54321, toptier_flag=False)
assert not check_valid_toptier_agency(54321)
def test_check_valid_toptier_agency_valid():
mommy.make("references.Agency", id=12345, toptier_flag=True)
assert check_valid_toptier_agency(12345)