Spaces:
Paused
Paused
| import time | |
| from asyncio import sleep | |
| from traceback import format_exc | |
| from apscheduler.schedulers.asyncio import AsyncIOScheduler | |
| from pyrogram.enums import ChatMemberStatus as CMS | |
| from pyrogram.errors import PeerIdInvalid, UserNotParticipant | |
| from Powers import LOGGER, MESSAGE_DUMP, TIME_ZONE | |
| from Powers.bot_class import Gojo | |
| from Powers.database.approve_db import Approve | |
| from Powers.database.blacklist_db import Blacklist | |
| from Powers.database.chats_db import Chats | |
| from Powers.database.disable_db import Disabling | |
| from Powers.database.filters_db import Filters | |
| from Powers.database.flood_db import Floods | |
| from Powers.database.greetings_db import Greetings | |
| from Powers.database.notes_db import Notes, NotesSettings | |
| from Powers.database.pins_db import Pins | |
| from Powers.database.reporting_db import Reporting | |
| # from Powers.database.users_db import Users | |
| from Powers.database.warns_db import Warns, WarnSettings | |
| from Powers.utils.custom_filters import command | |
| from Powers.vars import Config | |
| async def clean_my_db(c:Gojo,is_cmd=False, id=None): | |
| to_clean = list() | |
| chats_list = Chats.list_chats_by_id() | |
| to_clean.clear() | |
| start = time.time() | |
| for chats in chats_list: | |
| try: | |
| stat = await c.get_chat_member(chat_id=chats,user_id=Config.BOT_ID) | |
| if stat.status not in [CMS.MEMBER, CMS.ADMINISTRATOR, CMS.OWNER]: | |
| to_clean.append(chats) | |
| except UserNotParticipant: | |
| to_clean.append(chats) | |
| except Exception as e: | |
| LOGGER.error(e) | |
| LOGGER.error(format_exc()) | |
| if not is_cmd: | |
| return e | |
| else: | |
| to_clean.append(chats) | |
| for i in to_clean: | |
| Approve(i).clean_approve() | |
| Blacklist(i).clean_blacklist() | |
| Chats.remove_chat(i) | |
| Disabling(i).clean_disable() | |
| Filters().rm_all_filters(i) | |
| Floods().rm_flood(i) | |
| Greetings(i).clean_greetings() | |
| Notes().rm_all_notes(i) | |
| NotesSettings().clean_notes(i) | |
| Pins(i).clean_pins() | |
| Reporting(i).clean_reporting() | |
| Warns(i).clean_warn() | |
| WarnSettings(i).clean_warns() | |
| x = len(to_clean) | |
| txt = f"#INFO\n\nCleaned db:\nTotal chats removed: {x}" | |
| to_clean.clear() | |
| nums = time.time()-start | |
| if is_cmd: | |
| txt += f"\nClean type: Forced\nInitiated by: {(await c.get_users(user_ids=id)).mention}" | |
| txt += f"\nClean type: Manual\n\tTook {round(nums,2)} seconds to complete the process" | |
| await c.send_message(chat_id=MESSAGE_DUMP,text=txt) | |
| return txt | |
| else: | |
| txt += f"\nClean type: Auto\n\tTook {round(nums,2)} seconds to complete the process" | |
| await c.send_message(chat_id=MESSAGE_DUMP,text=txt) | |
| return txt | |