apache-airflow-providers-telegram 4.4.0rc1__tar.gz → 4.4.0rc2__tar.gz

{apache_airflow_providers_telegram-4.4.0rc1 → apache_airflow_providers_telegram-4.4.0rc2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: apache-airflow-providers-telegram
-Version: 4.4.0rc1
+Version: 4.4.0rc2
 Summary: Provider package apache-airflow-providers-telegram for Apache Airflow
 Keywords: airflow-provider,telegram,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
@@ -74,7 +74,7 @@ Project-URL: YouTube, https://www.youtube.com/channel/UCSXwxpWZQ7XZ1WL3wqevChA/
 
 Package ``apache-airflow-providers-telegram``
 
-Release: ``4.4.0.rc1``
+Release: ``4.4.0.rc2``
 
 
 `Telegram <https://telegram.org/>`__

{apache_airflow_providers_telegram-4.4.0rc1 → apache_airflow_providers_telegram-4.4.0rc2}/README.rst

@@ -42,7 +42,7 @@
 
 Package ``apache-airflow-providers-telegram``
 
-Release: ``4.4.0.rc1``
+Release: ``4.4.0.rc2``
 
 
 `Telegram <https://telegram.org/>`__

{apache_airflow_providers_telegram-4.4.0rc1 → apache_airflow_providers_telegram-4.4.0rc2}/airflow/providers/telegram/get_provider_info.py

@@ -28,9 +28,10 @@ def get_provider_info():
         "name": "Telegram",
         "description": "`Telegram <https://telegram.org/>`__\n",
         "state": "ready",
-        "source-date-epoch": 1705912297,
+        "source-date-epoch": 1709556330,
         "versions": [
             "4.4.0",
+            "4.3.1",
             "4.3.0",
             "4.2.0",
             "4.1.1",
@@ -64,6 +65,12 @@ def get_provider_info():
                 "python-modules": ["airflow.providers.telegram.operators.telegram"],
             }
         ],
+        "connection-types": [
+            {
+                "hook-class-name": "airflow.providers.telegram.hooks.telegram.TelegramHook",
+                "connection-type": "telegram",
+            }
+        ],
         "hooks": [
             {"integration-name": "Telegram", "python-modules": ["airflow.providers.telegram.hooks.telegram"]}
         ],

{apache_airflow_providers_telegram-4.4.0rc1 → apache_airflow_providers_telegram-4.4.0rc2}/airflow/providers/telegram/hooks/telegram.py

@@ -19,6 +19,7 @@
 from __future__ import annotations
 
 import asyncio
+from typing import Any
 
 import telegram
 import tenacity
@@ -38,14 +39,15 @@ class TelegramHook(BaseHook):
     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'
+    conn_type: 'telegram'
+    password: 'TELEGRAM_TOKEN' (optional)
     host: 'chat_id' (optional)
     Examples:
     .. code-block:: python
 
         # Create hook
         telegram_hook = TelegramHook(telegram_conn_id="telegram_default")
+        telegram_hook = TelegramHook()  # will use telegram_default
         # or telegram_hook = TelegramHook(telegram_conn_id='telegram_default', chat_id='-1xxx')
         # or telegram_hook = TelegramHook(token='xxx:xxx', chat_id='-1xxx')
 
@@ -58,21 +60,33 @@ class TelegramHook(BaseHook):
     :param chat_id: optional chat_id of the telegram chat/channel/group
     """
 
+    conn_name_attr = "telegram_conn_id"
+    default_conn_name = "telegram_default"
+    conn_type = "telegram"
+    hook_name = "Telegram"
+
     def __init__(
         self,
-        telegram_conn_id: str | None = None,
+        telegram_conn_id: str | None = default_conn_name,
         token: str | None = None,
         chat_id: str | None = None,
-        **kwargs,
     ) -> None:
-        super().__init__(**kwargs)
+        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()
 
+    @classmethod
+    def get_ui_field_behaviour(cls) -> dict[str, Any]:
+        """Return custom field behaviour."""
+        return {
+            "hidden_fields": ["schema", "extra", "login", "port", "extra"],
+            "relabeling": {},
+        }
+
     def get_conn(self) -> telegram.Bot:
         """
-        Returns the telegram bot client.
+        Return the telegram bot client.
 
         :return: telegram bot client
         """
@@ -80,7 +94,7 @@ class TelegramHook(BaseHook):
 
     def __get_token(self, token: str | None, telegram_conn_id: str | None) -> str:
         """
-        Returns the telegram API token.
+        Return the telegram API token.
 
         :param token: telegram API token
         :param telegram_conn_id: telegram connection name
@@ -101,7 +115,7 @@ class TelegramHook(BaseHook):
 
     def __get_chat_id(self, chat_id: str | None, telegram_conn_id: str | None) -> str | None:
         """
-        Returns the telegram chat ID for a chat/channel/group.
+        Return the telegram chat ID for a chat/channel/group.
 
         :param chat_id: optional chat ID
         :param telegram_conn_id: telegram connection name
@@ -123,7 +137,7 @@ class TelegramHook(BaseHook):
     )
     def send_message(self, api_params: dict) -> None:
         """
-        Sends the message to a telegram channel or chat.
+        Send the message to a telegram channel or chat.
 
         :param api_params: params for telegram_instance.send_message. It can also be used to override chat_id
         """

{apache_airflow_providers_telegram-4.4.0rc1 → apache_airflow_providers_telegram-4.4.0rc2}/airflow/providers/telegram/operators/telegram.py

@@ -72,7 +72,7 @@ class TelegramOperator(BaseOperator):
         super().__init__(**kwargs)
 
     def execute(self, context: Context) -> None:
-        """Calls the TelegramHook to post the provided Telegram message."""
+        """Call the TelegramHook to post the provided Telegram message."""
         if self.text:
             self.telegram_kwargs["text"] = self.text
 

{apache_airflow_providers_telegram-4.4.0rc1 → apache_airflow_providers_telegram-4.4.0rc2}/pyproject.toml

@@ -28,7 +28,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "apache-airflow-providers-telegram"
-version = "4.4.0.rc1"
+version = "4.4.0.rc2"
 description = "Provider package apache-airflow-providers-telegram for Apache Airflow"
 readme = "README.rst"
 authors = [