Function handle has a Cognitive Complexity of 40 (exceeds 15 allowed). Consider refactoring.

Open

    def handle(self, *args, **options):
        """Run the application."""

        # Make sure
        #   settings.BULK_DOWNLOAD_S3_BUCKET_NAME

• Read up
  Read up

File api_request_utils.py has 346 lines of code (exceeds 250 allowed). Consider refactoring.

Open

from datetime import date, time, datetime
from django.contrib.postgres.search import SearchVector
from django.db.models import Q
from django.utils import timezone
from usaspending_api.common.exceptions import InvalidParameterException

File elasticsearch_indexer.py has 344 lines of code (exceeds 250 allowed). Consider refactoring.

Open

import logging
from abc import ABC, abstractmethod
from datetime import datetime, timezone
from time import perf_counter

File source_procurement_transaction.py has 331 lines of code (exceeds 250 allowed). Consider refactoring.

Open

from django.db import models
from django.contrib.postgres.fields import ArrayField

from usaspending_api.common.custom_django_fields import NumericField, NaiveTimestampField

Function spending_filter has a Cognitive Complexity of 37 (exceeds 15 allowed). Consider refactoring.

Open

def spending_filter(alt_set, queryset, filters, _type):
    for key, value in filters.items():
        # check for valid key
        if value is None:
            raise InvalidParameterException("Invalid filter: " + key + " has null as its value.")

• Read up
  Read up

File detached_award_procurement.py has 316 lines of code (exceeds 250 allowed). Consider refactoring.

Open

DETACHED_AWARD_PROCUREMENT_COLUMNS = {
    "detached_award_procurement_id": {"delta": "INTEGER", "postgres": "INTEGER"},
    "detached_award_proc_unique": {"delta": "STRING", "postgres": "TEXT"},
    "a_76_fair_act_action": {"delta": "STRING", "postgres": "TEXT"},
    "a_76_fair_act_action_desc": {"delta": "STRING", "postgres": "TEXT"},

Function _handle_exit_signal has a Cognitive Complexity of 35 (exceeds 15 allowed). Consider refactoring.

Open

    def _handle_exit_signal(self, signum, frame, parent_dispatcher_signaled=True, is_retry=False):
        """Attempt to gracefully handle the exiting of the job as a result of receiving an exit signal.

        NOTE: This handler is only expected to be run from the parent dispatcher process. It is an error
        condition if it is invoked from the child worker process. Signals received in the child worker

• Read up
  Read up

File autocomplete.py has 312 lines of code (exceeds 250 allowed). Consider refactoring.

Open

from django.db.models import Case, F, IntegerField, Q, When
from django.db.models.functions import Upper
from rest_framework.response import Response
from rest_framework.views import APIView

File search_elasticsearch.py has 311 lines of code (exceeds 250 allowed). Consider refactoring.

Open

import copy
import logging

from django.conf import settings
from rest_framework.response import Response

File financial_accounts_by_awards.py has 310 lines of code (exceeds 250 allowed). Consider refactoring.

Open

FINANCIAL_ACCOUNTS_BY_AWARDS_COLUMNS = {
    "data_source": "STRING",
    "financial_accounts_by_awards_id": "INTEGER NOT NULL",
    "piid": "STRING",
    "parent_award_id": "STRING",

Function source_subquery_sql has a Cognitive Complexity of 34 (exceeds 15 allowed). Consider refactoring.

Open

    def source_subquery_sql(self, transaction_type=None):
        def build_date_format_sql(col: TransactionColumn, is_casted_to_date: bool = True) -> str:
            """Builder function to wrap a column in date-parsing logic.

            It will either parse it in mmddYYYY format with - or / as a required separator, or in YYYYmmdd format

• Read up
  Read up

File update_transactions.py has 303 lines of code (exceeds 250 allowed). Consider refactoring.

Open

import logging

from datetime import datetime
from usaspending_api.common.helpers.date_helper import fy
from django.core.management.base import BaseCommand

File spending.py has 296 lines of code (exceeds 250 allowed). Consider refactoring.

Open

import logging
from decimal import Decimal
from typing import List

from django.contrib.postgres.fields import ArrayField

Function __init__ has a Cognitive Complexity of 31 (exceeds 15 allowed). Consider refactoring.

Open

    def __init__(self, *args, **kwargs):
        # Grab any kwargs include and exclude fields, these are typically
        # passed in by a parent serializer to the child serializer
        kwargs_has_include = "fields" in kwargs
        kwargs_has_exclude = "exclude" in kwargs

• Read up
  Read up

File spending_by_geography.py has 285 lines of code (exceeds 250 allowed). Consider refactoring.

Open

from decimal import Decimal
from enum import Enum
from typing import Optional, List, Dict

from rest_framework.request import Request

Function _select_columns has a Cognitive Complexity of 30 (exceeds 15 allowed). Consider refactoring.

Open

def _select_columns(sql: str) -> Tuple[str, List[str]]:
    in_quotes = False
    in_cte = False
    parens_depth = 0
    last_processed_index = 0

• Read up
  Read up

File disaster_covid19_file_f_contracts.py has 275 lines of code (exceeds 250 allowed). Consider refactoring.

Open

f_contracts_sql_string = """
SELECT
    subaward_search.unique_award_key AS prime_award_unique_key,
    subaward_search.award_id AS prime_award_piid,
    subaward_search.parent_award_id AS prime_award_parent_piid,

Function load_data_into_model has a Cognitive Complexity of 29 (exceeds 15 allowed). Consider refactoring.

Open

def load_data_into_model(model_instance, data, **kwargs):
    """
    Loads data into a model instance
    Data should be a row, a dict of field -> value pairs
    Keyword args:

• Read up
  Read up

Function build_elasticsearch_result has a Cognitive Complexity of 29 (exceeds 15 allowed). Consider refactoring.

Open

    def build_elasticsearch_result(self, response: dict) -> List[dict]:
        def _key_to_geo_code(key):
            if self.location_type == LocationType.COUNTRY:
                return key
            return f"{code_to_state[key[:2]]['fips']}{key[2:]}" if (key and key[:2] in code_to_state) else None

• Read up
  Read up

File disaster_covid19_file_f_grants.py has 272 lines of code (exceeds 250 allowed). Consider refactoring.

Open

f_grants_sql_string = """
SELECT
    subaward_search.unique_award_key AS prime_award_unique_key,
    subaward_search.award_id AS prime_award_fain,
    subaward_search.award_amount AS prime_award_amount,