Exemplo n.º 1
0
def make_mocked_message(text, user_id=123, first_name='Юзер', username='******', message_id=None):
    if message_id is None:
        message_id = len(PROCESSED_MESSAGES)
    message = Message(
        message_id=message_id,
        from_user=User(id=user_id, is_bot=False, first_name=first_name, username=username),
        date=None,
        chat=Chat(id=user_id, type='private'),
        content_type=None,
        options={},
        json_string=None
    )
    message.text = text
    return message
Exemplo n.º 2
0
def check_name_surname(message: Message):
    """
    Обрабатывает введенные пользователем имя и фамилию
    при регистрации
    :param message: сообщение пользователя с именем и фамилией
    """
    user_id = message.chat.id

    if message.text in COMMANDS:
        return handle_cancel(message, 'регистрации')

    if not is_text(message) or not is_name_surname_correct(message):
        instruction = bot.send_message(user_id, INCORRECT_DATA_MSG)
        return bot.register_next_step_handler(instruction, check_name_surname)

    message.text = escape(message.text)

    state = get_state(user_id)
    state.registering_user = User(user_id=user_id, name=message.text)
    save_state(user_id, state)

    instruction = bot.send_message(user_id,
                                   REG_MAIL_MSG,
                                   reply_markup=ReplyKeyboardRemove())
    bot.register_next_step_handler(instruction, check_email)
Exemplo n.º 3
0
def check_material(message: Message):
    """
    Обрабатывает введенное пользователем название материала
    для загрузки
    :param message: сообщение пользователя с названием материала
    """
    author_id = message.from_user.id

    if message.text in COMMANDS:
        return handle_cancel(message, 'загрузки')

    if not is_text(message) or not is_title_correct(message):
        instruction = bot.send_message(author_id, INCORRECT_DATA_MSG)
        return bot.register_next_step_handler(instruction, check_material)

    # так как сообщения в поиске выводятся в HTML,
    # для защиты необходимо экранировать пользовательский ввод
    message.text = escape(message.text)

    state = get_state(author_id)
    state.uploading_material = Resource(title=message.text,
                                        author_id=author_id)
    save_state(author_id, state)

    instruction = bot.send_message(author_id, UPLOAD_COURSE_MSG)
    bot.register_next_step_handler(instruction, check_course)
Exemplo n.º 4
0
Arquivo: bot.py Projeto: Alex6430/bot
def upper(message: Message):
    message.text = message.text.lower()
    val = message.text.find("хочу посмотреть")
    if message.text == 'ютуб':
        key = types.InlineKeyboardMarkup()
        button_choose_youtube = types.InlineKeyboardButton(
            text="ютуб", callback_data="youtube")
        key.add(button_choose_youtube)
        bot.send_message(message.chat.id, "нажми кнопку", reply_markup=key)
        # bot.reply_to(message, 'https://www.youtube.com/')
    elif message.text == 'погода':
        key = types.InlineKeyboardMarkup()
        button_choose_weather = types.InlineKeyboardButton(
            text="погода", callback_data="Weather")
        key.add(button_choose_weather)
        bot.send_message(message.chat.id, "нажми кнопку", reply_markup=key)
        # bot.reply_to(message, 'https://yandex.ru/pogoda/moscow')
    elif message.text == 'кино':
        url = 'https://w25.zona.plus/movies?page='
        key = types.InlineKeyboardMarkup()
        button_choose_movi_yes = types.InlineKeyboardButton(
            text="да", callback_data="choose_movi_yes")
        button_choose_movi_not = types.InlineKeyboardButton(
            text="нет", callback_data="choose_movi_not")
        key.add(button_choose_movi_yes, button_choose_movi_not)
        bot.send_message(message.chat.id,
                         "хотите выбрать жанр?",
                         reply_markup=key)
    elif message.text == 'хочу посмотреть':
        bot.reply_to(message, 'https://yandex.ru/pogoda/moscow')
    else:
        bot.reply_to(message, "Я тупой и не понимаю команду")
