def process()

in processors/budget.py [0:0]
51 lines of code
10 McCabe index (conditional complexity)

    def process(self, output_var={'projects': 'projects', 'budget': 'budget'}):
        if 'budgetId' not in self.event[
                'attributes'] or 'billingAccountId' not in self.event[
                    'attributes']:
            raise MissingAttributesException(
                'Message attributes are missing budgetId and billingAccountId!')

        client = self._get_budget_service_client()
        request = self._get_budget_request()
        request.name = 'billingAccounts/%s/budgets/%s' % (
            self.event['attributes']['billingAccountId'],
            self.event['attributes']['budgetId'])
        self.logger.debug('Retrieving details for budget.',
                          extra={'budget': request.name})

        response = client.get_budget(request)

        projects = self.expand_projects(response.budget_filter.projects)

        data = json.loads(self.data)

        ret = {
            output_var['projects']: projects,
            output_var['budget']: {
                'name':
                    response.name,
                'display_name':
                    response.display_name,
                'cost_amount':
                    data['costAmount'] if 'costAmount' in data else '',
                'cost_interval_start':
                    data['costIntervalStart']
                    if 'costIntervalStart' in data else '',
                'alert_threshold_exceeded':
                    data['alertThresholdExceeded']
                    if 'alertThresholdExceeded' in data else '',
                'forecast_threshold_exceeded':
                    data['forecastThresholdExceeded']
                    if 'forecastThresholdExceeded' in data else '',
                'credit_types_treatment':
                    str(response.budget_filter.credit_types_treatment
                       ).replace('CreditTypesTreatment.', ''),
                'amount_type':
                    'last_period'
                    if response.amount.last_period_amount else 'specified',
                'amount_units':
                    response.amount.last_period_amount.units
                    if response.amount.last_period_amount else
                    response.amount.specified_amount.units,
                'amounts_currency_code':
                    response.amount.last_period_amount.currency_code
                    if response.amount.last_period_amount else
                    response.amount.specified_amount.currency_code,
            }
        }
        return ret