adjust according to core
This commit is contained in:
@@ -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):
|
||||
|
||||
@@ -1,7 +1,5 @@
|
||||
from ub_core import (
|
||||
BOT,
|
||||
DATABASE,
|
||||
DATABASE_CLIENT,
|
||||
LOGGER,
|
||||
Config,
|
||||
Convo,
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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)
|
||||
|
||||
@@ -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():
|
||||
|
||||
@@ -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():
|
||||
|
||||
@@ -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] = []
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user