Exemplo n.º 5
0
def fix_message(bot_instance, message: Message):
    bot_instance.is_from_chat = message.from_user.id != message.chat.id
    if bot_instance.is_from_chat:
        chat_administrators = bot.get_chat_administrators(message.chat.id)
        admin_ids = [admin.user.id for admin in chat_administrators]
        bot_instance.is_from_admin = message.from_user.id in admin_ids
    else:
        bot_instance.is_from_admin = False

    if message.content_type == 'text' and message.text.startswith('/poker'):
        if bot.is_from_chat and not bot.is_from_admin:
            bot.reply_to(message, strings['not_admin'])
            return
        elif not bot.is_from_chat:  # TODO изменить поведение
            bot.reply_to(message, strings['not_in_chat'])
            return

        chat_id = message.chat.id
        result_chat_id = re.search(regex_chat_id, message.text, re.MULTILINE)
        if result_chat_id:
            chat_id = int(regex_chat_id.group(1))

        if settings.DEBUG:
            chat_id = settings.DEBUG_CHAT_ID

        message.chat_id = chat_id
Exemplo n.º 6
0
    def add_course(self, message: Message):
        r1 = compile(
            r'^[ \n]*(?P<name>[\w\d ]+),[ \n]*(?P<duration>[\d]+),[\n ](?P<semester>[\d]+)[ \n]*$'
        )
        r2 = compile(
            r'^[ \n]*(?P<name>[\w\d ]+),[ \n]*(?P<duration>[\d]+),[\n ](?P<semester>\d\d\d\d-\d\d-\d\d)[ \n]*$'
        )

        data = r1.search(message.text)
        new = False
        if not data:
            data = r2.search(message.text)
            new = True

        if not data:
            return self.bot.send_message(message.chat.id,
                                         'Не удалось распознать данные')

        course_info = {
            'name': data.group('name'),
            'duration': data.group('duration'),
            'semester': data.group('semester'),
            'new': new
        }
        try:
            self.obj.create_course(course_info).commit()
        except MyException as error:
            return self.bot.send_message(message.chat.id, error)
        except Exception:
            return self.bot.send_message(message.chat.id,
                                         "Не удалось создать курс.")

        message.text = Bottoms_course.my
        return self.control(message)
Exemplo n.º 7
0
    def download_data(self, message: Message):

        if message.document:
            file = self.bot.get_file(message.document.file_id)
            self.download_state['file'] = self.bot.download_file(
                file.file_path)
        elif message.text:
            r = compile(r'^[ \n]*(?P<id>[\d]+)[ \n]*$')
            data = r.search(message.text)
            if data:
                self.download_state['course'] = data.group('id')
        elif message.text.lower() == 'stop':
            message.text = '/start'
            return self.bot.send_message(message.chat.id,
                                         **registration(message))

        if len(self.download_state) == 2:
            try:
                text = Attachment(message.from_user.id).write_state(
                    self.download_state['file'], self.download_state['course'])
            except Exception as error:
                print(error)
                text = 'При загрузки данных произошла ошибка'

            if not text:
                text = 'Готово'

            return self.bot.send_message(message.chat.id, text)
        else:
            return self.bot.register_next_step_handler(message,
                                                       self.download_data)
    def default_query(inline_query: types.InlineQuery):
        Message = str(inline_query.query)

        def DisplayArgs(args):
            Return = []
            for value, arg in enumerate(args):
                Return.append(
                    InlineQueryResultArticle(
                        id=value,
                        title=arg,
                        input_message_content=InputTextMessageContent(arg)))
            return Return

        #Ничего не написано
        if Message == "":
            Bot.answer_inline_query(
                inline_query.id,
                [command.ResultArticle for command in COMMAND_LIST])

        #Начал писать что то либо
        else:
            Commands = Message.lower().split(" ")
            SeemIt = [
                command for command in COMMAND_LIST
                if command.name.lower().startswith(Commands[0])
            ]

            #Проверяет закончилась ли название команды
            if len(SeemIt) == 1 and SeemIt[0].name == Commands[0]:
                command = SeemIt[0]

                for LEN in range(0, len(Commands) - 1, 1):
                    # Первый параметр
                    try:
                        ArgSeemLike = [
                            arg for arg in command.args[LEN]
                            if arg.lower().startswith(Commands[LEN + 1])
                        ]
                        Bot.answer_inline_query(inline_query.id,
                                                DisplayArgs(ArgSeemLike))
                    except:
                        Bot.answer_inline_query(inline_query.id, [
                            InlineQueryResultArticle(
                                id=0,
                                title="Команда завершена",
                                input_message_content=InputTextMessageContent(
                                    "Отправить команду типа"))
                        ])

            #Нет не закончилось так что продолжает показывать правильное написание команды
            else:
                Bot.answer_inline_query(inline_query.id, [
                    InlineQueryResultArticle(
                        id=0,
                        title=command.name,
                        input_message_content=InputTextMessageContent(
                            command.output),
                        description=command.description) for command in SeemIt
                ])
