qsc20001102/astrbot_plugin_jx3
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加避雷功能

Browse Source
This commit is contained in:
qsc
2026-02-07 14:29:46 +08:00
parent 57857aa84e
commit e502ebd5d5
8 changed files with 475 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/bilei_data.py
+220
View File
@@ -0,0 +1,220 @@
# pyright: reportArgumentType=false
# pyright: reportAttributeAccessIssue=false
# pyright: reportIndexIssue=false
# pyright: reportOptionalMemberAccess=false
from datetime import datetime
from typing import Dict, Any, Optional, List, Union
from astrbot.api import logger
from astrbot.api import AstrBotConfig
from .sqlite import AsyncSQLiteDB
from .fun_basic import load_template
class BiLeidata:
def __init__(self,sqlite:AsyncSQLiteDB):
# 引用sqlite
self._sql_db = sqlite
def _init_return_data(self) -> Dict[str, Any]:
"""初始化标准的返回数据结构"""
return {
"code": 0,
"msg": "功能函数未执行",
"data": {}
}
# --- 业务功能函数 ---
async def add(self,name: str, text: str ,user: str) -> Dict[str, Any]:
"""避雷添加"""
return_data = self._init_return_data()
# 获取系统时间
now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
# 添加数据
try:
await self._sql_db.insert(
"bilei",
{
"name": name,
"text": text,
"time": now,
"user": user,
}
)
except FileNotFoundError as e:
logger.error(f"添加避雷失败: {e}")
return_data["msg"] = "添加避雷失败"
return return_data
return_data["data"] = (
"避雷添加成功\n"
f"避雷名称：{name}\n"
f"避雷备注：{text}\n"
f"添加时间：{now}\n"
f"记录人：{user}\n"
)
return_data["code"] = 200
return return_data
async def all(self) -> Dict[str, Any]:
"""避雷查看"""
return_data = self._init_return_data()
# 查询数据
try:
data = await self._sql_db.select_all("bilei")
except FileNotFoundError as e:
logger.error(f"查看避雷失败: {e}")
return_data["msg"] = "查看避雷失败"
return return_data
if not data:
return_data["msg"] = "未找到避雷数据"
return return_data
# 加载模板
try:
return_data["temp"] = await load_template("bilei.html")
except FileNotFoundError as e:
logger.error(f"加载模板失败: {e}")
return_data["msg"] = "系统错误：模板文件不存在"
return return_data
# 数据处理
return_data["data"]["lists"] = data
return_data["code"] = 200
return return_data
async def select(self, name:str) -> Dict[str, Any]:
"""避雷查询 名称"""
return_data = self._init_return_data()
# 模糊拼接
like_name = f"%{name}%"
# 查询数据
try:
data = await self._sql_db.select_all(
"bilei",
"name LIKE ?",
(like_name,)
)
except FileNotFoundError as e:
logger.error(f"查询避雷失败: {e}")
return_data["msg"] = "查询避雷失败"
return return_data
if not data:
return_data["msg"] = "未查询到避雷数据"
return return_data
# 加载模板
try:
return_data["temp"] = await load_template("bilei.html")
except FileNotFoundError as e:
logger.error(f"加载模板失败: {e}")
return_data["msg"] = "系统错误：模板文件不存在"
return return_data
# 数据处理
return_data["data"]["lists"] = data
return_data["code"] = 200
return return_data
async def update(self, id:int, name: str, text: str ,user: str) -> Dict[str, Any]:
"""避雷修改 ID 名称 备注"""
return_data = self._init_return_data()
data = await self._sql_db.select_one(
"bilei",
"id=?",
(id,)
)
if not data:
return_data["msg"] = "没有当前ID"
return return_data
# 获取系统时间
now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
# 修改数据
try:
await self._sql_db.update(
"bilei",
{
"name": name,
"text": text,
"time": now,
"user": user,
},
"id=?",
(id,)
)
except FileNotFoundError as e:
logger.error(f"避雷修改失败: {e}")
return_data["msg"] = "避雷修改失败"
return return_data
return_data["data"] = (
"避雷修改成功\n"
f"ID：{id}\n"
f"避雷名称：{name}\n"
f"避雷备注：{text}\n"
f"修改时间：{now}\n"
f"修改人：{user}\n"
)
return_data["code"] = 200
return return_data
async def delete(self, id:int) -> Dict[str, Any]:
"""避雷删除 ID"""
return_data = self._init_return_data()
data = await self._sql_db.select_one(
"bilei",
"id=?",
(id,)
)
if not data:
return_data["msg"] = "没有当前ID"
return return_data
# 删除
try:
await self._sql_db.delete(
"bilei",
"id=?",
(id,)
)
except FileNotFoundError as e:
logger.error(f"避雷删除失败: {e}")
return_data["msg"] = "避雷删除失败"
return return_data
return_data["data"] = f"避雷删除成功。ID：{id}"
return_data["code"] = 200
return return_data
core/jx3_commands.py
+30 -4
View File
@@ -22,12 +22,13 @@ from astrbot.core.utils.session_waiter import (
from .jx3_data import JX3Service
from .async_task import AsyncTask
from .bilei_data import BiLeidata
class JX3Commands(Star):
def __init__(self, jx3_data:JX3Service, at:AsyncTask, server:str):
def __init__(self, jx3_data:JX3Service, at:AsyncTask, bilei:BiLeidata, server:str):
self.jx3fun = jx3_data
self.jx3at = at
self.bilie = bilei
self.server = server
@@ -307,9 +308,34 @@ class JX3Commands(Star):
return await self.plain_msg(event, lambda: self.jx3fun.pianzhi(qq))
async def jx3_bagua(self, event: AstrMessageEvent,type: str):
async def jx3_bagua(self, event: AstrMessageEvent,name: str,text: str):
"""剑三 八卦 类型"""
return await self.plain_msg(event, lambda: self.jx3fun.bagua(type))
return await self.plain_msg(event, lambda: self.jx3fun.bagua(name))
async def bilei_add(self, event: AstrMessageEvent,name: str, text: str):
"""避雷添加 名称 备注"""
return await self.plain_msg(event, lambda: self.bilie.add(name,text,event.get_sender_name()))
async def bilei_all(self, event: AstrMessageEvent):
"""避雷查看"""
return await self.T2I_image_msg(event, self.bilie.all)
async def bilei_select(self, event: AstrMessageEvent, name:str):
"""避雷查询"""
return await self.T2I_image_msg(event, lambda: self.bilie.select(name))
async def bilei_update(self, event: AstrMessageEvent, id:int, name: str, text: str):
"""避雷修改 ID 名称 备注"""
return await self.plain_msg(event, lambda: self.bilie.update(id,name,text,event.get_sender_name()))
async def bilei_delete(self, event: AstrMessageEvent, id:int):
"""避雷删除 ID"""
return await self.plain_msg(event, lambda: self.bilie.delete(id))
async def jx3_kaifhujiank(self, event: AstrMessageEvent):
core/jx3_data.py
+5 -2
View File
@@ -11,15 +11,17 @@ from astrbot.api import logger
from astrbot.api import AstrBotConfig
from .request import APIClient
from .sqlite import AsyncSQLiteDB
from .fun_basic import load_template,gold_to_string,week_to_num,compare_date_str
class JX3Service:
def __init__(self, api_config, config:AstrBotConfig):
self._api = APIClient()
# 获取API配置文件
# 引用API配置文件
self._api_config = api_config
# 获取插件配置文件
# 引用插件配置文件
self._config = config
# 获取配置中的 Token
self.token = self._config.get("jx3api_token", "")
if self.token == "":
@@ -1484,3 +1486,4 @@ class JX3Service:
return return_data
core/sqlite.py
+69
View File
@@ -0,0 +1,69 @@
import aiosqlite
from typing import Any, Dict, List, Optional, Tuple
class AsyncSQLiteDB:
def __init__(self, db_path: str = "data.db"):
self.db_path = db_path
self.conn: Optional[aiosqlite.Connection] = None
# ======================
# 生命周期
# ======================
async def connect(self):
self.conn = await aiosqlite.connect(self.db_path)
self.conn.row_factory = aiosqlite.Row
async def close(self):
if self.conn:
await self.conn.close()
# ======================
# 基础执行
# ======================
async def execute(self, sql: str, params: Tuple = ()):
async with self.conn.execute(sql, params):
await self.conn.commit()
async def fetch_one(self, sql: str, params: Tuple = ()) -> Optional[Dict[str, Any]]:
async with self.conn.execute(sql, params) as cursor:
row = await cursor.fetchone()
return dict(row) if row else None
async def fetch_all(self, sql: str, params: Tuple = ()) -> List[Dict[str, Any]]:
async with self.conn.execute(sql, params) as cursor:
rows = await cursor.fetchall()
return [dict(r) for r in rows]
# ======================
# CRUD
# ======================
async def insert(self, table: str, data: Dict[str, Any]):
keys = ", ".join(data.keys())
placeholders = ", ".join(["?"] * len(data))
sql = f"INSERT INTO {table} ({keys}) VALUES ({placeholders})"
await self.execute(sql, tuple(data.values()))
async def update(self, table: str, data: Dict[str, Any], where: str, params: Tuple):
set_clause = ", ".join([f"{k}=?" for k in data.keys()])
sql = f"UPDATE {table} SET {set_clause} WHERE {where}"
await self.execute(sql, tuple(data.values()) + params)
async def delete(self, table: str, where: str, params: Tuple):
sql = f"DELETE FROM {table} WHERE {where}"
await self.execute(sql, params)
async def select_one(self, table: str, where: str = "", params: Tuple = ()):
sql = f"SELECT * FROM {table}"
if where:
sql += f" WHERE {where}"
return await self.fetch_one(sql, params)
async def select_all(self, table: str, where: str = "", params: Tuple = ()):
sql = f"SELECT * FROM {table}"
if where:
sql += f" WHERE {where}"
return await self.fetch_all(sql, params)
main.py
+40 -3
View File
@@ -14,8 +14,10 @@ from astrbot.api import logger
from astrbot.api import AstrBotConfig
import astrbot.api.message_components as Comp
from .core.sqlite import AsyncSQLiteDB
from .core.jx3_data import JX3Service
from .core.async_task import AsyncTask
from .core.bilei_data import BiLeidata
from .core.jx3_commands import JX3Commands
@register("astrbot_plugin_jx3",
@@ -33,7 +35,11 @@ class Jx3ApiPlugin(Star):
# 本地数据存储路径
self.local_data_dir = StarTools.get_data_dir("astrbot_plugin_jx3")
# 插件数据文件路径
# SQLite本地路径
self.sqlite_path = Path(self.local_data_dir) /"sqlite.db"
logger.info(f"SQLite数据文件路径：{self.sqlite_path}")
# 插件自带数据文件路径
self.data_file_path = Path(__file__).parent / "data"
# 读取API配置文件
@@ -41,8 +47,9 @@ class Jx3ApiPlugin(Star):
with open(self.api_file_path, 'r', encoding='utf-8') as f:
self.api_config = json.load(f)
# 初始化数据
# 指令前缀
# 指令前缀功能
self.prefix_en = self.conf.get("prefix").get("enable")
self.prefix_text = self.conf.get("prefix").get("text")
if not self.prefix_text:
@@ -78,10 +85,27 @@ class Jx3ApiPlugin(Star):
raise
try:
# sqlite 实例化
self.sql_db = AsyncSQLiteDB(self.sqlite_path)
await self.sql_db.connect()
await self.sql_db.execute("""
CREATE TABLE IF NOT EXISTS bilei(
id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT,
text TEXT,
time TEXT,
user TEXT
)
""")
# 避雷功能 实例化
self.bilei = BiLeidata(self.sql_db)
# 剑三功能 实例化
self.jx3fun = JX3Service(self.api_config, self.conf)
# 后台推送 实例化
self.at = AsyncTask(self.context, self.conf, self.jx3fun)
await self.at.init_tasks()
self.jx3com = JX3Commands(self.jx3fun,self.at,self.server)
# 发送消息实例化
self.jx3com = JX3Commands(self.jx3fun, self.at, self.bilei, self.server)
except Exception as e:
if hasattr(self, "at"):
await self.at.destroy()
@@ -103,6 +127,14 @@ class Jx3ApiPlugin(Star):
if self.jx3fun:
await self.jx3fun.close()
self.jx3fun = None
if self.bilei:
await self.bilei.close()
self.bilei = None
if self.sql_db:
await self.sql_db.close()
self.sql_db = None
logger.info("jx3api插件已卸载/停用")
@@ -258,6 +290,11 @@ class Jx3ApiPlugin(Star):
"刷马": self.jx3com.jx3_shuma,
"骗子": self.jx3com.jx3_pianzhi,
"八卦": self.jx3com.jx3_bagua,
"避雷添加": self.jx3com.bilei_add,
"避雷查看": self.jx3com.bilei_all,
"避雷查询": self.jx3com.bilei_select,
"避雷修改": self.jx3com.bilei_update,
"避雷删除": self.jx3com.bilei_delete,
"开服监控": self.jx3com.jx3_kaifhujiank,
"新闻推送": self.jx3com.jx3_xinwenzhixun,
"刷马推送": self.jx3com.jx3_shuamamsg,
requirements.txt
+1 -1
View File
@@ -1,3 +1,3 @@
aiomysql
matplotlib
aiofiles
aiosqlite
templates/bilei.html
+97
View File
@@ -0,0 +1,97 @@
<!DOCTYPE html>
<html lang="zh-CN">
<head>
<meta charset="UTF-8">
<title>避雷查看</title>
<style>
body {
font-family: 'Microsoft YaHei', Arial, sans-serif;
background: #f1f2f6;
margin: 0;
padding: 20px;
}
.container {
max-width: 1000px;
margin: 0 auto;
text-align: center; /* ★ 让容器内标题更自然居中 */
}
h1 {
font-size: 32px;
margin-bottom: 25px;
color: #222;
font-weight: 800;
text-align: center; /* ★ 强制居中 */
}
table {
width: 100%;
border-collapse: collapse;
background: white;
border-radius: 12px;
overflow: hidden;
box-shadow: 0 4px 16px rgba(0,0,0,0.08);
margin-top: 10px;
text-align: left; /* 不影响表格内容 */
}
thead {
background: #d93939;
color: white;
font-size: 18px;
}
th, td {
padding: 14px 12px;
border-bottom: 1px solid #eee;
font-size: 16px;
}
tbody tr:hover {
background: #fff2f2;
}
.text-col {
max-width: 360px;
line-height: 1.5em;
word-break: break-word;
font-weight: 600;
color: #333;
}
</style>
</head>
<body>
<div class="container">
<h1>避雷查看</h1>
<table>
<thead>
<tr>
<th>ID</th>
<th>避雷名称</th>
<th>避雷备注</th>
<th>添加时间</th>
<th>添加人</th>
</tr>
</thead>
<tbody>
{% for m in lists %}
<tr>
<td>{{ m.id }}</td>
<td>{{ m.name }}</td>
<td class="text-col">{{ m.text }}</td>
<td>{{ m.time }}</td>
<td>{{ m.user }}</td>
</tr>
{% endfor %}
</tbody>
</table>
</div>
</body>
</html>
templates/helps.html
+11
View File
@@ -186,6 +186,17 @@
</div>
</div>
<div class="group">
<div class="group-title">避雷功能</div>
<div class="command-grid">
<div class="command"><div class="cmd-name">避雷查看</div><div class="cmd-usage">避雷查看</div></div>
<div class="command"><div class="cmd-name">避雷添加</div><div class="cmd-usage">避雷添加 名称 备注</div></div>
<div class="command"><div class="cmd-name">避雷删除</div><div class="cmd-usage">避雷删除 ID</div></div>
<div class="command"><div class="cmd-name">避雷修改</div><div class="cmd-usage">避雷修改 ID 名称 备注</div></div>
<div class="command"><div class="cmd-name">避雷查询</div><div class="cmd-usage">避雷查询 名称</div></div>
</div>
</div>
<div class="tip">
<strong>使用提示：</strong><br>
1. 中括号参数为可选内容，不填写则使用程序默认值<br>