|
|
import logging
|
|
|
import os
|
|
|
import re
|
|
|
|
|
|
import jsonpickle
|
|
|
from django.conf import settings
|
|
|
from werobot import WeRoBot
|
|
|
from werobot.replies import ArticlesReply, Article
|
|
|
from werobot.session.filestorage import FileStorage
|
|
|
|
|
|
from djangoblog.utils import get_sha256
|
|
|
from servermanager.api.blogapi import BlogApi
|
|
|
from servermanager.api.commonapi import ChatGPT, CommandHandler
|
|
|
from .MemcacheStorage import MemcacheStorage
|
|
|
|
|
|
# 初始化微信机器人实例，配置token和启用session功能
|
|
|
robot = WeRoBot(token=os.environ.get('DJANGO_WEROBOT_TOKEN')
|
|
|
or 'lylinux', enable_session=True)
|
|
|
# 创建Memcache存储实例用于session存储
|
|
|
memstorage = MemcacheStorage()
|
|
|
# 根据存储可用性配置机器人的session存储方式
|
|
|
if memstorage.is_available:
|
|
|
robot.config['SESSION_STORAGE'] = memstorage
|
|
|
else:
|
|
|
# 如果文件存储存在则删除旧文件，使用文件存储作为session存储
|
|
|
if os.path.exists(os.path.join(settings.BASE_DIR, 'werobot_session')):
|
|
|
os.remove(os.path.join(settings.BASE_DIR, 'werobot_session'))
|
|
|
robot.config['SESSION_STORAGE'] = FileStorage(filename='werobot_session')
|
|
|
|
|
|
# 初始化博客API和命令处理器实例
|
|
|
blogapi = BlogApi()
|
|
|
cmd_handler = CommandHandler()
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
|
def convert_to_article_reply(articles, message):
|
|
|
"""
|
|
|
将文章列表转换为微信文章回复格式
|
|
|
|
|
|
Args:
|
|
|
articles: 文章对象列表
|
|
|
message: 微信消息对象
|
|
|
|
|
|
Returns:
|
|
|
ArticlesReply: 微信文章回复对象
|
|
|
"""
|
|
|
reply = ArticlesReply(message=message)
|
|
|
from blog.templatetags.blog_tags import truncatechars_content
|
|
|
for post in articles:
|
|
|
# 提取文章中的图片URL
|
|
|
imgs = re.findall(r'(?:http\:|https\:)?\/\/.*\.(?:png|jpg)', post.body)
|
|
|
imgurl = ''
|
|
|
if imgs:
|
|
|
imgurl = imgs[0]
|
|
|
# 创建单篇文章对象
|
|
|
article = Article(
|
|
|
title=post.title,
|
|
|
description=truncatechars_content(post.body),
|
|
|
img=imgurl,
|
|
|
url=post.get_full_url()
|
|
|
)
|
|
|
reply.add_article(article)
|
|
|
return reply
|
|
|
|
|
|
|
|
|
@robot.filter(re.compile(r"^\?.*"))
|
|
|
def search(message, session):
|
|
|
"""
|
|
|
处理文章搜索请求，根据关键词搜索文章并返回结果
|
|
|
|
|
|
Args:
|
|
|
message: 微信消息对象，包含搜索关键词
|
|
|
session: 用户会话对象
|
|
|
|
|
|
Returns:
|
|
|
ArticlesReply或str: 搜索结果或提示信息
|
|
|
"""
|
|
|
search_str = str(message.content).replace('?', '')
|
|
|
result = blogapi.search_articles(search_str)
|
|
|
if result:
|
|
|
# 修复：将map替换为列表推导式
|
|
|
articles = [item.object for item in result]
|
|
|
reply = convert_to_article_reply(articles, message)
|
|
|
return reply
|
|
|
else:
|
|
|
return '没有找到相关文章。'
|
|
|
|
|
|
|
|
|
@robot.filter(re.compile(r'^category\s*$', re.I))
|
|
|
def category(message, session):
|
|
|
"""
|
|
|
获取所有文章分类目录信息
|
|
|
|
|
|
Args:
|
|
|
message: 微信消息对象
|
|
|
session: 用户会话对象
|
|
|
|
|
|
Returns:
|
|
|
str: 包含所有分类名称的字符串
|
|
|
"""
|
|
|
categories = blogapi.get_category_lists()
|
|
|
# 修复：将map替换为列表推导式
|
|
|
content = ','.join([category.name for category in categories])
|
|
|
return '所有文章分类目录：' + content
|
|
|
|
|
|
|
|
|
@robot.filter(re.compile(r'^recent\s*$', re.I))
|
|
|
def recents(message, session):
|
|
|
"""
|
|
|
获取最新发布的文章列表
|
|
|
|
|
|
Args:
|
|
|
message: 微信消息对象
|
|
|
session: 用户会话对象
|
|
|
|
|
|
Returns:
|
|
|
ArticlesReply或str: 最新文章列表或提示信息
|
|
|
"""
|
|
|
articles = blogapi.get_recent_articles()
|
|
|
if articles:
|
|
|
reply = convert_to_article_reply(articles, message)
|
|
|
return reply
|
|
|
else:
|
|
|
return "暂时还没有文章"
|
|
|
|
|
|
|
|
|
@robot.filter(re.compile('^help$', re.I))
|
|
|
def help(message, session):
|
|
|
"""
|
|
|
返回系统帮助信息，包含所有可用命令说明
|
|
|
|
|
|
Args:
|
|
|
message: 微信消息对象
|
|
|
session: 用户会话对象
|
|
|
|
|
|
Returns:
|
|
|
str: 帮助信息文本
|
|
|
"""
|
|
|
return '''欢迎关注!
|
|
|
默认会与图灵机器人聊天~~
|
|
|
你可以通过下面这些命令来获得信息
|
|
|
?关键字搜索文章.
|
|
|
如?python.
|
|
|
category获得文章分类目录及文章数.
|
|
|
category-***获得该分类目录文章
|
|
|
如category-python
|
|
|
recent获得最新文章
|
|
|
help获得帮助.
|
|
|
weather:获得天气
|
|
|
如weather:西安
|
|
|
idcard:获得身份证信息
|
|
|
如idcard:61048119xxxxxxxxxx
|
|
|
music:音乐搜索
|
|
|
如music:阴天快乐
|
|
|
PS:以上标点符号都不支持中文标点~~
|
|
|
'''
|
|
|
|
|
|
|
|
|
@robot.filter(re.compile(r'^weather\:.*$', re.I))
|
|
|
def weather(message, session):
|
|
|
"""
|
|
|
处理天气查询请求（待实现）
|
|
|
|
|
|
Args:
|
|
|
message: 微信消息对象
|
|
|
session: 用户会话对象
|
|
|
|
|
|
Returns:
|
|
|
str: 建设中提示信息
|
|
|
"""
|
|
|
return "建设中..."
|
|
|
|
|
|
|
|
|
@robot.filter(re.compile(r'^idcard\:.*$', re.I))
|
|
|
def idcard(message, session):
|
|
|
"""
|
|
|
处理身份证信息查询请求（待实现）
|
|
|
|
|
|
Args:
|
|
|
message: 微信消息对象
|
|
|
session: 用户会话对象
|
|
|
|
|
|
Returns:
|
|
|
str: 建设中提示信息
|
|
|
"""
|
|
|
return "建设中..."
|
|
|
|
|
|
|
|
|
@robot.handler
|
|
|
def echo(message, session):
|
|
|
"""
|
|
|
主消息处理函数，创建消息处理器并处理用户消息
|
|
|
|
|
|
Args:
|
|
|
message: 微信消息对象
|
|
|
session: 用户会话对象
|
|
|
|
|
|
Returns:
|
|
|
str或其他类型: 处理结果
|
|
|
"""
|
|
|
handler = MessageHandler(message, session)
|
|
|
return handler.handler()
|
|
|
|
|
|
|
|
|
class MessageHandler:
|
|
|
"""微信消息处理器类，负责处理各种用户消息和命令"""
|
|
|
|
|
|
def __init__(self, message, session):
|
|
|
"""
|
|
|
初始化消息处理器
|
|
|
|
|
|
Args:
|
|
|
message: 微信消息对象
|
|
|
session: 用户会话对象
|
|
|
"""
|
|
|
userid = message.source
|
|
|
self.message = message
|
|
|
self.session = session
|
|
|
self.userid = userid
|
|
|
try:
|
|
|
info = session[userid]
|
|
|
self.userinfo = jsonpickle.decode(info)
|
|
|
except Exception:
|
|
|
# 修复：移除了未使用的变量e
|
|
|
userinfo = WxUserInfo()
|
|
|
self.userinfo = userinfo
|
|
|
|
|
|
@property
|
|
|
def is_admin(self):
|
|
|
"""
|
|
|
判断当前用户是否为管理员
|
|
|
|
|
|
Returns:
|
|
|
bool: 是否为管理员
|
|
|
"""
|
|
|
return self.userinfo.is_admin
|
|
|
|
|
|
@property
|
|
|
def is_password_set(self):
|
|
|
"""
|
|
|
判断管理员密码是否已设置
|
|
|
|
|
|
Returns:
|
|
|
bool: 密码是否已设置
|
|
|
"""
|
|
|
return self.userinfo.is_password_set
|
|
|
|
|
|
def save_session(self):
|
|
|
"""
|
|
|
保存用户会话信息到session中
|
|
|
"""
|
|
|
info = jsonpickle.encode(self.userinfo)
|
|
|
self.session[self.userid] = info
|
|
|
|
|
|
def handler(self):
|
|
|
"""
|
|
|
主要的消息处理逻辑，根据用户状态和输入内容进行相应处理
|
|
|
|
|
|
Returns:
|
|
|
str: 处理结果响应文本
|
|
|
"""
|
|
|
info = self.message.content
|
|
|
|
|
|
# 处理管理员退出命令
|
|
|
if self.userinfo.is_admin and info.upper() == 'EXIT':
|
|
|
self.userinfo = WxUserInfo()
|
|
|
self.save_session()
|
|
|
return "退出成功"
|
|
|
|
|
|
# 处理管理员登录命令
|
|
|
if info.upper() == 'ADMIN':
|
|
|
self.userinfo.is_admin = True
|
|
|
self.save_session()
|
|
|
return "输入管理员密码"
|
|
|
|
|
|
# 处理管理员密码验证
|
|
|
if self.userinfo.is_admin and not self.userinfo.is_password_set:
|
|
|
return self._handle_password_verification(info)
|
|
|
|
|
|
# 处理管理员命令执行
|
|
|
if self.userinfo.is_admin and self.userinfo.is_password_set:
|
|
|
return self._handle_admin_commands(info)
|
|
|
|
|
|
# 默认使用ChatGPT处理普通消息
|
|
|
return ChatGPT.chat(info)
|
|
|
|
|
|
def _handle_password_verification(self, info):
|
|
|
"""处理管理员密码验证逻辑"""
|
|
|
passwd = settings.WXADMIN
|
|
|
if settings.TESTING:
|
|
|
passwd = '123'
|
|
|
|
|
|
if passwd.upper() == get_sha256(get_sha256(info)).upper():
|
|
|
self.userinfo.is_password_set = True
|
|
|
self.save_session()
|
|
|
return "验证通过,请输入命令或者要执行的命令代码:输入helpme获得帮助"
|
|
|
else:
|
|
|
# 处理密码错误次数限制
|
|
|
if self.userinfo.count >= 3:
|
|
|
self.userinfo = WxUserInfo()
|
|
|
self.save_session()
|
|
|
return "超过验证次数"
|
|
|
self.userinfo.count += 1
|
|
|
self.save_session()
|
|
|
return "验证失败，请重新输入管理员密码:"
|
|
|
|
|
|
def _handle_admin_commands(self, info):
|
|
|
"""处理管理员命令执行逻辑"""
|
|
|
if self.userinfo.command != '' and info.upper() == 'Y':
|
|
|
return cmd_handler.run(self.userinfo.command)
|
|
|
else:
|
|
|
if info.upper() == 'HELPME':
|
|
|
return cmd_handler.get_help()
|
|
|
self.userinfo.command = info
|
|
|
self.save_session()
|
|
|
return "确认执行: " + info + " 命令?"
|
|
|
|
|
|
|
|
|
class WxUserInfo:
|
|
|
"""微信用户信息类，存储用户的状态信息"""
|
|
|
|
|
|
def __init__(self):
|
|
|
"""
|
|
|
初始化用户信息，默认为非管理员状态
|
|
|
"""
|
|
|
# 修复：字段命名符合Python小写加下划线规范
|
|
|
self.is_admin = False
|
|
|
self.is_password_set = False
|
|
|
self.count = 0
|
|
|
self.command = '' |
