mirror of
https://github.com/zhayujie/bot-on-anything.git
synced 2026-02-19 00:37:13 +08:00
79 lines
2.7 KiB
Python
79 lines
2.7 KiB
Python
from concurrent.futures import ThreadPoolExecutor
|
||
import io
|
||
import requests
|
||
import telebot
|
||
from common import const
|
||
from common.log import logger
|
||
from channel.channel import Channel
|
||
from config import channel_conf_val, channel_conf
|
||
bot = telebot.TeleBot(token=channel_conf(const.TELEGRAM).get('bot_token'))
|
||
thread_pool = ThreadPoolExecutor(max_workers=8)
|
||
|
||
@bot.message_handler(commands=['help'])
|
||
def send_welcome(message):
|
||
bot.send_message(message.chat.id, "<a>我是chatGPT机器人，开始和我聊天吧!</a>", parse_mode = "HTML")
|
||
|
||
# 处理文本类型消息
|
||
@bot.message_handler(content_types=['text'])
|
||
def send_welcome(msg):
|
||
# telegram消息处理
|
||
TelegramChannel().handle(msg)
|
||
|
||
class TelegramChannel(Channel):
|
||
def __init__(self):
|
||
pass
|
||
|
||
def startup(self):
|
||
logger.info("开始启动[telegram]机器人")
|
||
bot.infinity_polling()
|
||
|
||
def handle(self, msg):
|
||
logger.debug("[Telegram] receive msg: " + msg.text)
|
||
img_match_prefix = self.check_prefix(msg, channel_conf_val(const.TELEGRAM, 'image_create_prefix'))
|
||
# 如果是图片请求
|
||
if img_match_prefix:
|
||
thread_pool.submit(self._do_send_img, msg, str(msg.chat.id))
|
||
else:
|
||
thread_pool.submit(self._dosend,msg.text,msg)
|
||
|
||
def _dosend(self,query,msg):
|
||
context= dict()
|
||
context['from_user_id'] = str(msg.chat.id)
|
||
reply_text = super().build_reply_content(query, context)
|
||
logger.info('[Telegram] reply content: {}'.format(reply_text))
|
||
bot.reply_to(msg,reply_text)
|
||
|
||
def _do_send_img(self, msg, reply_user_id):
|
||
try:
|
||
if not msg:
|
||
return
|
||
context = dict()
|
||
context['type'] = 'IMAGE_CREATE'
|
||
img_urls = super().build_reply_content(msg.text, context)
|
||
if not img_urls:
|
||
return
|
||
if not isinstance(img_urls, list):
|
||
bot.reply_to(msg,img_urls)
|
||
return
|
||
for url in img_urls:
|
||
# 图片下载
|
||
pic_res = requests.get(url, stream=True)
|
||
image_storage = io.BytesIO()
|
||
for block in pic_res.iter_content(1024):
|
||
image_storage.write(block)
|
||
image_storage.seek(0)
|
||
|
||
# 图片发送
|
||
logger.info('[Telegrame] sendImage, receiver={}'.format(reply_user_id))
|
||
bot.send_photo(msg.chat.id,image_storage)
|
||
except Exception as e:
|
||
logger.exception(e)
|
||
|
||
def check_prefix(self, msg, prefix_list):
|
||
if not prefix_list:
|
||
return None
|
||
for prefix in prefix_list:
|
||
if msg.text.startswith(prefix):
|
||
return prefix
|
||
return None
|