示例#1
0
 def __init__(
     self,
     *,
     flow_name: str,
     appflow_run_task_id: str,
     ignore_downstream_trigger_rules: bool = True,
     aws_conn_id: str = "aws_default",
     region: Optional[str] = None,
     **kwargs,
 ) -> None:
     if get_airflow_version() >= (2, 3):
         kwargs[
             "ignore_downstream_trigger_rules"] = ignore_downstream_trigger_rules
     else:
         self.log.warning(
             "Ignoring argument ignore_downstream_trigger_rules (%s) - Only supported for Airflow >= 2.3",
             ignore_downstream_trigger_rules,
         )
     super().__init__(
         python_callable=self._has_new_records_func,
         op_kwargs={
             "flow_name": flow_name,
             "appflow_run_task_id": appflow_run_task_id,
         },
         **kwargs,
     )
     self.aws_conn_id = aws_conn_id
     self.region = region
示例#2
0
    def get_conn_uri(self, conn_id: str) -> Optional[str]:
        """
        Return URI representation of Connection conn_id.

        As of Airflow version 2.3.0 this method is deprecated.

        :param conn_id: the connection id
        :return: deserialized Connection
        """
        if get_airflow_version() >= (2, 3):
            warnings.warn(
                f"Method `{self.__class__.__name__}.get_conn_uri` is deprecated and will be removed "
                "in a future release.  Please use method `get_conn_value` instead.",
                DeprecationWarning,
                stacklevel=2,
            )
        return self.get_conn_value(conn_id)