def decorator(func):
async def new_func(*def_args, **def_kwargs):
message = def_args[0]
if cmds:
message.conf["cmds"] = cmds
if allow_kwargs is False:
def_kwargs = dict()
with configure_scope() as scope:
parsed_update = parse_update(dict(message))
scope.set_extra("update", str(parsed_update))
if DEBUG_MODE:
# log.debug('[*] Starting {}.'.format(func.__name__))
# log.debug('Event: \n' + str(message))
start = time.time()
await func(*def_args, **def_kwargs)
log.debug("[*] {} Time: {} sec.".format(
func.__name__,
time.time() - start))
else:
await func(*def_args, **def_kwargs)
raise SkipHandler()
if f == "cb":
dp.register_callback_query_handler(new_func, *args,
**register_kwargs)
else:
dp.register_message_handler(new_func, *args, **register_kwargs)
if allow_edited is True:
dp.register_edited_message_handler(new_func, *args,
**register_kwargs)
async def sbroadcast(message):
data = await get_parsed_note_list(message, split_args=-1)
dp.register_message_handler(check_message_for_smartbroadcast)
await db.sbroadcast.drop({})
chats = mongodb.chat_list.distinct("chat_id")
data["chats_num"] = len(chats)
data["recived_chats"] = 0
data["chats"] = chats
await db.sbroadcast.insert_one(data)
await message.reply(
"Smart broadcast planned for <code>{}</code> chats".format(len(chats)))
async def continue_sbroadcast(message):
dp.register_message_handler(check_message_for_smartbroadcast)
return await message.reply("Re-registered the broadcast handler.")