Exemplo n.º 9
0
def input_command_start(message: Message) -> None:
    text = message.text.split(' ', maxsplit=1)
    if len(text) < 2:
        bot.send_message(chat_id=message.chat.id,
                         text='Need "/search %address%"')
        return
    message.text = text[1]
    controller.get_geo_step(bot=bot, message=message)
Exemplo n.º 10
0
def reply_on_mention(message: Message):
    global Clock
    global Stop
    global Count
    global UwUMode
    global ImagePng
    global Vocal
    if random.randint(1, 100)<15:
         Vocal=True
    Count=50
    Stop=True
    Clock=False
    user_obj: TGUserModel = get_user_from_message(message)
    group: GroupSettings = get_group_from_message(message) if message.chat.type != 'private' else None
    if(group and group.override_settings):  # Use group's settings
        settings: TGUserModel = group.user
    else:
        group = None
        settings = user_obj
    generated_message = generate_markov(fetch_messages(settings, group))
    if (generated_message and not check_duplicated(generated_message, user_obj, group)):
            if UwUMode==True:
               generated_message=generated_message.replace("l", 'w')
               generated_message=generated_message.replace("r", 'w')
            width, height = lcd.dimensions()
            spritemap = Image.open("/boot/TriggerBot-master/Peka.png").convert("PA")
            image = Image.new('1', (width, height),"black")
            image.paste(spritemap,(width-32,33))
            draw = ImageDraw.Draw(image)
            w, h = lcd.dimensions()
            font = ImageFont.truetype("/boot/TriggerBot-master/CCFONT.ttf", 12)
            lines = textwrap.wrap(generated_message, width=16)
            y_text = 0
            for line in lines:
                width, height = font.getsize(line)
                draw.text(((w- width)/4, y_text), line,1, font=font)
                y_text += height
            for x in range(128):
                for y in range(64):
                  pixel = image.getpixel((x, y))
                  lcd.set_pixel(x, y, pixel)
            backlight.set_all(random.randint(0,255),random.randint(0,255),random.randint(0,255))
            backlight.show()
            lcd.show()
            ImagePng=image
            if Vocal==False:
                bot.reply_to(message, generated_message)
            else:
                message.text=generated_message
                audio(message)
            Stop=False
    else:
            Stop=False
    LedOnOff()
Exemplo n.º 11
0
    def del_course(self, message: Message):
        r = compile(r'^[ \n]*(?P<id>[\d]+)[ \n]*$')
        data = r.search(message.text)

        if not data:
            return self.bot.send_message(
                message.chat.id, 'Не удалось распознать идентифиактор курса')

        try:
            self.obj.del_course(data.group('id'))
        except MyException as error:
            return self.bot.send_message(message.chat.id, error)
        except Exception:
            return self.bot.send_message(message.chat.id,
                                         "Не удалось удалить курс")

        message.text = Bottoms_course.my
        return self.control(message)
Exemplo n.º 12
0
def start(msg: types.Message):
    u_lang = msg.from_user.language_code if msg.from_user.language_code in const.languages else const.default_lang
    user, created = User.get_or_create(id=msg.from_user.id,
                                       defaults={
                                           'username': msg.from_user.username,
                                           'first_name':
                                           msg.from_user.first_name,
                                           'last_name':
                                           msg.from_user.last_name,
                                           'language_code': u_lang
                                       })
    params = msg.text.split(' ')
    if len(params) > 1 and params[1] == 'createlist':
        create_list(msg)
    elif len(params) > 1 and params[1].startswith("sub"):
        list_ = List.get_by_id(int(params[1][3:]))
        msg.text = f"📝 {list_.name}"
        new_list, created = List.get_or_create(
            name=list_.name, owner=user.id, defaults={"subscribed_by": list_})
        if created:
            bot.send_message(msg.from_user.id,
                             _("You subscribed to _{}_ list").format(
                                 list_.name),
                             parse_mode='markdown',
                             reply_markup=mps.main_mp(user.id))
            text = f"🔗 📝 *{new_list.name}*"
        else:
            delete_message(msg.chat.id, list_.last_message_id)
            bot.send_message(msg.chat.id,
                             _("You already have list with same name"),
                             reply_markup=mps.main_mp(user.id))
            text = f"📝 *{new_list.name}*"
        sent = bot.send_message(msg.chat.id,
                                text,
                                reply_markup=mps.list_mp(new_list.id),
                                parse_mode='markdown')
        new_list.last_message_id = sent.message_id
        new_list.save()
    else:
        bot.send_message(msg.from_user.id,
                         _("Hello. Create your first list."),
                         reply_markup=mps.main_mp(user.id))
