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 ```python
from app import BOT, bot, Message, CustomDB from app import BOT, bot, Message, CustomDB
TEST_COLLECTION = CustomDB("TEST_COLLECTION") TEST_COLLECTION = CustomDB["TEST_COLLECTION"]
@bot.add_cmd(cmd="add_data") @bot.add_cmd(cmd="add_data")
async def test_function(bot: BOT, message: Message): async def test_function(bot: BOT, message: Message):

View File

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

View File

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

View File

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

View File

@@ -1,6 +1,6 @@
from app import BOT, Config, CustomDB, Message from app import BOT, Config, CustomDB, Message
DB = CustomDB("SUDO_CMD_LIST") DB = CustomDB["SUDO_CMD_LIST"]
async def init_task(): 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 from app import BOT, Config, CustomDB, Message
SUDO = CustomDB("COMMON_SETTINGS") SUDO = CustomDB["COMMON_SETTINGS"]
SUDO_USERS = CustomDB("SUDO_USERS") SUDO_USERS = CustomDB["SUDO_USERS"]
async def init_task(): 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 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) MESSAGE_CACHE: dict[int, list[Message]] = defaultdict(list)
FLOOD_LIST: list[int] = [] 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 from app import BOT, CustomDB, Message, bot, extra_config
PM_USERS = CustomDB("PM_USERS") PM_USERS = CustomDB["PM_USERS"]
PM_GUARD = CustomDB("COMMON_SETTINGS") PM_GUARD = CustomDB["COMMON_SETTINGS"]
ALLOWED_USERS: list[int] = [] ALLOWED_USERS: list[int] = []
RECENT_USERS: dict = defaultdict(int) RECENT_USERS: dict = defaultdict(int)