You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@airflow.apache.org by GitBox <gi...@apache.org> on 2020/11/12 20:46:13 UTC
[GitHub] [airflow] kaxil commented on a change in pull request #11850: Add Telegram hook and operator
kaxil commented on a change in pull request #11850:
URL: https://github.com/apache/airflow/pull/11850#discussion_r522415435
##########
File path: airflow/providers/telegram/hooks/telegram.py
##########
@@ -0,0 +1,153 @@
+#
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+"""Hook for Telegram"""
+from typing import Optional
+
+import telegram
+import tenacity
+
+from airflow.exceptions import AirflowException
+from airflow.hooks.base_hook import BaseHook
+
+
+class TelegramHook(BaseHook):
+ """
+ This hook allows you to post messages to Telegram using the telegram python-telegram-bot library.
+
+ The library can be found here: https://github.com/python-telegram-bot/python-telegram-bot
+ It accepts both telegram bot API token directly or connection that has telegram bot API token.
+ If both supplied, token parameter will be given precedence, otherwise 'password' field in the connection
+ from telegram_conn_id will be used.
+ chat_id can also be provided in the connection using 'host' field in connection.
+ Following is the details of a telegram_connection:
+ name: 'telegram-connection-name'
+ conn_type: 'http'
+ password: 'TELEGRAM_TOKEN'
+ host: 'chat_id' (optional)
+ Examples:
+ .. code-block:: python
+
+ # Create hook
+ telegram_hook = TelegramHook(telegram_conn_id='telegram_default')
+ # or telegram_hook = TelegramHook(telegram_conn_id='telegram_default', chat_id='-1xxx')
+ # or telegram_hook = TelegramHook(token='xxx:xxx', chat_id='-1xxx')
+
+ # Call method from telegram bot client
+ telegram_hook.call(None', {"text": "message", "chat_id": "-1xxx"})
+ # or telegram_hook.call(None', {"text": "message"})
+
+ :param telegram_conn_id: connection that optionally has Telegram API token in the password field
+ :type telegram_conn_id: str
+ :param token: optional telegram API token
+ :type token: str
+ :param chat_id: optional chat_id of the telegram chat/channel/group
+ :type chat_id: str
+ """
+
+ def __init__(
+ self,
+ telegram_conn_id: Optional[str] = None,
+ token: Optional[str] = None,
+ chat_id: Optional[str] = None,
+ ) -> None:
+ super().__init__()
+ self.token = self.__get_token(token, telegram_conn_id)
+ self.chat_id = self.__get_chat_id(chat_id, telegram_conn_id)
+ self.connection = self.get_conn()
+
+ def get_conn(self) -> telegram.bot.Bot:
+ """
+ Returns the telegram bot client
+
+ :return: telegram bot client
+ :rtype: telegram.bot.Bot
+ """
+ return telegram.bot.Bot(token=self.token)
+
+ def __get_token(self, token: Optional[str], telegram_conn_id: str) -> str:
+ """
+ Returns the telegram API token
+
+ :param token: telegram API token
+ :type token: str
+ :param telegram_conn_id: telegram connection name
+ :type telegram_conn_id: str
+ :return: telegram API token
+ :rtype: str
+ """
+ if token is not None:
+ return token
+
+ if telegram_conn_id is not None:
+ conn = self.get_connection(telegram_conn_id)
+
+ if not conn.password:
+ raise AirflowException("Missing token(password) in Telegram connection")
+
+ return conn.password
+
+ raise AirflowException("Cannot get token: No valid Telegram connection supplied.")
+
+ def __get_chat_id(self, chat_id: Optional[str], telegram_conn_id: str) -> Optional[str]:
+ """
+ Returns the telegram chat ID for a chat/channel/group
+
+ :param chat_id: optional chat ID
+ :type chat_id: str
+ :param telegram_conn_id: telegram connection name
+ :type telegram_conn_id: str
+ :return: telegram chat ID
+ :rtype: str
+ """
+ if chat_id is not None:
+ return chat_id
+
+ if telegram_conn_id is not None:
+ conn = self.get_connection(telegram_conn_id)
+ return conn.host
+
+ return None
+
+ @tenacity.retry(
+ retry=tenacity.retry_if_exception_type(telegram.error.TelegramError),
+ stop=tenacity.stop_after_attempt(5),
+ wait=tenacity.wait_fixed(1),
+ )
+ def call(self, api_params: dict) -> None:
+ """
+ Sends the message to a telegram channel or chat.
+
+ :param method: not used
+ :type method: str
+ :param api_params: params for telegram_instance.send_message. It can also be used to override chat_id
+ :type api_params: dict
+ """
+ kwargs = {
+ "chat_id": self.chat_id,
+ "parse_mode": telegram.parsemode.ParseMode.HTML,
+ "disable_web_page_preview": True,
+ }
+ kwargs.update(api_params)
+
+ if 'text' not in kwargs or kwargs['text'] is None:
+ raise AirflowException("'text' must be provided for telegram message")
+
+ if kwargs['chat_id'] is None:
+ raise AirflowException("'chat_id' must be provided for telegram message")
+
+ self.log.debug(self.connection.send_message(**kwargs))
Review comment:
??? Is the wrapping in the DEBUG call intentional?
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
users@infra.apache.org