You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

83 lines
3.4 KiB

3 months ago
import requests
from bs4 import BeautifulSoup
from config.logger import setup_logging
from plugins_func.register import register_function, ToolType, ActionResponse, Action
import json
import base64
import urllib.parse
TAG = __name__
logger = setup_logging()
SEND_NOTICE_FUNCTION_DESC = {
"type": "function",
"function": {
"name": "send_notice",
"description": (
"用户发送通知,用户需要提供通知标题和通知内容"
"如果用户没有提供具体的通知标题,需要进行分析得出通知标题"
"注意时间的格式例如16:00决不能是1600"
),
"parameters": {
"type": "object",
"properties": {
"title": {
"type": "string",
"description": "通知标题"
},
"content": {
"type": "string",
"description": "通知内容"
},
"lang": {
"type": "string",
"description": "返回用户使用的语言code例如zh_CN/zh_HK/en_US/ja_JP等默认zh_CN"
}
},
"required": ["lang"]
}
}
}
HEADERS = {
'User-Agent': (
'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 '
'(KHTML, like Gecko) Chrome/92.0.4515.107 Safari/537.36'
)
}
@register_function('send_notice', SEND_NOTICE_FUNCTION_DESC, ToolType.SYSTEM_CTL)
def send_notice(conn, title: str = None, content: str = None, lang: str = "zh_CN"):
print("通知标题:"+title)
print("通知内容:"+content)
# 对中文内容进行BASE64编码
title_base64 = base64.b64encode(title.encode('utf-8')).decode('utf-8')
content_base64 = base64.b64encode(content.encode('utf-8')).decode('utf-8')
# 对BASE64结果进行URL编码
title_urlencoded = urllib.parse.quote(title_base64)
content_urlencoded = urllib.parse.quote(content_base64)
url = "https://www.edusoa.com/dsideal_yy/new_notice/kjAddNotice"
payload = f'notice_content=通知&notice_title=通知&noticeJson=%7B%22notice_title%22%3A%22{title_urlencoded}%22%2C%22notice_content%22%3A%22{content_urlencoded}%22%2C%22notice_level%22%3A%221%22%2C%22thumb_id%22%3A%22%22%2C%22notice_receipt%22%3A0%2C%22notice_public%22%3A0%2C%22sender_bureau_id%22%3A900008135%2C%22sender_person_id%22%3A%22546097%22%2C%22send_status%22%3A1%2C%22bureau_list%22%3A%5B900008135%5D%2C%22org_list%22%3A%5B%5D%2C%22person_list%22%3A%5B%5D%2C%22group_list%22%3A%5B%5D%2C%22attachments%22%3A%5B%5D%7D&noticeSourceJson=333'
headers = {
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:137.0) Gecko/20100101 Firefox/137.0',
'Content-Type': 'application/x-www-form-urlencoded',
'Cookie': 'HWWAFSESID=9e32376ecea1f188cd; HWWAFSESTIME=1744869321310; identity_id=5; person_id=546097; token=ae368e8a2633498a0217c67510668754'
}
response = requests.request("POST", url, headers=headers, data=payload)
print(response.text)
report= f"您的通知已成功发送!"
report += (
f"(只回复您的通知已成功发送即可,其他信息无需回复)"
)
return ActionResponse(Action.REQLLM, report, None)