Stable IMDB Lang For Every User

This commit is contained in:
yasir 2023-01-10 13:20:54 +07:00
parent cf255eaafb
commit f4ed6367e1
2 changed files with 12 additions and 3 deletions

View file

@ -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)
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})

View file

@ -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 = []