Spaces:
Runtime error
Runtime error
File size: 2,644 Bytes
78b07ad |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 |
import sys
from .clients import hellbot
from .config import Config, Symbols
from .database import db
from .logger import LOGS
async def _AuthUsers() -> None:
temp_list = []
temp_list.append(Config.OWNER_ID)
temp_list.extend([(await client.get_me()).id for client in hellbot.users])
stan_users = await db.get_all_stans()
for user in stan_users:
temp_list.append(user["user_id"])
users = list(set(temp_list))
for user in users:
Config.AUTH_USERS.add(user)
temp_list = None
LOGS.info(
f"{Symbols.arrow_right * 2} Added Authorized Users {Symbols.arrow_left * 2}"
)
async def _StanUsers() -> None:
users = await db.get_all_stans()
for user in users:
Config.STAN_USERS.add(user["user_id"])
LOGS.info(f"{Symbols.arrow_right * 2} Added Stan Users {Symbols.arrow_left * 2}")
async def _GbanUsers() -> None:
users = await db.get_gban()
for user in users:
Config.BANNED_USERS.add(user["user_id"])
LOGS.info(
f"{Symbols.arrow_right * 2} Added {len(users)} Gbanned Users {Symbols.arrow_left * 2}"
)
musers = await db.get_gmute()
for user in musers:
Config.MUTED_USERS.add(user["user_id"])
LOGS.info(
f"{Symbols.arrow_right * 2} Added {len(musers)} Gmuted Users {Symbols.arrow_left * 2}"
)
async def UserSetup() -> None:
"""Initialize Users Config"""
LOGS.info(f"{Symbols.bullet * 3} Setting Up Users {Symbols.bullet * 3}")
await _AuthUsers()
await _StanUsers()
await _GbanUsers()
async def ForcesubSetup() -> None:
"""Initialize Forcesub Config"""
chats = await db.get_all_forcesubs()
for chat in chats:
if chat not in Config.FORCESUBS:
Config.FORCESUBS.add(chat["chat"])
async def GachaBotsSetup() -> None:
"""Initialize GachaBots Config"""
bots = await db.get_all_gachabots_id()
for bot in bots:
Config.GACHA_BOTS.add(bot)
async def TemplateSetup() -> None:
"""Initialize Templates Config"""
module_name = "temp_module"
module = sys.modules.get(module_name)
if module is None:
module = type(sys)(module_name)
with open("Hellbot/functions/templates.py", "r", encoding="utf-8") as file:
exec(file.read(), module.__dict__)
global_vars = module.__dict__
var_n_value: dict[str, str] = {
var_name: global_vars[var_name][0]
for var_name in global_vars
if var_name.isupper() and not callable(global_vars[var_name])
}
Config.TEMPLATES = var_n_value
|