def privacy(bot, update): chat_id = get_chat_id(update) bot.send_message(chat_id=chat_id, text=R.get_string_resource("privacy_policy", TBDB.get_chat_lang(chat_id)), parse_mode='html', is_group=chat_id < 0)
def enable_qr(bot, update): chat_id = get_chat_id(update) TBDB.set_chat_qr_enabled(chat_id, 1) bot.send_message(chat_id=chat_id, text=R.get_string_resource("qr_enabled", TBDB.get_chat_lang(chat_id)), is_group=chat_id < 0)
def translate(bot, update): chat_id = get_chat_id(update) message = update.message or update.channel_post if not message: return lang = message.text lang = lang.replace("/translate", "").strip() logger.debug("Language %s", lang) if lang not in config.get_config_prop("app")["languages"]: bot.send_message(chat_id=chat_id, text=R.get_string_resource( "translate_language_not_found", TBDB.get_chat_lang(chat_id)).format(lang), is_group=chat_id < 0) return lang = config.get_config_prop("app")["languages"][lang].split('-')[0] if not message.reply_to_message: bot.send_message(chat_id=chat_id, text=R.get_string_resource( "translate_reply_to_message", TBDB.get_chat_lang(chat_id)), is_group=chat_id < 0) return translation = translator.translate(source=TBDB.get_chat_lang(chat_id), target=lang, text=message.reply_to_message.text) message.reply_text(translation)
def donate(bot, update): chat_id = get_chat_id(update) bot.send_message(chat_id=chat_id, text=R.get_string_resource("message_donate", TBDB.get_chat_lang(chat_id)), parse_mode="html", is_group=chat_id < 0)
def process_media_voice(bot, update, media, name): chat_id = get_chat_id(update) file_size = media.file_size if file_size >= 20 * (1024**2): message_id = get_message_id(update) bot.send_message(chat_id=chat_id, text=R.get_string_resource( "file_too_big", TBDB.get_chat_lang(chat_id)) + "\n", reply_to_message_id=message_id, parse_mode="html", is_group=chat_id < 0) return file_id = media.file_id file_path = os.path.join( config.get_config_prop("app")["media_path"], file_id) file = bot.get_file(file_id) file.download(file_path) try: transcribe_audio_file(bot, update, file_path) except Exception as e: logger.error("Exception handling %s from %d: %s", name, chat_id, traceback.format_exc()) finally: os.remove(file_path)
def disable_photos(bot, update): chat_id = get_chat_id(update) TBDB.set_chat_photos_enabled(chat_id, 0) bot.send_message(chat_id=chat_id, text=R.get_string_resource("photos_disabled", TBDB.get_chat_lang(chat_id)), is_group=chat_id < 0)
def post(bot, update): chat_id = get_chat_id(update) text = update.message.text[6:] def __post(): chats = TBDB.get_chats() sent = 0 for chat in chats: try: bot.send_message(chat_id=chat['chat_id'], text=text, parse_mode='html', is_group=int(chat['chat_id']) < 0) sent += 1 time.sleep(0.1) except Exception as e: logger.error("Exception sending broadcast to %d: (%s) %s", chat['chat_id'], e, traceback.format_exc()) bot.send_message(chat_id=chat_id, text='Broadcast sent to {}/{} chats'.format( sent, len(chats)), is_group=chat_id < 0) TranscriberBot.get().command_handlers['users'](bot, update) TranscriberBot.get().misc_thread_pool.submit(__post)
def left_chat_member(bot, update): chat_id = get_chat_id(update) message = update.message or update.channel_post if message.left_chat_member.id == bot.get_me().id: logger.info('Marking chat {} as inactive'.format(chat_id)) bot.active_chats_cache[chat_id] = 0 TBDB.set_chat_active(chat_id, bot.active_chats_cache[chat_id])
def users(bot, update): chat_id = get_chat_id(update) tot_chats = TBDB.get_chats_num() active_chats = TBDB.get_active_chats_num() bot.send_message(chat_id=chat_id, text='Total users: {}\nActive users: {}'.format( tot_chats, active_chats), parse_mode='html', is_group=chat_id < 0)
def photo(bot, update): chat_id = get_chat_id(update) chat = TBDB.get_chat_entry(chat_id) if chat["qr_enabled"] == 0 and chat["photos_enabled"] == 0: return message = update.message or update.channel_post if message.photo: TranscriberBot.get().photos_thread_pool.submit(process_media_photo, bot, update, message.photo, chat)
def voice(bot, update): chat_id = get_chat_id(update) voice_enabled = TBDB.get_chat_voice_enabled(chat_id) if voice_enabled == 0: return message = update.message or update.channel_post v = message.voice if voice_enabled == 2: pass else: TranscriberBot.get().voice_thread_pool.submit(process_media_voice, bot, update, v, "voice")
def document(bot, update): chat_id = get_chat_id(update) voice_enabled = TBDB.get_chat_voice_enabled(chat_id) m = update.message or update.channel_post file_name = m.document.file_name _, file_ext = os.path.splitext(file_name) if file_ext[1:] not in config.get_config_prop("app")["audio_ext"]: logger.info('extension %s not recognized', file_ext) return if voice_enabled == 0: return if voice_enabled == 2: pass else: TranscriberBot.get().voice_thread_pool.submit(process_media_voice, bot, update, m.document, 'audio_document')
def transcribe_audio_file(bot, update, path): chat_id = get_chat_id(update) lang = TBDB.get_chat_lang(chat_id) message_id = get_message_id(update) is_group = chat_id < 0 api_key = config.get_config_prop("wit").get(lang, None) if api_key is None: logger.error("Language not found in wit.json %s", lang) message = bot.send_message( chat_id=chat_id, text=R.get_string_resource("unknown_api_key", lang).format(language=lang) + "\n", reply_to_message_id=message_id, parse_mode="html", is_group=is_group).result() return logger.debug("Using key %s for lang %s", api_key, lang) message = bot.send_message( chat_id=chat_id, text=R.get_string_resource("transcribing", lang) + "\n", reply_to_message_id=message_id, parse_mode="html", is_group=is_group).result() TranscriberBot.get().start_thread(message_id) logger.debug("Starting thread %d", message_id) keyboard = InlineKeyboardMarkup( [[InlineKeyboardButton("Stop", callback_data=message_id)]]) text = "" if is_group: text = R.get_string_resource("transcription_text", lang) + "\n" success = False for speech in audiotools.transcribe(path, api_key): logger.debug("Thread %d running: %r", message_id, TranscriberBot.get().thread_running(message_id)) if TranscriberBot.get().thread_running(message_id) is False: TranscriberBot.get().del_thread(message_id) return retry = True retry_num = 0 while retry and TranscriberBot.get().thread_running(message_id): try: if len(text + " " + speech) >= 4000: text = R.get_string_resource("transcription_continues", lang) + "\n" message = bot.send_message( chat_id=chat_id, text=text + " " + speech + " <b>[...]</b>", reply_to_message_id=message.message_id, parse_mode="html", is_group=is_group, reply_markup=keyboard).result() else: message = bot.edit_message_text( text=text + " " + speech + " <b>[...]</b>", chat_id=chat_id, message_id=message.message_id, parse_mode="html", is_group=is_group, reply_markup=keyboard).result() text += " " + speech retry = False success = True except telegram.error.TimedOut as t: logger.error("Timeout error %s", traceback.format_exc()) retry_num += 1 if retry_num >= 3: retry = False except telegram.error.RetryAfter as r: logger.warning("Retrying after %d", r.retry_after) time.sleep(r.retry_after) except telegram.error.TelegramError as te: logger.error("Telegram error %s", traceback.format_exc()) retry = False except Exception as e: logger.error("Exception %s", traceback.format_exc()) retry = False retry = True retry_num = 0 while retry and TranscriberBot.get().thread_running(message_id): try: if success: bot.edit_message_text(text=text, chat_id=chat_id, message_id=message.message_id, parse_mode="html", is_group=is_group) else: bot.edit_message_text(R.get_string_resource( "transcription_failed", lang), chat_id=chat_id, message_id=message.message_id, parse_mode="html", is_group=is_group) retry = False except telegram.error.TimedOut as t: logger.error("Timeout error %s", traceback.format_exc()) retry_num += 1 if retry_num >= 3: retry = False except telegram.error.RetryAfter as r: logger.warning("Retrying after %d", r.retry_after) time.sleep(r.retry_after) except telegram.error.TelegramError as te: logger.error("Telegram error %s", traceback.format_exc()) retry = False except Exception as e: logger.error("Exception %s", traceback.format_exc()) retry = False TranscriberBot.get().del_thread(message_id)
def process_media_photo(bot, update, photo, chat): chat_id = get_chat_id(update) message_id = get_message_id(update) is_group = chat_id < 0 message = None if chat["photos_enabled"] == 1: message = bot.send_message(chat_id=chat_id, text=R.get_string_resource( "photo_recognizing", chat["lang"]), reply_to_message_id=message_id, parse_mode="html", is_group=is_group).result() file_id = photo[-1].file_id file_path = os.path.join( config.get_config_prop("app")["media_path"], file_id) bot.get_file(file_id).download(file_path) def process(message): if chat["qr_enabled"] == 1: qr = phototools.read_qr(file_path, chat["lang"]) if qr is not None: if is_group: qr = R.get_string_resource("qr_result", chat["lang"]) + "\n" + qr if message is not None: bot.edit_message_text(text=qr, chat_id=chat_id, message_id=message.message_id, parse_mode="html", is_group=is_group) return else: message = bot.send_message(chat_id=chat_id, text=qr, reply_to_message_id=message_id, parse_mode="html", is_group=is_group).result() if chat["photos_enabled"] == 1: text = phototools.image_ocr(file_path, chat["lang"]) if text is not None: if is_group: text = R.get_string_resource("ocr_result", chat["lang"]) + "\n" + text bot.edit_message_text(text=text, chat_id=chat_id, message_id=message.message_id, parse_mode="html", is_group=is_group) return bot.edit_message_text(text=R.get_string_resource( "photo_no_text", chat["lang"]), chat_id=chat_id, message_id=message.message_id, parse_mode="html", is_group=is_group) retry = True retry_num = 0 try: while retry: process(message) retry = False except telegram.error.TimedOut as t: logger.error("Timeout error %s", traceback.format_exc()) retry_num += 1 if retry_num >= 3: retry = False except telegram.error.RetryAfter as r: logger.warning("Retrying after %d", r.retry_after) time.sleep(r.retry_after) except telegram.error.TelegramError as te: logger.error("Telegram error %s", traceback.format_exc()) retry = False except Exception as e: logger.error("Exception %s", traceback.format_exc()) retry = False finally: os.remove(file_path)
def private_message(bot, update): chat_id = get_chat_id(update) bot.send_message(chat_id=chat_id, text=R.get_string_resource("message_private", TBDB.get_chat_lang(chat_id)))
def rate(bot, update): chat_id = get_chat_id(update) bot.send_message(chat_id=chat_id, text=R.get_string_resource("message_rate", TBDB.get_chat_lang(chat_id)), is_group=chat_id < 0)
def lang(bot, update): chat_id = get_chat_id(update) bot.send_message(chat_id=chat_id, text=TBDB.get_chat_lang(chat_id))