Exemplo n.º 13
0
    def control(self, message: Message):
        text = message.text.lower()
        self.obj = Teacher(message.from_user.id)
        answer = {}

        if Bottoms_course.list.lower() == text:
            answer['text'] = 'Список курсов:\n\n' + format_data(
                '{0}, {1} ({2}).\nНачало {3}\n',
                *self.obj.all_course().all())
        elif Bottoms_course.my.lower() == text:
            answer['text'] = "Список курсов у которых вы являетесь автором:\n"
            answer['text'] += format_data(
                '{0}) {1}, {2} ({3})\n{4}\n',
                *self.obj.my_course().all()) or 'Пока список пуст.'
        elif Bottoms_course.add_course.lower() == text:
            answer['text'] = NEW_COURSE.format(
                semesters=format_data('{0} - {1}',
                                      *self.obj.get_semesters().all()))
            self.bot.register_next_step_handler(message, self.add_course)
        elif Bottoms_course.del_course.lower() == text:
            answer['text'] = "Список ваших курсов:\n"
            my_course = format_data('{0}) {1}, {2} ({3})\n',
                                    *self.obj.my_course().all())
            if not my_course:
                answer['text'] += 'Пока список пуст.'
            else:
                answer[
                    'text'] += my_course + '\nВведите номер курса для удаления.'
            self.bot.register_next_step_handler(message, self.del_course)
        elif Bottoms_course.group_sing_up.lower() == text:
            answer[
                'text'] = "Введите номер курса и название группы. Например:\n1 зивтм-1-19"
            self.bot.register_next_step_handler(message, self.sing_up)
        elif Bottoms_course.back.lower() == text:
            message.text = '/start'
            answer = registration(message)

        if not answer:
            return LessonViewer(self.bot).control(message)

        return self.bot.send_message(message.chat.id, **answer)
Exemplo n.º 14
0
    def control(self, message: Message):
        text = message.text.lower()
        obj = Student(message.from_user.id)
        answer = None

        if Bottoms_stud.list.lower() == text:
            answer = f'Список курсов:\n\n' + format_data(
                '{0}) {1} ({2}).\nНачало {3}\n',
                *obj.all_course().all())
        elif Bottoms_stud.my.lower() == text:
            answer = f"Что бы просмотреть оценки или посещаемость по курсу введите:\nОценки [номер курса]\n"
            answer += f"Посещаемость [номер курса]\n\nСписок курсов на которые подписаннва ваша группа:\n"
            answer += format_data(
                '{0}) {1} ({2})\n',
                *obj.course_student().all()) or "Пока список пуст!"
        elif text == SECRET_COMMAND:
            obj.del_people().commit()
            message.text = '/start'
            self.bot.send_message(message.chat.id, **base_command(message))

        text_command = []
        if ' ' in text:
            text_command = text.split(' ')

        if len(text_command) < 2 or not text_command[1].isdigit():
            answer[
                'text'] = 'Вы ввели некоректные данные после пробела.  Или в вашем сообщении нехватает инфомерации.'

        elif Bottoms_stud.point.lower() == text[0]:
            data = obj.get_point_visible(int(text[1]), True).all()
            answer = f"Ваша успеваемость\n" + format_data('{0} - {1}', *data)
        elif Bottoms_stud.visits.lower() == text[0]:
            data = obj.get_point_visible(int(text[1]), False).all()
            answer = f"Ваша {Bottoms_stud.visits.lower()}" + format_data(
                '{0} - {1}', *data)

        if not answer:
            answer = "Попробуйте снова!!!"

        return self.bot.send_message(message.chat.id, answer)
