adjust according to core

This commit is contained in:
thedragonsinn
2025-03-10 18:24:29 +05:30
parent 70629d6495
commit b506aad02d
8 changed files with 9 additions and 12 deletions

View File

@@ -42,7 +42,7 @@ async def test_function(bot: BOT, message: Message):
```python
from app import BOT, bot, Message, CustomDB
TEST_COLLECTION = CustomDB("TEST_COLLECTION")
TEST_COLLECTION = CustomDB["TEST_COLLECTION"]
@bot.add_cmd(cmd="add_data")
async def test_function(bot: BOT, message: Message):

View File

@@ -1,7 +1,5 @@
from ub_core import (
BOT,
DATABASE,
DATABASE_CLIENT,
LOGGER,
Config,
Convo,

View File

@@ -10,7 +10,7 @@ from app import BOT, Config, CustomDB, Message, bot, extra_config
FBAN_TASK_LOCK = asyncio.Lock()
FED_DB = CustomDB("FED_LIST")
FED_DB = CustomDB["FED_LIST"]
BASIC_FILTER = filters.user([609517172, 2059887769]) & ~filters.service

View File

@@ -9,13 +9,12 @@ from google.genai.types import (
SafetySetting,
Tool,
)
from pyrogram import filters
from app import BOT, CustomDB, Message, extra_config
logging.getLogger("google_genai.models").setLevel(logging.WARNING)
DB_SETTINGS = CustomDB("COMMON_SETTINGS")
DB_SETTINGS = CustomDB["COMMON_SETTINGS"]
try:
client: Client = Client(api_key=extra_config.GEMINI_API_KEY)

View File

@@ -1,6 +1,6 @@
from app import BOT, Config, CustomDB, Message
DB = CustomDB("SUDO_CMD_LIST")
DB = CustomDB["SUDO_CMD_LIST"]
async def init_task():

View File

@@ -3,8 +3,8 @@ from ub_core.utils.helpers import extract_user_data, get_name
from app import BOT, Config, CustomDB, Message
SUDO = CustomDB("COMMON_SETTINGS")
SUDO_USERS = CustomDB("SUDO_USERS")
SUDO = CustomDB["COMMON_SETTINGS"]
SUDO_USERS = CustomDB["SUDO_USERS"]
async def init_task():

View File

@@ -8,7 +8,7 @@ from ub_core.utils.helpers import get_name
from app import BOT, Config, CustomDB, Message, bot, extra_config
LOGGER = CustomDB("COMMON_SETTINGS")
LOGGER = CustomDB["COMMON_SETTINGS"]
MESSAGE_CACHE: dict[int, list[Message]] = defaultdict(list)
FLOOD_LIST: list[int] = []

View File

@@ -7,8 +7,8 @@ from ub_core.utils.helpers import get_name
from app import BOT, CustomDB, Message, bot, extra_config
PM_USERS = CustomDB("PM_USERS")
PM_GUARD = CustomDB("COMMON_SETTINGS")
PM_USERS = CustomDB["PM_USERS"]
PM_GUARD = CustomDB["COMMON_SETTINGS"]
ALLOWED_USERS: list[int] = []
RECENT_USERS: dict = defaultdict(int)