[
https://issues.apache.org/jira/browse/AIRFLOW-4048?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
Ash Berlin-Taylor resolved AIRFLOW-4048.
----------------------------------------
Resolution: Fixed
> HttpSensor provide context to response_check
> --------------------------------------------
>
> Key: AIRFLOW-4048
> URL: https://issues.apache.org/jira/browse/AIRFLOW-4048
> Project: Apache Airflow
> Issue Type: Improvement
> Reporter: raphael auv
> Assignee: raphael auv
> Priority: Minor
> Fix For: 1.10.4
>
>
> Actually the response_check do not get by parameter the context , with a
> simple , provide_context option at the constructor , we could provide the
> context to the response_check function
> actual code :
> {code:java}
> ...
>
> def poke(self, context):
> self.log.info('Poking: %s', self.endpoint)
> try:
> response = self.hook.run(self.endpoint,
> data=self.request_params,
> headers=self.headers,
> extra_options=self.extra_options)
> if self.response_check:
> # run content check on response
> return self.response_check(response)
> {code}
>
> futur code:
> {code:java}
> def __init__(self,
> endpoint,
> http_conn_id='http_default',
> method='GET',
> request_params=None,
> headers=None,
> response_check=None,
> extra_options=None
> provide_context=False, *args, **kwargs):
> super(HttpSensor, self).__init__(*args, **kwargs)
> ...
> self.provide_context = provide_context
> def poke(self, context):
> self.log.info('Poking: %s', self.endpoint)
> try:
> response = self.hook.run(self.endpoint,
> data=self.request_params,
> headers=self.headers,
> extra_options=self.extra_options)
> if self.response_check:
> # run content check on response
> response_check_kwargs = {}
> if self.provide_context:
> response_check_kwargs["context"] = context
> return self.response_check(response, **response_check_kwargs)
> {code}
--
This message was sent by Atlassian JIRA
(v7.6.3#76005)
