yan
/
mdserver-web
mirror of https://github.com/midoks/mdserver-web.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

tgbot加入扩展机制

This commit is contained in:
midoks 2023-03-14 00:09:30 +08:00
parent da90386816
commit c9927c3fb9
5 changed files with 187 additions and 79 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
plugins/tgbot/index.html
View File

@ -19,7 +19,7 @@
}
</style>
<script type="text/javascript">
resetPluginWinHeight(300);
resetPluginWinHeight(350);
$.getScript( "/plugins/file?name=tgbot&f=js/tgbot.js", function(){
pluginService('tgbot');
});

85
plugins/tgbot/index.py
View File

@ -58,10 +58,7 @@ def getExtCfg():
def writeExtCfg(data):
cfg_path = getServerDir() + "/extend.cfg"
if not os.path.exists(cfg_path):
mw.writeFile(cfg_path, '{}')
t = mw.readFile(cfg_path)
return json.loads(t)
return mw.writeFile(cfg_path, json.dumps(data))
def getInitDTpl():
@ -147,9 +144,9 @@ def tbOp(method):
return data[1]
data = mw.execShell(file + ' ' + method)
print(data)
# if data[1] == '':
# return 'ok'
# print(data)
if data[1] == '':
return 'ok'
return 'ok'
@ -243,7 +240,21 @@ def uninstallPreInspection():
return "请手动删除<br/> rm -rf {}".format(getServerDir())
def getExtCfgByName(name):
elist = getExtCfg()
for x in elist:
if x['name'] == name:
return x
return None
def botExtList():
args = getArgs()
data_args = checkArgs(args, ['p'])
if not data_args[0]:
return data_args[1]
ext_path = getServerDir() + '/extend'
if not os.path.exists(ext_path):
return mw.returnJson(False, 'ok', [])
@ -254,9 +265,63 @@ def botExtList():
if e.endswith('py'):
elist.append(e)
# extList = getExtCfg()
page = int(args['p'])
page_size = 5
print(elist)
make_ext_list = []
for ex in elist:
tmp = {}
tmp['name'] = ex
edata = getExtCfgByName(ex)
if edata:
tmp['status'] = edata['status']
else:
tmp['status'] = 'stop'
tmp['tag'] = ex.split('_')[0]
make_ext_list.append(tmp)
writeExtCfg(make_ext_list)
dlist_sum = len(make_ext_list)
page_start = int((page - 1) * page_size)
page_end = page_start + page_size
if page_end >= dlist_sum:
ret_data = make_ext_list[page_start:]
else:
ret_data = make_ext_list[page_start:page_end]
data = {}
data['data'] = ret_data
data['args'] = args
data['list'] = mw.getPage(
{'count': dlist_sum, 'p': page, 'row': page_size, 'tojs': 'botExtListP'})
return mw.returnJson(True, 'ok', data)
def setExtStatus():
args = getArgs()
data_args = checkArgs(args, ['name', 'status'])
if not data_args[0]:
return data_args[1]
elist = getExtCfg()
name = args['name']
status = args['status']
for x in range(len(elist)):
if elist[x]['name'] == name:
elist[x]['status'] = status
break
writeExtCfg(elist)
action = '开启'
if status == 'stop':
action = '关闭'
return mw.returnJson(True, action + '[' + name + ']扩展成功')
def runLog():
@ -292,6 +357,8 @@ if __name__ == "__main__":
print(setBotConf())
elif func == 'bot_ext_list':
print(botExtList())
elif func == 'set_ext_status':
print(setExtStatus())
elif func == 'run_log':
print(runLog())

96
plugins/tgbot/js/tgbot.js
View File

