def register_handlers_common(dp: Dispatcher, admin_id: int):
dp.register_message_handler(cmd_start_help,
commands=['start', 'help'],
state='*')
dp.register_message_handler(cmd_start_help,
Text(startswith='⁉', ignore_case=True),
state='*')
dp.register_message_handler(day_handler,
Text(startswith=['🕌', '🕋']),
state='*')
dp.register_message_handler(date_handler, Text(startswith='📆'), state='*'),
dp.register_callback_query_handler(process_simple_calendar,
simple_cal_callback.filter(),
state='*')
dp.register_message_handler(next_handler, Text(startswith='⏰'), state='*')
def register_handlers(dp: Dispatcher):
dp.register_message_handler(cmd_cancel, commands="cancel", state="*")
dp.register_message_handler(cmd_cancel,
Text(equals="отмена", ignore_case=True),
state="*")
dp.register_message_handler(cmd_start, commands="start", state="*")
dp.register_message_handler(cmd_start,
Text(equals="привет", ignore_case=True),
state="*")
dp.register_message_handler(cmd_begin, commands="begin", state="*")
dp.register_message_handler(get_players_names,
state=States.waiting_for_players_list)
dp.register_message_handler(get_teams_number,
state=States.waiting_for_teams_number)
dp.register_message_handler(user_confirm,
state=States.waiting_for_user_confirm)
def bot():
client = MongoClient(mongoDbKey)
mydb = client['videmaker']
db = mydb['themes']
logging.basicConfig(level=logging.INFO)
bot = Bot(token=telegramKey)
dp = Dispatcher(bot)
content = {'theme': '', 'date': '', 'ready': 0}
@dp.message_handler(commands=['help'])
async def sendHelp(message: types.Message):
await message.reply('Me envie um tema para um video')
await message.reply('Exemplo: Tema god of war')
@dp.message_handler(Text(contains='Tema', ignore_case=True), state='*')
async def getTheme(message: types.Message):
try:
removeTheme = message.text.split()[1:]
theme = ' '.join(removeTheme)
content['theme'] = theme
content['date'] = datetime.datetime.utcnow()
db.update_one({'_id': theme}, {'$set': content}, upsert=True)
theme = f'Seu tema é {theme}'
except TypeError:
theme = "Erro"
print(theme)
await message.reply(theme)
executor.start_polling(dp, skip_updates=True)
def register_handlers_common(dp: Dispatcher):
"""Register command in this file."""
dp.register_message_handler(cmd_start, commands="start", state="*")
dp.register_message_handler(cmd_help, commands="help", state="*")
dp.register_message_handler(cmd_cancel, commands="cancel", state="*")
dp.register_message_handler(cmd_cancel,
Text(equals="Отмена", ignore_case=True),
state="*")
def register_handlers_login(dp: Dispatcher) -> None:
dp.register_message_handler(login_start, commands="login", state="*")
dp.register_message_handler(login_start,
Text(equals="вход", ignore_case=True),
state="*")
dp.register_message_handler(login_nickname,
state=OrderLogin.waiting_for_nickname)
dp.register_message_handler(login_password,
state=OrderLogin.waiting_for_password)
def register_handlers(dp: Dispatcher):
dp.register_message_handler(info_handler, commands='info')
dp.register_message_handler(contacts_handler, commands='contacts')
dp.register_message_handler(cancel_handler, state='*', commands='cancel')
dp.register_message_handler(cancel_handler, Text(equals='отмена', ignore_case=True), state='*')
dp.register_callback_query_handler(cancel_callback_handler, lambda callback_query: callback_query.data == "cancel",
state='*')
def register_handlers_common(dp: Dispatcher, admin_id: int):
dp.register_message_handler(cmd_start, commands="start", state="*")
dp.register_message_handler(cmd_cancel, commands="cancel", state="*")
dp.register_message_handler(cmd_cancel,
Text(equals="отмена", ignore_case=True),
state="*")
dp.register_message_handler(secret_command,
IDFilter(user_id=admin_id),
commands="abracadabra")
def register_handlers_fsm(dp: Dispatcher):
# text state
dp.register_message_handler(process_text,state=Form.text)
# Geo state
dp.register_message_handler(process_geo_text_check,Text(equals="Вести вручную", ignore_case=False), state=Form.geo)
dp.register_message_handler(process_geo_text, state=Form.geo,content_types=ContentType.TEXT)
dp.register_message_handler(process_geo_link, state=Form.geo,content_types=ContentType.LOCATION)
# Photo get date
dp.register_message_handler(process_photo,state=Form.photo, content_types=ContentType.PHOTO)
def register_handlers_common(dp: Dispatcher):
"""Register common handlers in Dispatcher."""
log.info('Configuring common handlers...')
dp.register_message_handler(common.cmd_start, commands='start')
dp.register_message_handler(common.cmd_about, commands='about')
dp.register_message_handler(common.cmd_help, commands='help')
dp.register_message_handler(common.cmd_cancel, commands='cancel', state='*')
dp.register_message_handler(
common.cmd_cancel, Text(equals='отмена', ignore_case=True), state='*'
)
def register_handlers_common(dp: Dispatcher):
""" Регистрация message handler для бота """
dp.register_message_handler(send_welcome,
commands=['start', 'help'],
state="*")
dp.register_message_handler(cancel_handler, state='*', commands='cancel')
dp.register_message_handler(cancel_handler,
Text(equals='cancel', ignore_case=True),
state='*')
def register_handler_add_post(dp: Dispatcher) -> None:
dp.register_message_handler(add_post_start, commands='add_post', state='*')
dp.register_message_handler(add_post_start,
Text(equals='добавить пост', ignore_case=True),
state='*')
dp.register_message_handler(add_post_description,
state=OrderAddPost.waiting_for_description)
dp.register_message_handler(add_post_photo,
content_types='photo',
state=OrderAddPost.waiting_for_photo)
def register_handlers_register(dp: Dispatcher) -> None:
dp.register_message_handler(register_start, commands='register', state='*')
dp.register_message_handler(register_start,
Text(equals='регистрация', ignore_case=True),
state='*')
dp.register_message_handler(register_nickname,
state=OrderRegister.waiting_for_nickname)
dp.register_message_handler(register_password,
state=OrderRegister.waiting_for_password)
dp.register_message_handler(register_password_again,
state=OrderRegister.waiting_for_password_again)
def register_main_group_user(dp: Dispatcher, main_group_id: int):
dp.register_message_handler(error_no_reply,
chat_id=main_group_id,
is_reply=False,
commands=["lapor", "report"])
dp.register_message_handler(cmd_report,
chat_id=main_group_id,
is_reply=True,
commands=["lapor", "report"])
dp.register_message_handler(calling_all_units,
Text(startswith="@lapor", ignore_case=True),
chat_id=main_group_id)
def register_handlers_common(dp: Dispatcher, admin_id: int):
dp.register_message_handler(cmd_process_start,
state='*',
commands=['start'])
dp.register_message_handler(cmd_cancel_process,
state='*',
commands=['cancel'])
dp.register_message_handler(cmd_cancel_process,
Text(equals="отмена", ignore_case=True),
state='*')
dp.register_message_handler(admin_cmd,
IDFilter(user_id=admin_id),
commands=['abracadabra'],
state='*')
async def test_equals_string(self, test_filter_text, test_text, ignore_case):
test_filter = Text(equals=test_filter_text, ignore_case=ignore_case)
async def check(obj):
result = await test_filter.check(obj)
if ignore_case:
_test_filter_text = test_filter_text.lower()
_test_text = test_text.lower()
else:
_test_filter_text = test_filter_text
_test_text = test_text
return result is (_test_text == _test_filter_text)
await self._run_check(check, test_text)
def register_handlers_order(dp: dp):
dp.register_message_handler(cancel_order,
Text(equals=['НАЗАД']),
state='*')
dp.register_callback_query_handler(start_order, Text(equals=['to_order']))
dp.register_message_handler(get_time, state=Cart.time)
dp.register_message_handler(get_name, state=Cart.name)
dp.register_message_handler(get_phone_invalid,
lambda message: not message.text.isdigit(
) and message.text.lower() not in
('назад', 'заказ верен'),
state=Cart.phone)
dp.register_message_handler(get_phone_contact,
state=Cart.phone,
content_types=['contact'])
dp.register_message_handler(get_phone_digit,
state=Cart.phone,
content_types=['text'])
dp.register_message_handler(check_order,
lambda message: message.text.lower() in
('назад', 'заказ верен'),
state=Cart.user_id)
dp.register_callback_query_handler(admin,
order_user_id.filter(user='******'))
async def test_contains_list(self, test_filter_list, test_text, ignore_case):
test_filter = Text(contains=test_filter_list, ignore_case=ignore_case)
async def check(obj):
result = await test_filter.check(obj)
if ignore_case:
_test_filter_list = list(map(str.lower, test_filter_list))
_test_text = test_text.lower()
else:
_test_filter_list = test_filter_list
_test_text = test_text
return result is all(map(_test_text.__contains__, _test_filter_list))
await self._run_check(check, test_text)
async def test_startswith_list(self, test_prefix_list, test_text, ignore_case):
test_filter = Text(startswith=test_prefix_list, ignore_case=ignore_case)
async def check(obj):
result = await test_filter.check(obj)
if ignore_case:
_test_prefix_list = map(str.lower, test_prefix_list)
_test_text = test_text.lower()
else:
_test_prefix_list = test_prefix_list
_test_text = test_text
return result is any(map(_test_text.startswith, _test_prefix_list))
await self._run_check(check, test_text)
async def test_endswith(self, test_postfix, test_text, ignore_case):
test_filter = Text(endswith=test_postfix, ignore_case=ignore_case)
async def check(obj):
result = await test_filter.check(obj)
if ignore_case:
_test_postfix = test_postfix.lower()
_test_text = test_text.lower()
else:
_test_postfix = test_postfix
_test_text = test_text
return result is _test_text.endswith(_test_postfix)
await self._run_check(check, test_text)
async def test_contains(self, test_string, test_text, ignore_case):
test_filter = Text(contains=test_string, ignore_case=ignore_case)
async def check(obj):
result = await test_filter.check(obj)
if ignore_case:
_test_string = test_string.lower()
_test_text = test_text.lower()
else:
_test_string = test_string
_test_text = test_text
return result is (_test_string in _test_text)
await self._run_check(check, test_text)
def register_user(dp: Dispatcher):
dp.register_message_handler(show_about, commands=["about"], state="*")
dp.register_message_handler(set_state, commands=["set"], state="*")
dp.register_message_handler(show_shop, commands=["shop"], state="*")
dp.register_message_handler(user_info,
commands=["me"],
state=UserStatus.access_true)
dp.register_message_handler(secret_code,
Text(equals="пожалуйста", ignore_case=True),
state=UserStatus.waiting_code)
dp.register_message_handler(
start_ref_link,
CommandStart(deep_link=re.compile(r"ref_([\w]+)")),
state="*")
dp.register_message_handler(reading_code, state=UserStatus.waiting_code)
dp.register_message_handler(start_base, commands=["start"], state="*")
def register_cart(dp: Dispatcher):
dp.register_callback_query_handler(show_item, Text(startswith="show_item_"), state=UserStatus.access_true)
dp.register_callback_query_handler(add_item_to_cart, Text(startswith="add_item_"), state=UserStatus.access_true)
dp.register_message_handler(show_cart, commands=["cart"], state=UserStatus.access_true)
dp.register_callback_query_handler(cart_clear, Text(equals="cart_clear"), state="*")
dp.register_callback_query_handler(cart_buy, Text(equals="cart_buy"), state="*")
dp.register_message_handler(cart_cancel, Text(equals="отмена", ignore_case=True), state=UserStatus.waiting_address)
dp.register_message_handler(add_address, state=UserStatus.waiting_address)
dp.register_callback_query_handler(check_payment, Text(equals="pay_check"), state="*")
dp.register_callback_query_handler(cart_item_plus, cb_items.filter(action="cart_plus"), state="*")
dp.register_callback_query_handler(cart_item_minus, cb_items.filter(action="cart_minus"), state="*")
async def test_equals_list(self, test_filter_list, test_text, ignore_case):
test_filter = Text(equals=test_filter_list, ignore_case=ignore_case)
async def check(obj):
result = await test_filter.check(obj)
if ignore_case:
_test_filter_list = list(map(str.lower, test_filter_list))
_test_text = test_text.lower()
else:
_test_filter_list = test_filter_list
_test_text = test_text
assert result is (_test_text in _test_filter_list)
await check(Message(text=test_text))
await check(CallbackQuery(data=test_text))
await check(InlineQuery(query=test_text))
await check(Poll(question=test_text))
async def test_equals_string(self, test_filter_text, test_text,
ignore_case):
test_filter = Text(equals=test_filter_text, ignore_case=ignore_case)
async def check(obj):
result = await test_filter.check(obj)
if ignore_case:
_test_filter_text = test_filter_text.lower()
_test_text = test_text.lower()
else:
_test_filter_text = test_filter_text
_test_text = test_text
return result is (_test_text == _test_filter_text)
assert await check(Message(text=test_text))
assert await check(CallbackQuery(data=test_text))
assert await check(InlineQuery(query=test_text))
assert await check(Poll(question=test_text))
async def test_startswith(self, test_prefix, test_text, ignore_case):
test_filter = Text(startswith=test_prefix, ignore_case=ignore_case)
async def check(obj):
result = await test_filter.check(obj)
if ignore_case:
_test_prefix = test_prefix.lower()
_test_text = test_text.lower()
else:
_test_prefix = test_prefix
_test_text = test_text
return result is _test_text.startswith(_test_prefix)
assert await check(Message(text=test_text))
assert await check(CallbackQuery(data=test_text))
assert await check(InlineQuery(query=test_text))
assert await check(Poll(question=test_text))
def register_handlers_settings(dp: Dispatcher):
"""Register routes handlers in Dispatcher."""
log.info('Configuring settings handlers...')
dp.register_message_handler(settings.settings_list, commands='settings')
dp.register_callback_query_handler(
settings.settings_list, cd_settings.filter(action='list')
)
dp.register_callback_query_handler(
settings.settings_tz, cd_settings.filter(action='tz')
)
dp.register_callback_query_handler(
settings.settings_tz_change, cd_settings.filter(action='tz-change')
)
dp.register_message_handler(
settings.settings_tz_set,
Text(startswith='UTC', ignore_case=True),
state=SetTimezone,
)
dp.register_message_handler(settings.settings_tz_error, state=SetTimezone)
async def test_endswith_list(self, test_postfix_list, test_text,
ignore_case):
test_filter = Text(endswith=test_postfix_list, ignore_case=ignore_case)
async def check(obj):
result = await test_filter.check(obj)
if ignore_case:
_test_postfix_list = map(str.lower, test_postfix_list)
_test_text = test_text.lower()
else:
_test_postfix_list = test_postfix_list
_test_text = test_text
return result is any(map(_test_text.endswith, _test_postfix_list))
assert await check(Message(text=test_text))
assert await check(CallbackQuery(data=test_text))
assert await check(InlineQuery(query=test_text))
assert await check(Poll(question=test_text))
async def test_contains_list(self, test_filter_list, test_text,
ignore_case):
test_filter = Text(contains=test_filter_list, ignore_case=ignore_case)
async def check(obj):
result = await test_filter.check(obj)
if ignore_case:
_test_filter_list = list(map(str.lower, test_filter_list))
_test_text = test_text.lower()
else:
_test_filter_list = test_filter_list
_test_text = test_text
return result is all(
map(_test_text.__contains__, _test_filter_list))
assert await check(Message(text=test_text))
assert await check(CallbackQuery(data=test_text))
assert await check(InlineQuery(query=test_text))
assert await check(Poll(question=test_text))
def register_admin(dp: Dispatcher):
dp.register_message_handler(cancel_add_item,
IDFilter(user_id=config.admins),
Text(equals="отбой", ignore_case=True),
state="*")
dp.register_message_handler(add_item,
IDFilter(user_id=config.admins),
commands=["add"],
state="*")
dp.register_message_handler(add_name,
IDFilter(user_id=config.admins),
state=UserStatus.waiting_name)
dp.register_message_handler(add_price,
IDFilter(user_id=config.admins),
state=UserStatus.waiting_price)
dp.register_message_handler(add_description,
IDFilter(user_id=config.admins),
state=UserStatus.waiting_description)
dp.register_message_handler(add_img_link,
IDFilter(user_id=config.admins),
content_types=types.ContentTypes.PHOTO,
state=UserStatus.waiting_img_link)
def register_cart_handlers(dp: dp):
dp.register_callback_query_handler(add_in_cart,
order_cart.filter(action='product_add'))
dp.register_message_handler(cart_view_from_msg, Text(equals='КОРЗИНА'))
dp.register_callback_query_handler(cart_view_from_call,
Text(equals='move_to_cart'))
dp.register_callback_query_handler(get_prev,
Text(equals=['prev']),
state='*')
dp.register_callback_query_handler(get_next,
Text(equals=['next']),
state='*')
dp.register_callback_query_handler(get_up_item,
Text(equals=['up']),
state='*')
dp.register_callback_query_handler(get_down_item,
Text(equals=['down']),
state='*')
dp.register_callback_query_handler(delete_item,
Text(equals=['delete']),
state='*')