Exemplo n.º 15
0
    def control(self, message: Message):
        text = message.text.lower()
        obj = Teacher(message.from_user.id)
        answer = {}

        if Bottoms_teacher.course.lower() == text:
            answer['text'] = 'Курсы'
            answer['reply_markup'] = ReplyKeyboardMarkup(row_width=2)
            answer['reply_markup'].add(*map(KeyboardButton, Bottoms_course))
        elif Bottoms_teacher.lessons.lower() == text:
            answer['text'] = 'Занятия'
            answer['reply_markup'] = ReplyKeyboardMarkup(row_width=2)
            answer['reply_markup'].add(*map(KeyboardButton, Bottoms_lesson))
        elif Bottoms_teacher.settings.lower() == text:
            pass
        elif text == SECRET_COMMAND:
            obj.del_people().commit()
            message.text = '/start'
            return base_command(message)

        if not answer:
            return Course(self.bot).control(message)

        return self.bot.send_message(message.chat.id, **answer)
Exemplo n.º 16
0
class TestUserRegistration(unittest.TestCase):
    @parameterized.expand([(Message(from_user=User(first_name="Oleg",
                                                   id=3847590291,
                                                   is_bot=False,
                                                   username="******"),
                                    message_id=2211,
                                    date=datetime.datetime.now(),
                                    chat=Chat(id=381278125782,
                                              type="private",
                                              username="******"),
                                    content_type="text",
                                    options=[],
                                    json_string=""), 12)])
    def test_registration(self, message, value):
        message.text = "/configure_schedule"
        get_user = Mock(return_value=["Hi"])

        with patch('modules.schedule.controller.get_user', new=get_user):
            with patch('modules.schedule.controller.set_user_configured'):
                with patch('modules.core.source.bot.send_message',
                           side_effect=send_message):
                    with patch(
                            'modules.core.source.bot.register_next_step_handler'
                    ):
                        schedule_command_handler(message)
        assert_equal(messages.pop(), permanent.REQUEST_COURSE)
        messages.clear()

    @parameterized.expand([(Message(from_user=User(first_name="Oleg",
                                                   id=3847590291,
                                                   is_bot=False,
                                                   username="******"),
                                    message_id=2211,
                                    date=datetime.datetime.now(),
                                    chat=Chat(id=381278125782,
                                              type="private",
                                              username="******"),
                                    content_type="text",
                                    options=[],
                                    json_string=""), 12)])
    def test_course_set(self, message, value):
        message.text = "B19"
        with patch('modules.schedule.controller.append_user_group'):
            with patch('modules.core.source.bot.send_message',
                       side_effect=send_message):
                with patch(
                        'modules.core.source.bot.register_next_step_handler'):
                    process_course_step(message)
        assert_equal(messages.pop(), permanent.REQUEST_GROUP)

    @parameterized.expand([(Message(from_user=User(first_name="Oleg",
                                                   id=3847590291,
                                                   is_bot=False,
                                                   username="******"),
                                    message_id=2211,
                                    date=datetime.datetime.now(),
                                    chat=Chat(id=381278125782,
                                              type="private",
                                              username="******"),
                                    content_type="text",
                                    options=[],
                                    json_string=""), 12)])
    def test_group_set(self, message, value):
        message.text = "B16-SE-01"
        with patch('modules.schedule.controller.append_user_group'):
            with patch('modules.core.source.bot.send_message',
                       side_effect=send_message):
                with patch(
                        'modules.core.source.bot.register_next_step_handler'):
                    with patch(
                            'modules.schedule.controller.set_user_configured'):
                        process_group_step(message)
        assert_equal(messages.pop(), permanent.MESSAGE_SETTINGS_SAVED)

    @parameterized.expand([(Message(from_user=User(first_name="Oleg",
                                                   id=3847590291,
                                                   is_bot=False,
                                                   username="******"),
                                    message_id=2211,
                                    date=datetime.datetime.now(),
                                    chat=Chat(id=381278125782,
                                              type="private",
                                              username="******"),
                                    content_type="text",
                                    options=[],
                                    json_string=""), 12)])
    def test_wrong_course_set(self, message, value):
        message.text = "Cake is a lie"
        with patch('modules.schedule.controller.append_user_group'):
            with patch('modules.core.source.bot.send_message',
                       side_effect=send_message):
                with patch(
                        'modules.core.source.bot.register_next_step_handler'):
                    process_course_step(message)
        assert_equal(messages.pop(), permanent.MESSAGE_ERROR)