@ -86,7 +86,6 @@ function botExtList(){
<th width="20">脚本</th>\
<th width="120">类型</th>\
<th width="10">状态</th>\
<th style="text-align: right;" width="50">操作</th>\
</tr>\
</thead>\
<tbody id="ext_list"></tbody>\
@ -96,55 +95,54 @@ function botExtList(){
</div>\
</div>';
$('.soft-man-con').html(body);
botExtListP(1)
botExtListP(1);
}
function botExtListP(p=1){
appPost('bot_ext_list',{'page':p}, function(rdata){
function setBotExtStatus(name,status){
appPost('set_ext_status',{'name':name,'status':status}, function(rdata){
var rdata = $.parseJSON(rdata.data);
var tBody = '';
if (rdata.data.length == 0 ){
var tBody = '<tr><td colspan="4"><div style="text-align:center;">无数据</div></td></tr>';
}
var ldata = rdata.data;
for (var i = 0; i < ldata.length; i++) {
tBody += '<tr data-id="'+ldata[i]['name']+'">'
tBody += '<td>'+ldata[i]['id']+'</td>';
tBody += '<td>'+ldata[i]['name']+'</td>';
if (ldata[i]['status'] == 'start'){
tBody += '<td><span style="color:#20a53a;cursor: pointer;" class="strategy_status glyphicon glyphicon-play"></span></td>';
} else{
tBody += '<td><span style="color:red;cursor: pointer;" class="strategy_status glyphicon glyphicon-pause"></span></td>';
}
tBody += "<td style='text-align: right;'><a class='btlink restart'>重启</a> | <a class='btlink edit'>编辑</a></td>";
tBody +='<tr>';
};
$('#ext_list').html(tBody);
// $('#strategy_list_page').html(rdata.data.list);
// $('#strategy_list .strategy_status').click(function(){
// var id = $(this).parent().parent().data('id');
// var status = 'stop';
// if ($(this).hasClass('glyphicon-pause')){
// status = 'start';
// }
// setStrategyStatus(id,status);
// });
// $('#strategy_list .restart').click(function(){
// var id = $(this).parent().parent().data('id');
// setStrategyRestart(id);
// });
// $('#strategy_list .edit').click(function(){
// var id = $(this).parent().parent().data('id');
// setStrategyEdit(id);
// });
layer.msg(rdata['msg'],);
showMsg(rdata['msg'], function(){
botExtListP(1);
},{icon:rdata['status']?1:2,shade: [0.3, '#000']},2000);
});
}
function botExtListP(p=1){
appPost('bot_ext_list',{'p':p}, function(rdata){
// console.log(rdata);
var rdata = $.parseJSON(rdata.data);
// console.log(rdata);
var tBody = '';
if (!rdata.status && rdata.data.length == 0 ){
var tBody = '<tr><td colspan="4"><div style="text-align:center;">无数据</div></td></tr>';
} else{
var ldata = rdata.data.data;
for (var i = 0; i < ldata.length; i++) {
tBody += '<tr data-name="'+ldata[i]['name']+'">'
tBody += '<td>'+ldata[i]['name']+'</td>';
tBody += '<td>'+ldata[i]['tag']+'</td>';
if (ldata[i]['status'] == 'start'){
tBody += '<td><span style="color:#20a53a;cursor: pointer;" class="ext_status glyphicon glyphicon-play"></span></td>';
} else{
tBody += '<td><span style="color:red;cursor: pointer;" class="ext_status glyphicon glyphicon-pause"></span></td>';
}
tBody +='<tr>';
}
}
$('#ext_list').html(tBody);
$('#ext_list_page').html(rdata.data.list);
$('#ext_list .ext_status').click(function(){
var name = $(this).parent().parent().data('name');
var status = 'stop';
if ($(this).hasClass('glyphicon-pause')){
status = 'start';
}
setBotExtStatus(name,status);
});
});
}

14
plugins/tgbot/startup/extend/receive_faq.py Normal file
View File

@ -0,0 +1,14 @@
import sys
import io
import os
import time
import re
import json
import base64
import threading
import telebot
def run(bot, message):
return bot

69
plugins/tgbot/startup/tgbot.py
View File

@ -28,6 +28,9 @@ def getServerDir():
return mw.getServerDir() + '/' + getPluginName()
sys.path.append(getServerDir() + "/extend")
def getConfigData():
cfg_path = getServerDir() + "/data.cfg"
if not os.path.exists(cfg_path):
@ -42,6 +45,24 @@ def writeConf(data):
return True
def getExtCfg():
cfg_path = getServerDir() + "/extend.cfg"
if not os.path.exists(cfg_path):
mw.writeFile(cfg_path, '{}')
t = mw.readFile(cfg_path)
return json.loads(t)
def getStartExtCfgByTag(tag='push'):
# 获取开启的扩展
elist = getExtCfg()
rlist = []
for x in elist:
if x['tag'] == tag and x['status'] == 'start':
rlist.append(x)
return rlist
def writeLog(log_str):
if __name__ == "__main__":
print(log_str)
@ -60,6 +81,7 @@ while True:
writeLog('等待输入配置,填写app_token')
time.sleep(3)
bot = telebot.TeleBot(cfg['bot']['app_token'])
@ -76,9 +98,6 @@ bot.set_my_commands(
telebot.types.BotCommand("start", "查看帮助信息"),
telebot.types.BotCommand("mw_chat_id", "查看群组ChatID")
],
# scope=telebot.types.BotCommandScopeChat(12345678) # use for personal command for users
# scope=telebot.types.BotCommandScopeAllPrivateChats() # use for all
# private chats
)
@ -87,13 +106,8 @@ def hanle_start_help(message):
bot.reply_to(message, "hello world")
@bot.message_handler(commands=['mw'])
def hanle_start_mw(message):
bot.reply_to(message, "我就是最靓的仔!")
@bot.message_handler(commands=['mw_echo'])
def hanle_start_help(message):
def hanle_mw_echo(message):
bot.reply_to(message, message.text)
@ -103,9 +117,30 @@ def hanle_get_chat_id(message):
@bot.message_handler(func=lambda message: True)
def echo_message(message):
print(message)
# bot.reply_to(message, "拦截所有消息:" + message.text)
def all_message(message):
rlist = getStartExtCfgByTag('receive')
for r in rlist:
try:
script = r['name'].split('.')[0]
__import__(script).run(bot, message)
except Exception as e:
pass
def runBotPushTask():
plist = getStartExtCfgByTag('push')
for p in plist:
try:
script = p['name'].split('.')[0]
__import__(script).run(bot)
except Exception as e:
pass
def botPush():
while True:
runBotPushTask()
time.sleep(3)
def setDaemon(t):
@ -113,19 +148,13 @@ def setDaemon(t):
t.daemon = True
else:
t.setDaemon(True)
return t
def botPush():
while True:
print('a')
time.sleep(3)
if __name__ == "__main__":
# 机器人推送任务
botPushTask = threading.Thread(target=botPush)
botPushTask = setDaemon(botPushTask)
# print(botPushTask)
# botPushTask = setDaemon(botPushTask)
botPushTask.start()
writeLog('启动成功')