def my_d6(message): """ рандомно выбирает элементы из списка значков TODO: желательно найти способ их увеличить или заменить на ASCII арт :param message: :return: """ d6 = ["⚀", "⚁", "⚂", "⚃", "⚄", "⚅"] dice = 2 roll_sum = 0 symbols = '' for _ in str(message.text).lower().split(): if not len(message.text.split()) == 1: try: dice = int(message.text.split()[1]) except ValueError: my_bot.reply_to(message, "Не понял число костей. " "Пожалуйста, введи команду " "в виде \'/d6 <int>\', " "где <int> — целое от 1 до 10.") return if 0 < dice <= 10: max_result = dice * 6 for count in range(dice): roll_index = random.randint(0, len(d6) - 1) roll_sum += roll_index + 1 if count < dice - 1: symbols += '{0} + '.format(d6[roll_index]) elif count == dice - 1: symbols += '{0} = {1} ({2})'.format(d6[roll_index], roll_sum, max_result) my_bot.reply_to(message, symbols) user_action_log(message, "got that D6 output: {0}".format(symbols))
def check_disa(message): # добавления счетчика в функцию if not hasattr(check_disa, "disa_counter"): check_disa.disa_counter = 0 # проверяем Диса ли это if message.from_user.id != config.disa_id: return # проверяем что идет серия из коротких предложений if len(message.text) > config.length_of_stupid_message: check_disa.disa_counter = 0 return check_disa.disa_counter += 1 # проверяем, будем ли отвечать Дисе disa_trigger = random.randint(1, 6) if check_disa.disa_counter >= config.too_many_messages and disa_trigger == 2: my_bot.reply_to(message, random.choice(config.stop_disa)) check_disa.disa_counter = 0 # записываем в файл увеличенный счетчик хромосом try: with open(config.file_location_disa, 'r+', encoding='utf-8') as file: disa_chromo = str(int(file.read()) + 1) file.seek(0) file.write(disa_chromo) file.truncate() except Exception as ex: logging.error(ex) pass
def vk_post(message): if len(message.text.split()) > 1: post_id = message.text.split('wall')[-1] user_action_log(message, "has requested vk post: {}".format(post_id)) response = requests.get('https://api.vk.com/method/wall.getById', params={ 'access_token': tokens.vk, 'posts': post_id, 'extended': '1', 'v': '5.68' }) response_list = response.json()['response']['items'] if len(response_list) == 0: my_bot.reply_to( message, "Неудача! Использование: `/vk_post vk.com/wall-51776562_939`", parse_mode="Markdown") return post = vk_utils.VkPost(response_list[0]) post.prepare_post() post.send_post(message.chat.id) else: my_bot.reply_to(message, "Использование: `/vk_post vk.com/wall-51776562_939`", parse_mode="Markdown")
def welcoming_task(message): chat_id = message.chat.id new_members_names = [] new_members_info = [] users = [] for member in message.new_chat_members: new_members_names.append(member.first_name) new_members_info.append(user_info(member)) if not member.is_bot: user = { 'user_id': member.id, 'chat_id': chat_id, 'first_name': member.first_name, 'last_name': member.last_name, 'is_member': True } users.append(user) if users: # Bot check with db.atomic(): User.insert_many(users).on_conflict_replace().execute() if str(message.chat.id) == config.mm_chat: welcoming_msg = "{}, {}!\nЕсли здесь впервые, то ознакомься с правилами — /rules, " \ "и представься, если несложно.".format(random.choice(config.welcome_list), ', '.join(new_members_names)) else: welcoming_msg = "{}, {}!\n".format(random.choice(config.welcome_list), ', '.join(new_members_names)) my_bot.reply_to(message, welcoming_msg) action_log("User(s) {} joined the chat.".format( ', '.join(new_members_info)))
def my_dn(message): roll_sum = 0 symbols = '' if len(message.text.split()) == 3: try: dice_max = int(message.text.split()[1]) dice_n = int(message.text.split()[2]) except ValueError: return max_result = dice_n * dice_max for count in range(dice_n): try: roll = random.randint(0, dice_max) roll_sum += roll if count < dice_n - 1: symbols += '{0} + '.format(roll) elif count == dice_n - 1: symbols += '{0} = {1} ({2})'.format(roll, roll_sum, max_result) except ValueError: pass if not len(symbols) > 4096: my_bot.reply_to(message, symbols) user_action_log(message, "knew about /dn and got that output: {0}".format(symbols)) else: my_bot.reply_to(message, "Слишком большие числа. " "Попробуй что-нибудь поменьше") user_action_log(message, "knew about /dn " "and the answer was too long " "to fit one message")
def admin_post(message): if len(message.text.split()) > 1: global_lock.acquire() if message.text.split()[1] == "edit": try: with open(config.file_location_lastbotpost, 'r', encoding='utf-8') as file: last_msg_id = int(file.read()) my_edited_message = ' '.join(message.text.split()[2:]) my_bot.edit_message_text(my_edited_message, config.my_chatID, last_msg_id, parse_mode="Markdown") user_action_log( message, "has edited message {}:\n{}".format( last_msg_id, my_edited_message)) except (IOError, OSError): my_bot.reply_to(message, "Мне нечего редактировать.") else: my_message = ' '.join(message.text.split()[1:]) sent_message = my_bot.send_message(config.my_chatID, my_message, parse_mode="Markdown") with open(config.file_location_lastbotpost, 'w', encoding='utf-8') as file_lastmsgID_write: file_lastmsgID_write.write(str(sent_message.message_id)) user_action_log(message, "has posted this message:\n{}".format(my_message)) global_lock.release() else: my_bot.reply_to(message, "Мне нечего постить.")
def welcoming_task(message): chat_id = message.chat.id new_members_names = [] new_members_info = [] users = [] for member in message.new_chat_members: new_members_names.append(member.first_name) new_members_info.append(user_info(member)) if not member.is_bot: user = { 'user_id' : member.id, 'chat_id' : chat_id, 'first_name': member.first_name, 'last_name' : member.last_name, 'is_member' : True } users.append(user) if users: # Bot check with db.atomic(): User.insert_many(users).on_conflict_replace().execute() if str(message.chat.id) == config.mm_chat: welcoming_msg = "{}, {}!\nЕсли здесь впервые, то ознакомься с правилами — /rules, " \ "и представься, если несложно.".format(random.choice(config.welcome_list), ', '.join(new_members_names)) else: welcoming_msg = "{}, {}!\n".format(random.choice(config.welcome_list), ', '.join(new_members_names)) my_bot.reply_to(message, welcoming_msg) action_log("User(s) {} joined the chat.".format(', '.join(new_members_info)))
def arxiv_search(query, message): try: arxiv_search_res = arxiv.query(search_query=query, max_results=3) query_answer = '' for paper in arxiv_search_res: end = '…' if len(paper['summary']) > 251 else '' a_name = paper['authors'][0] if len(paper['authors']) > 1: a_name += 'et al.' query_answer += \ '• {0}. <a href="{1}">{2}</a>. {3}{4}\n'.format( a_name, paper['arxiv_url'], escape(paper['title'].replace('\n', ' ')), escape(paper['summary'][0:250].replace('\n', ' ')), end) print(query_answer) user_action_log(message, "called arxiv search with query {}".format(query)) my_bot.reply_to(message, query_answer, parse_mode="HTML") except Exception as ex: exc_type, exc_obj, exc_tb = sys.exc_info() fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1] action_log("Unknown Exception:\n{}: {}\nat {} line {}\n" "Creating the alert file.".format(exc_type, ex, fname, exc_tb.tb_lineno))
def your_gender(message): with open(config.file_location['/gender'], 'r', encoding='utf-8') as file_gender: gender = random.choice(file_gender.readlines()) my_bot.reply_to(message, gender) user_action_log( message, "has discovered his gender:\n{0}".format( str(gender).replace("<br>", "\n")))
def ro_roll(text, chat_id=config.mm_chat, max_time=100): time.sleep(0.25) ro_roll_val = random.randint(1, max_time) release_time = time.time() + ro_roll_val * 60 roll_bot_msg = my_bot.send_message(chat_id, "/ROll") my_bot.reply_to(roll_bot_msg, str(ro_roll_val).zfill(2)) my_bot.send_message(chat_id, text.format(str(ro_roll_val).zfill(2))) return release_time
def update_bot(message): if not hasattr(update_bot, "check_sure"): update_bot.check_sure = True return my_bot.reply_to(message, "Ух, ухожу на обновление...") user_action_log(message, "remotely ran update script.") os.execl('/bin/bash', 'bash', 'bot_update.sh')
def my_truth(message): answers = [ "да", "нет", "это не важно", "да, хотя зря", "никогда", "100%", "1 из 100" ] truth = random.choice(answers) my_bot.reply_to(message, truth) user_action_log(message, "has discovered the Truth:\n{0}".format(truth))
def rules_command(message): if str(message.chat.id) == config.mm_chat: with open(config.file_location['/rules'], 'r', encoding='utf-8') as file: my_bot.reply_to(message, file.read(), parse_mode="HTML", disable_web_page_preview=True) user_action_log(message, "called rules")
def my_new_data(message): command = message.text.lower().split()[0] command_raw = re.split("@+", command)[0] with open(config.file_location[command_raw], 'r', encoding='utf-8') as file: my_bot.reply_to(message, file.read(), parse_mode="HTML", disable_web_page_preview=True) user_action_log(message, "called that command: {}".format(command))
def admin_clean(message): if not hasattr(admin_clean, "allow_long"): admin_clean.allow_long = False if not hasattr(admin_clean, "allow_long_id"): admin_clean.allow_long_id = -1 if len(message.text.split()) == 1: if admin_clean.allow_long: user_action_log(message, "cancelled big cleanup") admin_clean.allow_long = False return else: num_str = message.text.split()[1] if not num_str.isdigit(): if admin_clean.allow_long: user_action_log(message, "cancelled big cleanup") admin_clean.allow_long = False return num = int(num_str) allow_long_str = 'Long cleanup is allowed' if admin_clean.allow_long else 'Long cleanup is not allowed' user_action_log( message, "has launched cleanup of {} messages. {}".format(num, allow_long_str)) if num > 500: my_bot.reply_to( message, "Тааак, падажжи, слишком большое число указал, больше 500 не принимаю" ) return if num > 128 and (not admin_clean.allow_long or admin_clean.allow_long_id != message.from_user.id): my_bot.reply_to( message, "Вы запросили очистку более 128 сообщений. Для подтверждения отправьте " "команду еще раз. Для отмены отправльте команду с текстовым параметром. " "С уважением, ваш раб") admin_clean.allow_long = True admin_clean.allow_long_id = message.from_user.id return count = 0 msg_id = message.message_id while count < num: try: my_bot.delete_message(chat_id=message.chat.id, message_id=msg_id) count += 1 except: pass msg_id -= 1 user_action_log(message, "cleaned up {} messages".format(count))
def welcoming_task(message): new_members_names = [] new_members_info = [] for member in message.new_chat_members: new_members_names.append(member.first_name) new_members_info.append(user_info(member)) welcoming_msg = "{}, {}!\nЕсли здесь впервые, то ознакомься с правилами — /rules, " \ "и представься, если несложно.".format(random.choice(config.welcome_list), ', '.join(new_members_names)) my_bot.reply_to(message, welcoming_msg) action_log("User(s) {} joined the chat.".format( ', '.join(new_members_info)))
def get_random_user(message): if message.chat.type != 'private': chat_id = message.chat.id query = User.select().where(User.chat_id == chat_id, User.is_member).order_by(fn.Random()) if query: user = query[0] my_bot.reply_to(message, 'Вам выпал: <a href="tg://user?id={}">{}</a>'. format(user.user_id, strip_tags(user.first_name)), parse_mode='HTML') else: pass # TODO: log that we do not have user base else: my_bot.reply_to(message, "КТО ВЫ ТО? Я ТУТ ОДИН!")
def wolfram_solver(message): """ обрабатывает запрос и посылает пользователю картинку с результатом в случае удачи :param message: :return: """ # сканируем и передаём всё, что ввёл пользователь после '/wolfram ' или '/wf ' if not len(message.text.split()) == 1: my_bot.send_chat_action(message.chat.id, 'upload_photo') your_query = ' '.join(message.text.split()[1:]) user_action_log( message, "entered this query for /wolfram:\n{0}".format(your_query)) response = requests.get( "https://api.wolframalpha.com/v1/simple?appid=" + tokens.wolfram, params={'i': your_query}) # если всё хорошо, и запрос найден if response.status_code == 200: img_original = Image.open(io.BytesIO(response.content)) img_cropped = img_original.crop( (0, 95, 540, img_original.size[1] - 50)) io_img = io.BytesIO() io_img.name = "wolfram {}.png".format(your_query.replace("/", "_")) img_cropped.save(io_img, format="png") io_img.seek(0) wolfram_max_ratio = 2.5 if img_cropped.size[1] / img_cropped.size[0] > wolfram_max_ratio: my_bot.send_document(message.chat.id, io_img, reply_to_message_id=message.message_id) else: my_bot.send_photo(message.chat.id, io_img, reply_to_message_id=message.message_id) user_action_log( message, "has received this Wolfram output:\n{0}".format(response.url)) # если всё плохо else: my_bot.reply_to( message, "Запрос не найдён.\nЕсли ты ввёл его на русском, " "то попробуй ввести его на английском.") user_action_log(message, "didn't received any data") # если пользователь вызвал /wolfram без аргумента else: my_bot.reply_to( message, "Использование: `/wolfram <запрос>` или `/wf <запрос>`", parse_mode="Markdown") user_action_log(message, "called /wolfram without any arguments")
def get_random_user(message): if message.chat.type != 'private': chat_id = message.chat.id query = User.select().where(User.chat_id == chat_id, User.is_member).order_by(fn.Random()) if query: user = query[0] my_bot.reply_to( message, 'Вам выпал: <a href="tg://user?id={}">{}</a>'.format( user.user_id, strip_tags(user.first_name)), parse_mode='HTML') else: pass # TODO: log that we do not have user base else: my_bot.reply_to(message, "КТО ВЫ ТО? Я ТУТ ОДИН!")
def admin_clean(message): if not hasattr(admin_clean, "allow_long"): admin_clean.allow_long = False if not hasattr(admin_clean, "allow_long_id"): admin_clean.allow_long_id = -1 if len(message.text.split()) == 1: if admin_clean.allow_long: user_action_log(message, "cancelled big cleanup") admin_clean.allow_long = False return else: num_str = message.text.split()[1] if not num_str.isdigit(): if admin_clean.allow_long: user_action_log(message, "cancelled big cleanup") admin_clean.allow_long = False return num = int(num_str) allow_long_str = 'Long cleanup is allowed' if admin_clean.allow_long else 'Long cleanup is not allowed' user_action_log(message, "has launched cleanup of {} messages. {}".format(num, allow_long_str)) if num > 500: my_bot.reply_to(message, "Тааак, падажжи, слишком большое число указал, больше 500 не принимаю") return if num > 128 and (not admin_clean.allow_long or admin_clean.allow_long_id != message.from_user.id): my_bot.reply_to(message, "Вы запросили очистку более 128 сообщений. Для подтверждения отправьте " "команду еще раз. Для отмены отправльте команду с текстовым параметром. " "С уважением, ваш раб") admin_clean.allow_long = True admin_clean.allow_long_id = message.from_user.id return count = 0 msg_id = message.message_id while count < num: try: my_bot.delete_message(chat_id=message.chat.id, message_id=msg_id) count += 1 except: pass msg_id -= 1 user_action_log(message, "cleaned up {} messages".format(count))
def vk_post(message): if len(message.text.split()) > 1: post_id = message.text.split('wall')[-1] user_action_log(message, "has requested vk post: {}".format(post_id)) response = requests.get('https://api.vk.com/method/wall.getById', params={'access_token': tokens.vk, 'posts': post_id, 'extended': '1', 'v': config.vk_ver}) response_list = response.json()['response']['items'] if len(response_list) == 0: my_bot.reply_to(message, "Неудача! Использование: `/vk vk.com/wall-51776562_939`", parse_mode="Markdown") return post = vk_utils.VkPost(response_list[0]) post.prepare_post() post.send_post(message.chat.id) else: my_bot.reply_to(message, "Использование: `/vk vk.com/wall-51776562_939`", parse_mode="Markdown")
def rand_image_task(message): path = config.task_dir difficulty = ["1", "2", "3"] user_action_log(message, "asked for a challenge") if not len(message.text.split()) == 1: your_difficulty = message.text.split()[1] if your_difficulty in difficulty: all_imgs = os.listdir(path) rand_img = random.choice(all_imgs) while not rand_img.startswith(your_difficulty): rand_img = random.choice(all_imgs) your_img = open(path + rand_img, "rb") my_bot.send_photo(message.chat.id, your_img, reply_to_message_id=message.message_id) user_action_log( message, "chose a difficulty level '{0}' " "and got that image:\n{1}".format(your_difficulty, your_img.name)) your_img.close() else: my_bot.reply_to( message, "Доступно только три уровня сложности:\n" "{0}" "\nВыбираю рандомную задачу:".format(difficulty)) all_imgs = os.listdir(path) rand_img = random.choice(all_imgs) your_img = open(path + rand_img, "rb") my_bot.send_photo(message.chat.id, your_img, reply_to_message_id=message.message_id) user_action_log( message, "chose a non-existent difficulty level '{0}' " "and got that image:\n{1}".format(your_difficulty, your_img.name)) your_img.close() else: all_imgs = os.listdir(path) rand_img = random.choice(all_imgs) your_img = open(path + rand_img, "rb") my_bot.send_photo(message.chat.id, your_img, reply_to_message_id=message.message_id) user_action_log(message, "got that image:\n{0}".format(your_img.name)) your_img.close()
def rand_image_maths(message): path = config.math_dir subjects = ["algebra", "calculus", "funcan"] user_action_log(message, "asked for maths") if not len(message.text.split()) == 1: your_subject = message.text.split()[1].lower() if your_subject in subjects: all_imgs = os.listdir(path) rand_img = random.choice(all_imgs) while not rand_img.startswith(your_subject): rand_img = random.choice(all_imgs) your_img = open(path + rand_img, "rb") my_bot.send_photo(message.chat.id, your_img, reply_to_message_id=message.message_id) user_action_log( message, "chose subject '{0}' " "and got that image:\n" "{1}".format(your_subject, your_img.name)) your_img.close() else: my_bot.reply_to( message, "На данный момент доступны факты" " только по следующим предметам:\n{0}\n" "Выбираю рандомный факт:".format(subjects)) all_imgs = os.listdir(path) rand_img = random.choice(all_imgs) your_img = open(path + rand_img, "rb") my_bot.send_photo(message.chat.id, your_img, reply_to_message_id=message.message_id) user_action_log( message, "chose a non-existent subject '{0}' " "and got that image:\n" "{1}".format(your_subject, your_img.name)) your_img.close() else: all_imgs = os.listdir(path) rand_img = random.choice(all_imgs) your_img = open(path + rand_img, "rb") my_bot.send_photo(message.chat.id, your_img, reply_to_message_id=message.message_id) user_action_log(message, "got that image:\n{0}".format(your_img.name)) your_img.close()
def rand_image_maths(message): path = config.math_dir subjects = ["algebra", "calculus", "funcan"] user_action_log(message, "asked for maths") if not len(message.text.split()) == 1: your_subject = message.text.split()[1].lower() if your_subject in subjects: all_imgs = os.listdir(path) rand_img = random.choice(all_imgs) while not rand_img.startswith(your_subject): rand_img = random.choice(all_imgs) your_img = open(path + rand_img, "rb") my_bot.send_photo(message.chat.id, your_img, reply_to_message_id=message.message_id) user_action_log(message, "chose subject '{0}' " "and got that image:\n" "{1}".format(your_subject, your_img.name)) your_img.close() else: my_bot.reply_to(message, "На данный момент доступны факты" " только по следующим предметам:\n{0}\n" "Выбираю рандомный факт:".format(subjects) ) all_imgs = os.listdir(path) rand_img = random.choice(all_imgs) your_img = open(path + rand_img, "rb") my_bot.send_photo(message.chat.id, your_img, reply_to_message_id=message.message_id) user_action_log(message, "chose a non-existent subject '{0}' " "and got that image:\n" "{1}".format(your_subject, your_img.name)) your_img.close() else: all_imgs = os.listdir(path) rand_img = random.choice(all_imgs) your_img = open(path + rand_img, "rb") my_bot.send_photo(message.chat.id, your_img, reply_to_message_id=message.message_id) user_action_log(message, "got that image:\n{0}".format(your_img.name)) your_img.close()
def check_disa(message): empty_name = '' # добавления счетчика в функцию if not hasattr(check_disa, "disa_counter"): check_disa.disa_counter = 0 # проверяем что идет серия из коротких предложений try: msg_len = len(message.text) except TypeError: msg_len = 0 if message.from_user.id != config.disa_id or msg_len > config.length_of_stupid_message: check_disa.disa_counter = 0 return check_disa.disa_counter += 1 # проверяем, будем ли отвечать Дисе disa_trigger = random.randint(1, 2) if check_disa.disa_counter >= config.too_many_messages and disa_trigger == 2: # my_bot.reply_to(message, random.choice(config.stop_disa)) ro_roll = random.randint(0, 100) release_time = time.time() + ro_roll * 60 my_bot.restrict_chat_member(chat_id=config.mm_chat, user_id=config.disa_id, until_date=release_time, can_send_messages=False, can_send_media_messages=False, can_send_other_messages=False, can_add_web_page_previews=False) compress_msgs(message, empty_name, message.from_user.first_name, message.from_user.last_name, message.from_user.id, 4) time.sleep(0.25) roll_bot_msg = my_bot.send_message(config.mm_chat, "/ROll") my_bot.reply_to(roll_bot_msg, str(ro_roll).zfill(2)) my_bot.send_message( config.mm_chat, "Эй, {}.\nТвой флуд обеспечил тебе {} мин. РО. Поздравляю!".format( message.from_user.first_name, str(ro_roll).zfill(2))) check_disa.disa_counter = 0 # записываем в файл увеличенный счетчик хромосом disa_chromo = value_from_file(config.file_location['chromo'], 46) value_to_file(config.file_location['chromo'], disa_chromo + 1)
def update_bot(message): if not hasattr(update_bot, "check_sure"): update_bot.check_sure = True return global_lock.acquire() try: file_update_write = open(config.bot_update_filename, 'w', encoding='utf-8') file_update_write.close() except RuntimeError: pass global_lock.release() my_bot.reply_to(message, "Ух, ухожу на обновление...") user_action_log(message, "remotely ran update script.") os.execl('/bin/bash', 'bash', 'bot_update.sh')
def rand_image_task(message): path = config.task_dir difficulty = ["1", "2", "3"] user_action_log(message, "asked for a challenge") if not len(message.text.split()) == 1: your_difficulty = message.text.split()[1] if your_difficulty in difficulty: all_imgs = os.listdir(path) rand_img = random.choice(all_imgs) while not rand_img.startswith(your_difficulty): rand_img = random.choice(all_imgs) your_img = open(path + rand_img, "rb") my_bot.send_photo(message.chat.id, your_img, reply_to_message_id=message.message_id) user_action_log(message, "chose a difficulty level '{0}' " "and got that image:\n{1}".format(your_difficulty, your_img.name)) your_img.close() else: my_bot.reply_to(message, "Доступно только три уровня сложности:\n" "{0}" "\nВыбираю рандомную задачу:".format(difficulty)) all_imgs = os.listdir(path) rand_img = random.choice(all_imgs) your_img = open(path + rand_img, "rb") my_bot.send_photo(message.chat.id, your_img, reply_to_message_id=message.message_id) user_action_log(message, "chose a non-existent difficulty level '{0}' " "and got that image:\n{1}".format(your_difficulty, your_img.name)) your_img.close() else: all_imgs = os.listdir(path) rand_img = random.choice(all_imgs) your_img = open(path + rand_img, "rb") my_bot.send_photo(message.chat.id, your_img, reply_to_message_id=message.message_id) user_action_log(message, "got that image:\n{0}".format(your_img.name)) your_img.close()
def command_or(message): user_action_log(message, "called: " + message.text) # Shitcode alert! or_lang = "ru" if len(message.text.split()) < 4: return or_message = message.text.split(' ', 1)[1] if "or" in message.text.split(): make_choice = re.split(r'[ ](?:or)[, ]', or_message) or_lang = "en" else: make_choice = re.split(r'[ ](?:или)[, ]', or_message) if len(make_choice) > 1 and not ((message.text.split()[1] == "или") or (message.text.split()[1] == "or")): choosen_answer = random.choice(make_choice) if or_lang == "ru": choosen_answer = re.sub(r'(?i)\bя\b', 'ты', choosen_answer) else: choosen_answer = re.sub(r'(?i)\bi\b', 'you', choosen_answer) # more subs to come my_bot.reply_to(message, choosen_answer)
def disa_vk_report(disa_chromo, message): login, password = tokens.vk_disa_login, tokens.vk_disa_password vk_session = vk_api.VkApi(login, password) vk_session.auth() vk = vk_session.get_api() wall = vk.wall.get(owner_id=config.vk_disa_groupID, count=1) if time.localtime(wall['items'][0]['date'])[2] == time.localtime()[2]: disa_chromo_post = disa_chromo - 46 try: old_chromo = int(wall['items'][0]['text']) disa_chromo_post += old_chromo except Exception as ex: logging.error(ex) disa_chromo_post = disa_chromo vk.wall.edit(owner_id=config.vk_disa_groupID, post_id=wall['items'][0]['id'], message=str(disa_chromo_post)) else: disa_chromo_post = 46 + disa_chromo vk.wall.post(owner_id=config.vk_disa_groupID, message=str(disa_chromo_post)) if 1 < disa_chromo - 46 % 10 < 5: chromo_end = "ы" elif disa_chromo - 46 % 10 == 1: chromo_end = "а" else: chromo_end = "" my_bot.reply_to( message, "С последнего репорта набежало {0} хромосом{1}.\n" "Мы успешно зарегистрировали этот факт: " "https://vk.com/disa_count".format((disa_chromo - 46), chromo_end)) action_log("Disa summary printed") disa_chromo = 46 global_lock.acquire() with open(config.file_location_disa, 'w', encoding='utf-8') as file_disa_write: file_disa_write.write(str(disa_chromo)) global_lock.release() disa.disa_first = True
def command_or(message): user_action_log(message, "called: " + message.text) # Shitcode alert! or_lang = "ru" if len(message.text.split()) < 4: return or_message = message.text.split(' ', 1)[1] if "or" in message.text.split(): make_choice = re.split(r'[ ](?:or)[, ]', or_message) or_lang = "en" else: make_choice = re.split(r'[ ](?:или)[, ]', or_message) if len(make_choice) > 1 and not ((message.text.split()[1] == "или") or (message.text.split()[1] == "or")): choosen_answer = random.choice(make_choice) if or_lang == "ru": choosen_answer = re.sub(r'(?i)\bя\b', 'ты', choosen_answer) else: choosen_answer = re.sub(r'(?i)\bi\b', 'you', choosen_answer) ## more subs to come my_bot.reply_to(message, choosen_answer)
def my_weather(message): """ Получает погоду в Москве на сегодня и на три ближайших дня, пересылает пользователю :param message: :return: """ try: my_owm = pyowm.OWM(tokens.owm) # где мы хотим узнать погоду my_obs = my_owm.weather_at_place('Moscow') except pyowm.exceptions.unauthorized_error.UnauthorizedError: print("Your API subscription level does not allow to check weather") return w = my_obs.get_weather() # статус погоды сейчас status = w.get_detailed_status() # температура сейчас temp_now = w.get_temperature('celsius') # limit=4, т.к. первый результат — текущая погода my_forecast = my_owm.daily_forecast('Moscow,RU', limit=4) my_fc = my_forecast.get_forecast() # температуры на следующие три дня my_fc_temps = [] # статусы на следующие три дня my_fc_statuses = [] for wth in my_fc: my_fc_temps.append(str(wth.get_temperature('celsius')['day'])) my_fc_statuses.append(str(wth.get_status())) # если всё нормально, то выводим результаты else: forecast = "The current temperature in Moscow is {} C, " \ "and it is {}.\n\n" \ "Tomorrow it will be {} C, {}.\n" \ "In 2 days it will be {}, {}.\n" \ "In 3 days it will be {} C, {}.".format(temp_now['temp'], status, my_fc_temps[1], my_fc_statuses[1], my_fc_temps[2], my_fc_statuses[2], my_fc_temps[3], my_fc_statuses[3]) my_bot.reply_to(message, forecast) user_action_log(message, "got that weather forecast")
def vk_post_last(message): if len(message.text.split()) > 1: count = message.text.split()[1] if not count.isdigit(): my_bot.reply_to(message, "Неудача! Использование: `/vk_post_last [N]`", parse_mode="Markdown") return response = requests.get('https://api.vk.com/method/wall.get', params={'access_token': tokens.vk, 'owner_id': config.mm_vk_group, 'count': count, 'offset': 0, 'v': config.vk_ver}) posts = response.json()['response']['items'] if len(posts) == 0: my_bot.reply_to(message, "Неудача! Использование: `/vk_post_last [N]`", parse_mode="Markdown") return for post in reversed(posts): post_vk = vk_utils.VkPost(post) post_vk.prepare_post() try: if config.mm_chat != '': post_vk.send_post(config.mm_chat) if config.mm_channel != '': post_vk.send_post(config.mm_channel) if tokens.fb != '' and config.mm_fb_album != '': post_vk.send_post_fb(tokens.fb, config.mm_fb_album) except Exception as ex: logging.exception(ex) else: my_bot.reply_to(message, "Использование: `/vk_post_last [N]`", parse_mode="Markdown")
def wolfram_solver(message): """ обрабатывает запрос и посылает пользователю картинку с результатом в случае удачи :param message: :return: """ # сканируем и передаём всё, что ввёл пользователь после '/wolfram ' или '/wf ' if not len(message.text.split()) == 1: my_bot.send_chat_action(message.chat.id, 'upload_photo') your_query = ' '.join(message.text.split()[1:]) user_action_log(message, "entered this query for /wolfram:\n{0}".format(your_query)) response = requests.get("https://api.wolframalpha.com/v1/simple?appid=" + tokens.wolfram, params={'i': your_query}) # если всё хорошо, и запрос найден if response.status_code == 200: img_original = Image.open(io.BytesIO(response.content)) img_cropped = img_original.crop((0, 95, 540, img_original.size[1] - 50)) io_img = io.BytesIO() io_img.name = "wolfram {}.png".format(your_query.replace("/", "_")) img_cropped.save(io_img, format="png") io_img.seek(0) wolfram_max_ratio = 2.5 if img_cropped.size[1] / img_cropped.size[0] > wolfram_max_ratio: my_bot.send_document(message.chat.id, io_img, reply_to_message_id=message.message_id) else: my_bot.send_photo(message.chat.id, io_img, reply_to_message_id=message.message_id) user_action_log(message, "has received this Wolfram output:\n{0}".format(response.url)) # если всё плохо else: my_bot.reply_to(message, "Запрос не найдён.\nЕсли ты ввёл его на русском, " "то попробуй ввести его на английском.") user_action_log(message, "didn't received any data") # если пользователь вызвал /wolfram без аргумента else: my_bot.reply_to(message, "Использование: `/wolfram <запрос>` или `/wf <запрос>`", parse_mode="Markdown") user_action_log(message, "called /wolfram without any arguments")
def admin_post(message): if len(message.text.split()) > 1: global_lock.acquire() if message.text.split()[1] == "edit": try: with open(config.file_location['last_post'], 'r', encoding='utf-8') as file: last_msg_id = int(file.read()) my_edited_message = ' '.join(message.text.split()[2:]) my_bot.edit_message_text(my_edited_message, config.mm_chat, last_msg_id, parse_mode="Markdown") user_action_log(message, "has edited message {}:\n{}".format(last_msg_id, my_edited_message)) except (IOError, OSError): my_bot.reply_to(message, "Мне нечего редактировать.") else: my_message = ' '.join(message.text.split()[1:]) sent_message = my_bot.send_message(config.mm_chat, my_message, parse_mode="Markdown") with open(config.file_location['last_post'], 'w', encoding='utf-8') as file_lastmsgID_write: file_lastmsgID_write.write(str(sent_message.message_id)) user_action_log(message, "has posted this message:\n{}".format(my_message)) global_lock.release() else: my_bot.reply_to(message, "Мне нечего постить.")
def disa_vk_report(disa_chromo, message): login, password = tokens.vk_disa_login, tokens.vk_disa_password vk_session = vk_api.VkApi( login, password, config_filename=config.file_location['vk_config']) vk_session.auth() vk = vk_session.get_api() wall = vk.wall.get(owner_id=config.disa_vk_group, count=1) if time.localtime(wall['items'][0]['date'])[2] == time.localtime()[2]: disa_chromo_post = disa_chromo - 46 try: old_chromo = int(wall['items'][0]['text']) disa_chromo_post += old_chromo except Exception as ex: logging.error(ex) disa_chromo_post = disa_chromo vk.wall.edit(owner_id=config.disa_vk_group, post_id=wall['items'][0]['id'], message=str(disa_chromo_post)) else: disa_chromo_post = 46 + disa_chromo vk.wall.post(owner_id=config.disa_vk_group, message=str(disa_chromo_post)) if 1 < disa_chromo - 46 % 10 < 5: chromo_end = "ы" elif disa_chromo - 46 % 10 == 1: chromo_end = "а" else: chromo_end = "" my_bot.reply_to( message, "С последнего репорта набежало {0} хромосом{1}.\n" "Мы успешно зарегистрировали этот факт: " "https://vk.com/disa_count".format((disa_chromo - 46), chromo_end)) action_log("Disa summary printed") value_to_file(config.file_location['chromo'], 46) disa.disa_first = True
def disa_vk_report(disa_chromo, message): login, password = tokens.vk_disa_login, tokens.vk_disa_password vk_session = vk_api.VkApi(login, password, config_filename=config.file_location['vk_config']) vk_session.auth() vk = vk_session.get_api() wall = vk.wall.get(owner_id=config.disa_vk_group, count=1) if time.localtime(wall['items'][0]['date'])[2] == time.localtime()[2]: disa_chromo_post = disa_chromo - 46 try: old_chromo = int(wall['items'][0]['text']) disa_chromo_post += old_chromo except Exception as ex: logging.error(ex) disa_chromo_post = disa_chromo vk.wall.edit(owner_id=config.disa_vk_group, post_id=wall['items'][0]['id'], message=str(disa_chromo_post)) else: disa_chromo_post = 46 + disa_chromo vk.wall.post(owner_id=config.disa_vk_group, message=str(disa_chromo_post)) if 1 < disa_chromo - 46 % 10 < 5: chromo_end = "ы" elif disa_chromo - 46 % 10 == 1: chromo_end = "а" else: chromo_end = "" my_bot.reply_to(message, "С последнего репорта набежало {0} хромосом{1}.\n" "Мы успешно зарегистрировали этот факт: " "https://vk.com/disa_count".format((disa_chromo - 46), chromo_end)) action_log("Disa summary printed") value_to_file(config.file_location['chromo'], 46) disa.disa_first = True
def arxiv_search(query, message): try: arxiv_search_res = arxiv.query(search_query=query, max_results=3) query_answer = '' for paper in arxiv_search_res: end = '…' if len(paper['summary']) > 251 else '' a_name = paper['authors'][0] if len(paper['authors'])>1: a_name += 'et al.' query_answer += \ '• {0}. <a href="{1}">{2}</a>. {3}{4}\n'.format( a_name, paper['arxiv_url'], escape(paper['title'].replace('\n', ' ')), escape(paper['summary'][0:250].replace('\n', ' ')), end) print(query_answer) user_action_log(message, "called arxiv search with query {}".format(query)) my_bot.reply_to(message, query_answer, parse_mode="HTML") except Exception as ex: exc_type, exc_obj, exc_tb = sys.exc_info() fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1] action_log("Unknown Exception:\n{}: {}\nat {} line {}\n" "Creating the alert file.".format(exc_type, ex, fname, exc_tb.tb_lineno))
def vk_post_last(message): if len(message.text.split()) > 1: count = message.text.split()[1] if not count.isdigit(): my_bot.reply_to(message, "Неудача! Использование: `/vk_post_last [N]`", parse_mode="Markdown") return response = requests.get('https://api.vk.com/method/wall.get', params={ 'access_token': tokens.vk, 'owner_id': config.mm_vk_group, 'count': count, 'offset': 0, 'v': config.vk_ver }) posts = response.json()['response']['items'] if len(posts) == 0: my_bot.reply_to(message, "Неудача! Использование: `/vk_post_last [N]`", parse_mode="Markdown") return for post in reversed(posts): vk_post = vk_utils.VkPost(post) vk_post.prepare_post() try: if config.mm_chat != '': vk_post.send_post(config.mm_chat) if config.mm_channel != '': vk_post.send_post(config.mm_channel) if tokens.fb != '' and config.mm_fb_album != '': vk_post.send_post_fb(tokens.fb, config.mm_fb_album) except Exception as ex: logging.exception(ex) else: my_bot.reply_to(message, "Использование: `/vk_post_last [N]`", parse_mode="Markdown")
def get_log(message): user_action_log(message, "requested bot logs") with open(config.file_location['bot_logs'], 'r', encoding='utf-8') as file: lines = file.readlines()[-100:] for text in cut_long_text(''.join(lines), max_len=3500): my_bot.reply_to(message, "{}".format(text))
def get_uid(message): if getattr(message, 'reply_to_message') is None: return my_bot.reply_to(message, '`{}`'.format(str(message.reply_to_message.from_user.id)), parse_mode="Markdown")
def id_reply(message): my_bot.reply_to(message, message.text) user_action_log(message, "called echo:\n{}".format(message.text))
def underscope_reply(message): my_bot.reply_to(message, "_\\") user_action_log(message, "called the _\\")
def my_wiki(message): """ Обрабатывает запрос и пересылает результат. Если запроса нет, выдаёт рандомный факт. :param message: :return: """ # обрабатываем всё, что пользователь ввёл после '/wiki ' if not len(message.text.split()) == 1: your_query = ' '.join(message.text.split()[1:]) user_action_log(message, "entered this query for /wiki:\n{0}".format(your_query)) try: # определяем язык запроса. Эвристика для английского и русского if all(ord(x) < 127 or not x.isalpha() for x in your_query): wikipedia.set_lang('en') # TODO: a bit dirty condition elif all(ord(x) < 127 or (ord('Ё') <= ord(x) <= ord('ё')) or not x.isalpha() for x in your_query): wikipedia.set_lang('ru') else: wikipedia.set_lang(detect(your_query)) wiki_response = wikipedia.summary(your_query, sentences=7) if '\n \n' in str(wiki_response): wiki_response = "{}...\n\n" \ "<i>В данной статье " \ "имеется математическая вёрстка. " \ "Ссылка на статью:</i>".format(str(wiki_response).split('\n \n', 1)[0]) # print(wiki_response) # извлекаем ссылку на саму статью wiki_url = parse.unquote(wikipedia.page(your_query).url) # извлекаем название статьи wiki_title = wikipedia.page(your_query).title my_bot.reply_to(message, "<b>{0}.</b>\n{1}\n\n{2}".format( wiki_title, wiki_response, wiki_url), parse_mode="HTML") user_action_log(message, "got Wikipedia article\n{0}".format(str(wiki_title))) # всё плохо, ничего не нашли except wikipedia.exceptions.PageError: my_bot.reply_to(message, "Запрос не найден.") user_action_log(message, "didn't received any data.") # нашли несколько статей, предлагаем пользователю список except wikipedia.exceptions.DisambiguationError as ex: wiki_options = ex.options my_bot.reply_to(message, "Пожалуйста, уточни запрос. " "Выбери, что из перечисленного имелось в виду, " "и вызови /wiki ещё раз.\n" + "\n".join(map(str, wiki_options))) print("There are multiple possible pages for that article.\n") # берём рандомную статью на рандомном языке (языки в config.py) else: wikipedia.set_lang(random.choice(['en', 'ru'])) wikp = wikipedia.random(pages=1) try: print("Trying to get Wikipedia article\n{0}".format(str(wikp))) wikpd = wikipedia.page(wikp) wiki_fact = wikipedia.summary(wikp, sentences=3) my_bot.reply_to(message, "<b>{0}.</b>\n{1}".format(wikpd.title, wiki_fact), parse_mode="HTML") user_action_log(message, "got Wikipedia article\n{0}".format(str(wikp))) except wikipedia.exceptions.DisambiguationError: wikp = wikipedia.random(pages=1) try: print("Trying to get Wikipedia article\n{0}".format(str(wikp))) wiki_var = wikipedia.search(wikp, results=1) print("There are multiple possible pages for that article.\n") # wikpd = wikipedia.page(str(wiki_var[0])) wiki_fact = wikipedia.summary(wiki_var, sentences=4) my_bot.reply_to(message, "<b>{0}.</b>\n{1}".format(wikp, wiki_fact), parse_mode="HTML") except wikipedia.exceptions.PageError: print("Page error for Wikipedia article\n{0}".format(str(wikp))) my_bot.reply_to(message, "<b>{0}.</b>".format("You're even unluckier"), parse_mode="HTML") except Exception as ex: exc_type, exc_obj, exc_tb = sys.exc_info() fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1] action_log("Unknown Exception in Wikipedia: {}: {}\nat {} line {}".format(exc_type, ex, fname, exc_tb.tb_lineno)) except wikipedia.exceptions.PageError: print("Page error for Wikipedia article\n{0}".format(str(wikp))) my_bot.reply_to(message, "<b>{0}.</b>".format("You're very unlucky bastard"), parse_mode="HTML") except Exception as ex: exc_type, exc_obj, exc_tb = sys.exc_info() fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1] action_log("Unknown Exception in Wikipedia: {}: {}\nat {} line {}".format(exc_type, ex, fname, exc_tb.tb_lineno))
def your_gender(message): with open(config.file_location['/gender'], 'r', encoding='utf-8') as file_gender: gender = random.choice(file_gender.readlines()) my_bot.reply_to(message, gender) user_action_log(message, "has discovered his gender:\n{0}".format(str(gender).replace("<br>", "\n")))
def my_roll(message): rolled_number = random.randint(0, 100) my_bot.reply_to(message, str(rolled_number).zfill(2)) user_action_log(message, "recieved {0}".format(rolled_number))
def my_truth(message): answers = ["да", "нет", "это не важно", "да, хотя зря", "никогда", "100%", "1 из 100"] truth = random.choice(answers) my_bot.reply_to(message, truth) user_action_log(message, "has discovered the Truth:\n{0}".format(truth))
def id_reply(message): my_bot.reply_to(message, "/id") user_action_log(message, "called the id")
def arxiv_random(message): user_action_log(message, "made arxiv random query") try: eastern = pytz.timezone('US/Eastern') eastern_time = datetime.datetime.now(eastern) # publications on 20:00 if eastern_time.hour < 20: eastern_time -= datetime.timedelta(days=1) # no publications on friday and saturday if eastern_time.weekday() == 5: eastern_time -= datetime.timedelta(days=2) elif eastern_time.weekday() == 4: eastern_time -= datetime.timedelta(days=1) last_published_date = eastern_time.strftime("%Y-%m-%d") response = requests.get('http://export.arxiv.org/oai2', params={'verb' : 'ListIdentifiers', 'set' : 'math', 'metadataPrefix': 'oai_dc', 'from' : last_published_date}) action_log("Random arxiv paper since {}".format(last_published_date)) # если всё хорошо if response.status_code == 200: response_tree = ElementTree.fromstring(response.content) num_of_papers = len(response_tree[2]) paper_index = random.randint(0, num_of_papers) paper_arxiv_id = response_tree[2][paper_index][0].text.split(':')[-1] # hardcoded papep_obj = arxiv.query(id_list=[paper_arxiv_id])[0] paper_link = papep_obj['pdf_url'].replace('http://', 'https://') + '.pdf' paper_link_name = paper_link.split("/pdf/")[1] print(paper_link) print(paper_link_name) req_pdf_size = requests.head(paper_link) pdf_size = round(int(req_pdf_size.headers["Content-Length"]) / 1024 / 1024, 2) a_name = papep_obj['authors'][0] if len(papep_obj['authors'])>1: a_name += 'et al.' query_answer = '{}. <a href="{}">{}</a>. {}\n\n— <a href="{}">{}</a>, {} Мб\n'.format( a_name, papep_obj['arxiv_url'], escape(papep_obj['title'].replace('\n', ' ')), escape(papep_obj['summary'].replace('\n', ' ')), paper_link, paper_link_name, pdf_size ) my_bot.reply_to(message, query_answer, parse_mode="HTML", disable_web_page_preview=False) user_action_log(message, "arxiv random query was successful: " "got paper {}".format(papep_obj['arxiv_url'])) # TODO(randl): doesn't send. Download and delete? # my_bot.send_document(message.chat.id, data=paper_link) elif response.status_code == 503: # слишком часто запрашиваем action_log("Too much queries. 10 minutes break should be enough") arxiv_checker.last_call = datetime.datetime.utcnow() - datetime.timedelta(seconds=610) else: # если всё плохо user_action_log(message, "arxiv random query failed: response {}".format(response.status_code)) except Exception as ex: exc_type, exc_obj, exc_tb = sys.exc_info() fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1] action_log("Unknown Exception: {}: {}\nat {} line {}".format(exc_type, ex, fname, exc_tb.tb_lineno))