def send_media(message):
if media_type == "image":
send_typing_image(message)
photo = open(media_path, "rb")
bot.send_photo(
message.from_user.id,
photo,
reply_markup=get_keyboard_reply_markup(
get_telegram_config()["built_in"]["main_keyboard"]),
)
if media_type == "audio":
send_typing_audio(message)
audio = open(media_path, "rb")
bot.send_audio(
message.from_user.id,
audio,
reply_markup=get_keyboard_reply_markup(
get_telegram_config()["built_in"]["main_keyboard"]),
)
if media_type == "video":
send_typing_video(message)
video = open(media_path, "rb")
bot.send_video(
message.from_user.id,
video,
reply_markup=get_keyboard_reply_markup(
get_telegram_config()["built_in"]["main_keyboard"]),
)
def send_welcome(message):
send_typing(message)
bot.send_message(
message.from_user.id,
text=get_telegram_config()["built_in"]["main_menu_message"],
reply_markup=get_keyboard_reply_markup(
get_telegram_config()["built_in"]["main_menu"]),
)
def cancel(message):
global answers_question_form
global question_form_count
global question_dict
send_typing(message)
question_form_count = 1
answers_question_form = []
bot.send_message(
message.from_user.id,
text=get_telegram_config()["built_in"]["main_menu_message"],
reply_markup=get_keyboard_reply_markup(
get_telegram_config()["built_in"]["main_menu"]),
)
def question_form(message):
global answers_question_form
global question_form_count
global question_dict
if (message.text == get_telegram_config()["buttons"][
get_telegram_config()["built_in"]["cancel_button"]]):
send_typing(message)
bot.send_message(
message.from_user.id,
text=get_telegram_config()["built_in"]["cancel_message"],
reply_markup=get_keyboard_reply_markup(
get_telegram_config()["built_in"]["main_keyboard"]),
)
bot.register_next_step_handler(message, cancel)
else:
send_typing(message)
if question_form_count < total_steps:
bot.send_message(
message.from_user.id,
text=get_telegram_config()[question_dict]
[f"{[k for k, v in get_telegram_config()[question_dict].items()][question_form_count]}"],
)
answers_question_form.append(message.text)
question_form_count += 1
bot.register_next_step_handler(message, question_form)
else:
answers_question_form.append(message.text)
bot.send_message(
message.from_user.id,
text=get_telegram_config()["built_in"]["thanks_question_form"],
reply_markup=get_keyboard_reply_markup(
get_telegram_config()["built_in"]["main_keyboard"]),
)
send_email(str(answers_question_form))
answers_question_form = []
question_form_count = 1
def get_bot():
global _bot
if _bot is None:
_logger.debug('Creating new Bot')
_bot = telepot.Bot(config.get_telegram_config()['Token'])
return _bot
import socket
import sys
_logger = logging.getLogger(__name__)
_filePath = os.path.dirname(os.path.realpath(__file__))
_logger.debug("Filepath: "+_filePath)
_bot = None
_module_manager = None
_git_branch = None
restart_keyboard = InlineKeyboardMarkup(inline_keyboard=[
[InlineKeyboardButton(text="Restart now", callback_data="restart")]])
try:
_admins = config.get_telegram_config()['Admins']
_admins = _admins.split(',')
except KeyError as e:
err_msg = "Cannot load Admins from Config! Key {} does not exist".format(e)
_logger.error(err_msg)
raise RuntimeError(err_msg)
def config_logger(log_level=logging.INFO):
root = logging.getLogger('')
root.setLevel(log_level)
for h in root.handlers:
root.removeHandler(h)
log_dir = "log"
if not os.path.exists(log_dir):
def get_keyboard_reply_markup(btn):
markup = types.ReplyKeyboardMarkup(resize_keyboard=True)
button_a = ""
button_b = ""
button_c = ""
button_d = ""
button_e = ""
if (len(
get_telegram_config()["keyboards"]
[f"keyboard_{get_telegram_config()['button_on_keyboard'][btn][9:]}_btn_1"]
) > 0):
btns_btn_a = get_telegram_config()["keyboards"][
f"keyboard_{get_telegram_config()['button_on_keyboard'][btn][9:]}_btn_1"]
button_a = types.KeyboardButton(
get_telegram_config()["buttons"][btns_btn_a])
else:
pass
if (len(
get_telegram_config()["keyboards"]
[f"keyboard_{get_telegram_config()['button_on_keyboard'][btn][9:]}_btn_2"]
) > 0):
btns_btn_b = get_telegram_config()["keyboards"][
f"keyboard_{get_telegram_config()['button_on_keyboard'][btn][9:]}_btn_2"]
button_b = types.KeyboardButton(
get_telegram_config()["buttons"][btns_btn_b])
else:
pass
if (len(
get_telegram_config()["keyboards"]
[f"keyboard_{get_telegram_config()['button_on_keyboard'][btn][9:]}_btn_3"]
) > 0):
btns_btn_c = get_telegram_config()["keyboards"][
f"keyboard_{get_telegram_config()['button_on_keyboard'][btn][9:]}_btn_3"]
button_c = types.KeyboardButton(
get_telegram_config()["buttons"][btns_btn_c])
else:
pass
if (len(
get_telegram_config()["keyboards"]
[f"keyboard_{get_telegram_config()['button_on_keyboard'][btn][9:]}_btn_4"]
) > 0):
btns_btn_d = get_telegram_config()["keyboards"][
f"keyboard_{get_telegram_config()['button_on_keyboard'][btn][9:]}_btn_4"]
button_d = types.KeyboardButton(
get_telegram_config()["buttons"][btns_btn_d])
else:
pass
if (len(
get_telegram_config()["keyboards"]
[f"keyboard_{get_telegram_config()['button_on_keyboard'][btn][9:]}_btn_5"]
) > 0):
btns_btn_e = get_telegram_config()["keyboards"][
f"keyboard_{get_telegram_config()['button_on_keyboard'][btn][9:]}_btn_5"]
button_e = types.KeyboardButton(
get_telegram_config()["buttons"][btns_btn_e])
else:
pass
markup.row(button_a, button_b)
markup.row(button_c, button_d, button_e)
# if (
# get_telegram_config()["keyboards"][
# f"keyboard_{keyboard_number}_ReplyKeyboardRemove"
# ]
# == "True"
# ):
# markup = types.ReplyKeyboardRemove(selective=True)
return markup
def handler(message):
global question_form_count
global total_steps
global question_dict
global media_type
global media_path
try:
if message.text in [
v for v in get_telegram_config()["buttons"].values()
]:
btn = [
k for k, v in get_telegram_config()["buttons"].items()
if message.text == v
][0]
msg_content = get_telegram_config()["messages"][f"msg{btn[3:]}"]
if type(msg_content) is str:
msg = msg_content
send_typing(message)
bot.send_message(
message.from_user.id,
text=msg,
reply_markup=get_keyboard_reply_markup(btn),
)
if type(msg_content) is list and msg_content[0] == "question_form":
func = eval(msg_content[0])
question_dict = msg_content[1]
start_msg = get_telegram_config()[question_dict][
msg_content[2]]
total_steps = int([
k[9:]
for k, v in get_telegram_config()[question_dict].items()
][-1])
if question_form_count == 0:
question_form_count += 1
bot.send_message(
message.from_user.id,
text=start_msg,
reply_markup=get_keyboard_reply_markup(btn),
)
bot.register_next_step_handler(message, func)
if type(msg_content) is list and msg_content[0] == "send_media":
func = eval(msg_content[0])
media_type = msg_content[1]
media_path = msg_content[2]
bot.send_message(
message.from_user.id,
text=msg_content[0],
reply_markup=get_keyboard_reply_markup(btn),
)
bot.register_next_step_handler(message, func)
except Exception as e:
bot.send_message(message.from_user.id, text=str(e))
import telebot
from telebot import types
from config import get_telegram_config
from email_send import send_email
answers_question_form = []
question_form_count = 0
question_dict = ""
total_steps = 0
media_path = ""
media_type = ""
# https://github.com/python-telegram-bot/python-telegram-bot
bot = telebot.TeleBot(get_telegram_config()["telegram_bot_token"])
def send_typing(message):
bot.send_chat_action(message.from_user.id, "typing")
def send_typing_image(message):
bot.send_chat_action(message.from_user.id, "upload_photo")
def send_typing_audio(message):
bot.send_chat_action(message.from_user.id, "record_audio")
def send_typing_video(message):
bot.send_chat_action(message.from_user.id, "upload_video")
def __init__(driver, *args, **kwargs):
super().__init__(*args, **kwargs)
driver.messager = BotMessager(*get_telegram_config())
