airflow.providers.http.hooks.http

Module Contents

Classes

HttpHook

Interact with HTTP servers.
class airflow.providers.http.hooks.http.HttpHook(method: str = 'POST', http_conn_id: str = default_conn_name, auth_type: Any = HTTPBasicAuth)[source]

Bases: airflow.hooks.base.BaseHook

Interact with HTTP servers.

Parameters

  • method (str) -- the API method to be called

  • http_conn_id (str) -- http connection that has the base API url i.e https://www.google.com/ and optional authentication credentials. Default headers can also be specified in the Extra field in json format.

  • auth_type (AuthBase of python requests lib) -- The auth type for the service

conn_name_attr = http_conn_id[source]
default_conn_name = http_default[source]
conn_type = http[source]
hook_name = HTTP[source]
get_conn(self, headers: Optional[Dict[Any, Any]] = None) → requests.Session[source]

Returns http session for use with requests

Parameters

headers (dict) -- additional headers to be passed through as a dictionary

run(self, endpoint: Optional[str] = None, data: Optional[Union[Dict[str, Any], str]] = None, headers: Optional[Dict[str, Any]] = None, extra_options: Optional[Dict[str, Any]] = None, **request_kwargs: Any) → Any[source]

Performs the request

Parameters

  • endpoint (str) -- the endpoint to be called i.e. resource/v1/query?

  • data (dict) -- payload to be uploaded or request parameters

  • headers (dict) -- additional headers to be passed through as a dictionary

  • extra_options (dict) -- additional options to be used when executing the request i.e. {'check_response': False} to avoid checking raising exceptions on non 2XX or 3XX status codes

  • request_kwargs -- Additional kwargs to pass when creating a request. For example, run(json=obj) is passed as requests.Request(json=obj)

check_response(self, response: requests.Response) → None[source]

Checks the status code and raise an AirflowException exception on non 2XX or 3XX status codes

Parameters

response (requests.response) -- A requests response object

run_and_check(self, session: requests.Session, prepped_request: requests.PreparedRequest, extra_options: Dict[Any, Any]) → Any[source]

Grabs extra options like timeout and actually runs the request, checking for the result

Parameters

  • session (requests.Session) -- the session to be used to execute the request

  • prepped_request (session.prepare_request) -- the prepared request generated in run()

  • extra_options (dict) -- additional options to be used when executing the request i.e. {'check_response': False} to avoid checking raising exceptions on non 2XX or 3XX status codes

run_with_advanced_retry(self, _retry_args: Dict[Any, Any], *args: Any, **kwargs: Any) → Any[source]

Runs Hook.run() with a Tenacity decorator attached to it. This is useful for connectors which might be disturbed by intermittent issues and should not instantly fail.

Parameters

_retry_args (dict) -- Arguments which define the retry behaviour. See Tenacity documentation at https://github.com/jd/tenacity

hook = HttpHook(http_conn_id="my_conn", method="GET")
retry_args = dict(
    wait=tenacity.wait_exponential(),
    stop=tenacity.stop_after_attempt(10),
    retry=requests.exceptions.ConnectionError,
)
hook.run_with_advanced_retry(endpoint="v1/test", _retry_args=retry_args)
test_connection(self)[source]

Test HTTP Connection

Was this entry helpful?