def __get_cb()

in libmozdata/connection.py [0:0]

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

    def __get_cb(self, query):
        """Get the callback to use when data have been retrieved

        Args:
            query (Query): the query

        Returns:
            function: the callback for the query
        """

        def cb(res, *args, **kwargs):
            if res.status_code == 200:
                try:
                    response = res.json()
                except ValueError:
                    response = res.text

                if query.handlerdata is not None:
                    query.handler(response, query.handlerdata)
                else:
                    query.handler(response)
            elif self.RAISE_ERROR:
                res.raise_for_status()
            else:
                print("Connection error:")
                print("   url: ", res.url)
                print("   text: ", res.text)

        return cb