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