def special_query()

in mwaairflow/assets/plugins/operators/salesforce_to_s3_operator.py [0:0]

• 12 lines of code
• 5 McCabe index (conditional complexity)

    def special_query(self, query, sf_hook, relationship_object=None):
        if not query:
            raise ValueError("Query is None.  Cannot query nothing")

        sf_hook.sign_in()

        results = sf_hook.make_query(query)
        if relationship_object:
            records = []
            for r in results["records"]:
                if r.get(relationship_object, None):
                    records.extend(r[relationship_object]["records"])
            results["records"] = records

        return results