Exemplo n.º 17
0
def fake_message(fake_message_dict):
    """Return a fake, minimalist Telegram message."""
    return Message.de_json(fake_message_dict)
Exemplo n.º 18
0
          'chat': {'id': -383813333,
                   'title': 'Мой техдолг',
                   'type': 'group',
                   'all_members_are_administrators': True},
          'date': 1580657779,
          'text': '/bless_me@every_day_bible_bot',
          'entities': [{'offset': 0,
                        'length': 29,
                        'type': 'bot_command'}]}

test_user = User(id=362857450,
                 is_bot= False,
                 first_name='Nikolas',
                 username='******',
                 last_name='Luchanos',
                 language_code='ru')


test_chat = Chat(type='group',
                 id=-383813333,
                 title='Мой техдолг',
                 all_members_are_administrators=True)

test_msg = Message(message_id=test_message_id,
                   from_user=test_user,
                   date=test_date,
                   chat=test_chat,
                   content_type=test_content_type,
                   options=[],
                   json_string=test_json)
Exemplo n.º 19
0
def echo_digits(message: Message):

	message.text = message.text.lower()

	if 'скинь дз' in message.text:
		if check(message.text.split()[2].strip()):
			if dz[message.text.split()[2].strip()] == 'фото':
				directory = '../'
				all_files_in_directory = os.listdir(directory)
				for name in all_files_in_directory:
					if name == message.text.split()[2].strip()+".jpg":
						print('Скидываю фото',message.text.split()[2].strip()+".jpg")
						img = open(directory+'/'+name,'rb')
						bot.send_chat_action(message.from_user.id,'upload_photo')
						bot.send_photo(message.from_user.id,img,message.text.split()[2].strip())
						img.close()
			else:
				bot.send_message(message.chat.id,dz[message.text.split()[2].strip()])

	elif 'расписание' in message.text:

		if 'на' in message.text:
			if checkR(message.text.split()[2].strip()):
				y = checkF(message.text.split()[2].strip())
				f = date.get(y)
				bot.send_message(message.chat.id,"Расписание на {0}:\n {1}\n {2}\n {3}\n {4}\n {5}\n {6}\n {7}\n {8}\n".format(f[0],f[1],f[2],f[3],f[4],f[5],f[6],f[7],f[8]))

		if 'на завтра' in message.text:
			x = datetime.datetime.now()
			y = x.strftime("%w")
			y = m[int(y)]
			c = date.get(y)
			bot.send_message(message.chat.id,"Расписание на {0}:\n {1}\n {2}\n {3}\n {4}\n {5}\n {6}\n {7}\n {8}\n".format(c[0],c[1],c[2],c[3],c[4],c[5],c[6],c[7],c[8]))

	elif 'скинь все дз' in message.text:
		if check(message.text.split()[2].strip()):
			for k, v in dz.items():
				print(v)
				if v == 'фото':
					directory = '../'
					all_files_in_directory = os.listdir(directory)
					for name in all_files_in_directory:
						if name == k+".jpg":
							print('Скидываю фото',k+".jpg")
							img = open(directory+'/'+name,'rb')
							bot.send_chat_action(message.from_user.id,'upload_photo')
							bot.send_photo(message.from_user.id,img,k)
							img.close()
				else:
					bot.send_message(message.chat.id,k+' '+v)

	elif 'добавь дз' in message.text:
		if check(message.text.split()[2].strip()):
			l = len(message.text.split()[2].strip())
			dz[message.text.split()[2].strip()] = message.text[11+l:] 
			bot.send_message(message.chat.id,'Дз добавлено')

	elif 'удалить дз' in message.text:
		if check(message.text.split()[2].strip()):
			dz[message.text.split()[2].strip()] = 'ничего'
			os.remove(message.text.split()[2].strip()+".jpg")
			bot.send_message(message.chat.id,'Дз удалено')

	else:	
		bot.send_message(message.chat.id,'Я незнаю эту команду')
