Try fix broadcast to use data from peer db

Signed-off-by: Yasir Aris M <git@yasirdev.my.id>
This commit is contained in:
Yasir Aris M 2023-10-09 22:15:48 +07:00
parent e8d99b5190
commit c934ca1e40
3 changed files with 12 additions and 9 deletions

View file

@ -7,7 +7,7 @@ class UsersData:
def __init__(self, uri, database_name): def __init__(self, uri, database_name):
self._client = AsyncClient(uri) self._client = AsyncClient(uri)
self.db = self._client[database_name] self.db = self._client[database_name]
self.col = self.db["users"] self.col = self.db["userlist"]
self.grp = self.db["groups"] self.grp = self.db["groups"]
@staticmethod @staticmethod

View file

@ -5,27 +5,30 @@ import time
from pyrogram import filters from pyrogram import filters
from pyrogram.types import Message from pyrogram.types import Message
from database.users_chats_db import db from async_pymongo import AsyncClient
from misskaty import app from misskaty import app, DATABASE_URI
from misskaty.vars import SUDO from misskaty.vars import SUDO
from utils import broadcast_messages from utils import broadcast_messages
@app.on_message(filters.command("broadcast") & filters.user(SUDO) & filters.reply) @app.on_message(filters.command("broadcast") & filters.user(SUDO) & filters.reply)
async def broadcast(_, ctx: Message): async def broadcast(_, ctx: Message):
users = await db.get_all_users() mongo = AsyncClient(DATABASE_URI)
userdb = mongo["MissKatyBot"]["peers"]
b_msg = ctx.reply_to_message b_msg = ctx.reply_to_message
sts = await ctx.reply_msg("Broadcasting your messages...") sts = await ctx.reply_msg("Broadcasting your messages...")
start_time = time.time() start_time = time.time()
total_users = await db.total_users_count() total_users = await userdb.count_documents({})
done = 0 done = 0
blocked = 0 blocked = 0
deleted = 0 deleted = 0
failed = 0 failed = 0
success = 0 success = 0
async for user in users: async for chat in userdb.find({}):
pti, sh = await broadcast_messages(int(user["id"]), b_msg) if chat["type"] == "user":
continue
pti, sh = await broadcast_messages(int(chat["id"]), b_msg)
if pti: if pti:
success += 1 success += 1
elif pti is False: elif pti is False:

View file

@ -308,8 +308,8 @@ async def getDataNodrakor(msg, kueri, CurrentPage, user, strings):
) )
return None, 0, None return None, 0, None
for i in entry: for i in entry:
genre = i.find(class_="gmr-movie-on").text genre = i.find(class_="gmr-movie-on")
genre = f"{genre}" if genre != "" else "N/A" genre = f"{genre.text}" if genre != "" else "N/A"
judul = i.find(class_="entry-title").find("a").text judul = i.find(class_="entry-title").find("a").text
link = i.find(class_="entry-title").find("a").get("href") link = i.find(class_="entry-title").find("a").get("href")
nodrakordata.append({"judul": judul, "link": link, "genre": genre}) nodrakordata.append({"judul": judul, "link": link, "genre": genre})