[PATCH v1 14/18] Add delete_broadcast() in scraper/db.py

[Muhammad Rizki <kiizuha@gnuweeb.org>]

Add delete_broadcast() for delete a chat in the database from receiving
email messages.

Signed-off-by: Muhammad Rizki <kiizuha@gnuweeb.org>
---
 daemon/scraper/db.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/daemon/scraper/db.py b/daemon/scraper/db.py
index 805b591..6faedb6 100644
--- a/daemon/scraper/db.py
+++ b/daemon/scraper/db.py
@@ -199,6 +199,24 @@ class Db():
 		return self.cur.lastrowid
 
 
+	def delete_broadcast(self, chat_id: Union[int, str]):
+		get_query = """
+			SELECT id FROM broadcast_chats
+			WHERE chat_id = %(chat_id)s
+		"""
+		delete_query = """
+			DELETE FROM broadcast_chats
+			WHERE chat_id = %(chat_id)s
+		"""
+		self.cur.execute(get_query, {"chat_id": chat_id})
+		exist = self.cur.fetchone()
+		if bool(exist):
+			self.cur.execute(delete_query, {"chat_id": chat_id})
+			return True
+
+		return False
+
+
 	def get_broadcast_chats(self):
 		q = """
 			SELECT *
-- 
Muhammad Rizki