From 2e57d481916fbd78b1b3ca2c5446b09828562bbd Mon Sep 17 00:00:00 2001
From: "Andrey \"Limych\" Khrolenok" <andrey@khrolenok.ru>
Date: Wed, 17 Apr 2019 20:15:33 +0300
Subject: [PATCH] Adding Telegram bot leave_chat() service (#22259)

* Adding leave_chat() service

* Fix indent
---
 homeassistant/components/telegram_bot/__init__.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/homeassistant/components/telegram_bot/__init__.py b/homeassistant/components/telegram_bot/__init__.py
index fe582867bce..43f8a26644c 100644
--- a/homeassistant/components/telegram_bot/__init__.py
+++ b/homeassistant/components/telegram_bot/__init__.py
@@ -67,6 +67,7 @@ SERVICE_EDIT_CAPTION = 'edit_caption'
 SERVICE_EDIT_REPLYMARKUP = 'edit_replymarkup'
 SERVICE_ANSWER_CALLBACK_QUERY = 'answer_callback_query'
 SERVICE_DELETE_MESSAGE = 'delete_message'
+SERVICE_LEAVE_CHAT = 'leave_chat'
 
 EVENT_TELEGRAM_CALLBACK = 'telegram_callback'
 EVENT_TELEGRAM_COMMAND = 'telegram_command'
@@ -167,6 +168,10 @@ SERVICE_SCHEMA_DELETE_MESSAGE = vol.Schema({
         vol.Any(cv.positive_int, vol.All(cv.string, 'last')),
 }, extra=vol.ALLOW_EXTRA)
 
+SERVICE_SCHEMA_LEAVE_CHAT = vol.Schema({
+    vol.Required(ATTR_CHAT_ID): vol.Coerce(int),
+})
+
 SERVICE_MAP = {
     SERVICE_SEND_MESSAGE: SERVICE_SCHEMA_SEND_MESSAGE,
     SERVICE_SEND_PHOTO: SERVICE_SCHEMA_SEND_FILE,
@@ -179,6 +184,7 @@ SERVICE_MAP = {
     SERVICE_EDIT_REPLYMARKUP: SERVICE_SCHEMA_EDIT_REPLYMARKUP,
     SERVICE_ANSWER_CALLBACK_QUERY: SERVICE_SCHEMA_ANSWER_CALLBACK_QUERY,
     SERVICE_DELETE_MESSAGE: SERVICE_SCHEMA_DELETE_MESSAGE,
+    SERVICE_LEAVE_CHAT: SERVICE_SCHEMA_LEAVE_CHAT,
 }
 
 
@@ -580,6 +586,15 @@ class TelegramNotificationService:
                            chat_id=chat_id,
                            latitude=latitude, longitude=longitude, **params)
 
+    def leave_chat(self, chat_id=None):
+        """Remove bot from chat."""
+        chat_id = self._get_target_chat_ids(chat_id)[0]
+        _LOGGER.debug("Leave from chat ID %s", chat_id)
+        leaved = self._send_msg(self.bot.leaveChat,
+                                "Error leaving chat",
+                                chat_id)
+        return leaved
+
 
 class BaseTelegramBotEntity:
     """The base class for the telegram bot."""