From f4ed6367e119860d9d361222433243d071ed6b36 Mon Sep 17 00:00:00 2001
From: yasir <yasiramunandar@gmail.com>
Date: Tue, 10 Jan 2023 13:20:54 +0700
Subject: [PATCH] Stable IMDB Lang For Every User

---
 database/imdb_db.py             | 7 ++++++-
 misskaty/plugins/imdb_search.py | 8 ++++++--
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/database/imdb_db.py b/database/imdb_db.py
index de25290a..ad805797 100644
--- a/database/imdb_db.py
+++ b/database/imdb_db.py
@@ -8,4 +8,9 @@ async def is_imdbset(user_id: int) -> bool:
 
 
 async def add_imdbset(user_id: int, lang):
-    await imbd_db.update_one({"user_id": user_id}, {"$set": {"lang": lang}}, upsert=True)
\ No newline at end of file
+    await imbd_db.update_one({"user_id": user_id}, {"$set": {"lang": lang}}, upsert=True)
+
+async def remove_imdbset(user_id: int):
+    user = await imbd_db.find_one({"user_id": user_id})
+    if user:
+        return await imbd_db.delete_one({"user_id": user_id})
\ No newline at end of file
diff --git a/misskaty/plugins/imdb_search.py b/misskaty/plugins/imdb_search.py
index 901d0d3d..b27b3f9f 100644
--- a/misskaty/plugins/imdb_search.py
+++ b/misskaty/plugins/imdb_search.py
@@ -3,7 +3,7 @@ import logging
 import re
 
 from bs4 import BeautifulSoup
-from database.imdb_db import add_imdbset, is_imdbset
+from database.imdb_db import *
 from deep_translator import GoogleTranslator
 from pykeyboard import InlineButton, InlineKeyboard
 from pyrogram import filters
@@ -68,6 +68,7 @@ async def imdbsetlang(client, query):
     buttons.row(
         InlineButton("🇺🇸 English", f"setimdb#eng#{query.from_user.id}"),
         InlineButton("🇮🇩 Indonesia", f"setimdb#ind#{query.from_user.id}")
+        InlineButton("🗑 Remove UserSetting", f"setimdb#rm#{query.from_user.id}")
     )
     buttons.row(
         InlineButton("❌ Close", f"close#{query.from_user.id}")
@@ -82,9 +83,12 @@ async def imdbsetlang(client, query):
     if lang == "eng":
         await add_imdbset(query.from_user.id, lang)
         await query.message.edit_caption("Language interface for IMDB has been changed to English.")
-    else:
+    elif lang == "ind":
         await add_imdbset(query.from_user.id, lang)
         await query.message.edit_caption("Bahasa tampilan IMDB sudah diubah ke Indonesia.")
+    else:
+        await remove_imdbset(query.from_user.id)
+        await query.message.edit_caption("UserSetting for IMDB has been deleted from database.")
 
 async def imdb_search_id(kueri, message):
     BTN = []