Exemplo n.º 20
0
class TestCourseAttachment(unittest.TestCase):
    @parameterized.expand([(Message(from_user=User(first_name="Oleg",
                                                   id=3847590291,
                                                   is_bot=False,
                                                   username="******"),
                                    message_id=2211,
                                    date=datetime.datetime.now(),
                                    chat=Chat(id=381278125782,
                                              type="private",
                                              username="******"),
                                    content_type="text",
                                    options=[],
                                    json_string=""), 12)])
    def test_add_course(self, message, value):
        message.text = "/add_course"
        with patch('modules.core.source.bot.register_next_step_handler'):
            with patch('modules.core.source.bot.send_message',
                       side_effect=send_message):
                with patch(
                        'modules.electives_schedule.controller.get_electives_course'
                ):
                    add_course_handler(message)
        assert_equal(messages.pop(), str("What course you want to add?"))

    @parameterized.expand([(Message(from_user=User(first_name="Oleg",
                                                   id=3847590291,
                                                   is_bot=False,
                                                   username="******"),
                                    message_id=2211,
                                    date=datetime.datetime.now(),
                                    chat=Chat(id=381278125782,
                                              type="private",
                                              username="******"),
                                    content_type="text",
                                    options=[],
                                    json_string=""), 12)])
    def test_wrong_add_course(self, message, value):
        message.text = "/some_staff"
        with patch('modules.core.source.bot.register_next_step_handler'):
            with patch('modules.core.source.bot.send_message',
                       side_effect=send_message):
                with patch(
                        'modules.electives_schedule.controller.get_electives_course'
                ):
                    add_course_handler(message)
        assert_equal(len(messages), 0)

    @parameterized.expand([(Message(from_user=User(first_name="Oleg",
                                                   id=3847590291,
                                                   is_bot=False,
                                                   username="******"),
                                    message_id=2211,
                                    date=datetime.datetime.now(),
                                    chat=Chat(id=381278125782,
                                              type="private",
                                              username="******"),
                                    content_type="text",
                                    options=[],
                                    json_string=""), 12)])
    def test_course_set(self, message, value):
        message.text = "B19"
        with patch(
                'modules.electives_schedule.controller.check_electives_course',
                return_value=True):
            with patch('modules.core.source.bot.send_message',
                       side_effect=send_message):
                with patch(
                        'modules.electives_schedule.controller.set_electives_user'
                ):
                    process_electives(message)
        assert_equal(messages.pop(), permanent.MESSAGE_SUCCESS)

    @parameterized.expand([(Message(from_user=User(first_name="Oleg",
                                                   id=3847590291,
                                                   is_bot=False,
                                                   username="******"),
                                    message_id=2211,
                                    date=datetime.datetime.now(),
                                    chat=Chat(id=381278125782,
                                              type="private",
                                              username="******"),
                                    content_type="text",
                                    options=[],
                                    json_string=""), 12)])
    def test_wrong_course_set(self, message, value):
        message.text = "BBBSBS"
        with patch(
                'modules.electives_schedule.controller.check_electives_course',
                return_value=False):
            with patch('modules.core.source.bot.send_message',
                       side_effect=send_message):
                with patch(
                        'modules.electives_schedule.controller.set_electives_user'
                ):
                    process_electives(message)
        assert_equal(messages.pop(), permanent.MESSAGE_UNKNOWN_LESSON)
Exemplo n.º 21
0
 def setUpClass(cls) -> None:
     jsonstring = r'{"message_id":1,"from":{"id":108929734,"first_name":"Frank","last_name":"Wang","username":"******","is_bot":true},"chat":{"id":1734,"first_name":"F","type":"private","last_name":"Wa","username":"******"},"date":1435296025,"text":"/start"}'
     cls.message = Message.de_json(jsonstring)
Exemplo n.º 22
0
def new_user(message: Message):
    try:
        create_user(message)
    except MyException as error:
        message.return_text = error
    return registration_user(message)
Exemplo n.º 23
0
def fake_message(fake_message_dict):
    """Return a fake, minimalist Telegram message."""
    return Message.de_json(fake_message_dict)
Exemplo n.º 24
0
 def get_message(self) -> Message or None:
     if self.chat_type == 'callback_query':
         return None
     return Message.de_json(self.message)
Exemplo n.º 25
0
def _request_lang(language):
    return text_xo.request_admin_support(
        Message(0, User(first_name='', id=0, is_bot=True), 0, 0, 'text',
                dict(text=f'AUTO_REQUEST: {language=}'), ''))