From df9e14de6003e894b0c2ff226d5cf1bcebb71820 Mon Sep 17 00:00:00 2001 From: iouAkira Date: Wed, 24 Mar 2021 23:02:09 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E9=80=9A=E8=BF=87bot?= =?UTF-8?q?=E4=B8=8A=E4=BC=A0js=E6=96=87=E4=BB=B6=E5=88=B0/scripts?= =?UTF-8?q?=E7=9B=AE=E5=BD=95=E5=B9=B6=E5=8F=AF=E4=BB=A5=E9=80=89=E6=8B=A9?= =?UTF-8?q?=E6=89=A7=E8=A1=8C=E7=9A=84=E4=BA=A4=E4=BA=92?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docker/bot/jd_bot | 219 ++++++++++++++++++++++++++-------------------- 1 file changed, 125 insertions(+), 94 deletions(-) diff --git a/docker/bot/jd_bot b/docker/bot/jd_bot index b45d7c68..bd131d80 100644 --- a/docker/bot/jd_bot +++ b/docker/bot/jd_bot @@ -951,129 +951,160 @@ def parseJDCookies(headers): jd_cookie = f'pt_key={pt_key};pt_pin={pt_pin};' -def unknown(update, context): - """回复用户输入不存在的指令 - """ +def saveFile(update, context): from_user_id = update.message.from_user.id if admin_id == str(from_user_id): - spnode_readme = "" - if "DISABLE_SPNODE" not in os.environ: - spnode_readme = "/spnode 获取可执行脚本的列表，选择对应的按钮执行。(拓展使用：运行指定路径脚本，例：/spnode /scripts/jd_818.js)\n\n" \ - "使用bot交互+spnode后 后续用户的cookie维护更新只需要更新logs/cookies.list即可\n" \ - "使用bot交互+spnode后 后续执行脚本命令请使用spnode否者无法使用logs/cookies.list的cookies执行脚本，定时任务也将自动替换为spnode命令执行\n" \ - "spnode功能概述示例\n\n" \ - "spnode conc /scripts/jd_bean_change.js 为每个cookie单独执行jd_bean_change脚本（伪并发\n" \ - "spnode 1 /scripts/jd_bean_change.js 为logs/cookies.list文件里面第一行cookie账户单独执行jd_bean_change脚本\n" \ - "spnode jd_XXXX /scripts/jd_bean_change.js 为logs/cookies.list文件里面pt_pin=jd_XXXX的cookie账户单独执行jd_bean_change脚本\n" \ - "spnode /scripts/jd_bean_change.js 为logs/cookies.list所有cookies账户一起执行jd_bean_change脚本\n" \ - "请仔细阅读并理解上面的内容，使用bot交互默认开启spnode指令功能功能。\n" \ - "如需____停用___请配置环境变量 -DISABLE_SPNODE=True" - update.message.reply_text(text="⚠️ 您输入了一个错误的指令，请参考说明使用\n" \ - "\n" \ - "/start 开始并获取指令说明\n" \ - "/node 获取可执行脚本的列表，选择对应的按钮执行。(拓展使用：运行指定路径脚本，例：/node /scripts/jd_818.js) \n" \ - "/git 获取可执行git指令列表，选择对应的按钮执行。(拓展使用：运行指定路径脚本，例：/git -C /scripts/ pull)\n" \ - "/logs 获取logs下的日志文件列表，选择对应名字可以下载日志文件\n" \ - "/env 获取系统环境变量列表。(拓展使用：设置系统环境变量，例：/env export JD_DEBUG=true，环境变量只针对当前bot进程生效) \n" \ - "/cmd 执行执行命令。参考：/cmd ls -l 涉及目录文件操作请使用绝对路径,部分shell命令开放使用\n" \ - "/gen_long_code 长期活动互助码提交消息生成\n" \ - "/gen_temp_code 短期临时活动互助码提交消息生成\n" \ - "/gen_daily_code 每天变化互助码活动提交消息生成\n\n%s" % spnode_readme, - parse_mode=ParseMode.HTML) - else: - update.message.reply_text(text='此为私人使用bot,不能执行您的指令！') + js_file_name = update.message.document.file_name + if str(js_file_name).endswith(".js"): + save_path = f"{_base_dir}{js_file_name}" + try: + # logger.info(update.message) + file = context.bot.getFile(update.message.document.file_id) + file.download(save_path) + keyboard_line = [[InlineKeyboardButton('node 执行', callback_data='node %s ' % save_path), + InlineKeyboardButton('spnode 执行', + callback_data='spnode %s' % save_path)], + [InlineKeyboardButton('取消操作', callback_data='cancel')]] + reply_markup = InlineKeyboardMarkup(keyboard_line) -def error(update, context): - """Log Errors caused by Updates.""" - logger.warning('Update "%s" caused error "%s"', update, context.error) - context.bot.send_message( - 'Update "%s" caused error "%s"', update, context.error) + update.message.reply_text( + text='```{}```'.format( + helpers.escape_markdown(' ↓↓↓ %s 上传至/scripts完成，请请选择需要的操作 ↓↓↓ ' % js_file_name)), + reply_markup=reply_markup, parse_mode=ParseMode.MARKDOWN_V2) + except Exception as e: + update.message.reply_text(text='```{}```'.format( + helpers.escape_markdown(" →→→ %s js上传至/scripts过程中出错，请重新尝试。 ←←← " % js_file_name)), + parse_mode=ParseMode.MARKDOWN_V2) + else: + update.message.reply_text(text='```{}```'.format( + helpers.escape_markdown(" →→→ 抱歉，暂时只开放上传js文件至/scripts目录 ←←← ")), + parse_mode=ParseMode.MARKDOWN_V2) -def main(): - global admin_id, bot_token, crontab_list_file + def unknown(update, context): + """回复用户输入不存在的指令 + """ + from_user_id = update.message.from_user.id + if admin_id == str(from_user_id): + spnode_readme = "" + if "DISABLE_SPNODE" not in os.environ: + spnode_readme = "/spnode 获取可执行脚本的列表，选择对应的按钮执行。(拓展使用：运行指定路径脚本，例：/spnode /scripts/jd_818.js)\n\n" \ + "使用bot交互+spnode后 后续用户的cookie维护更新只需要更新logs/cookies.list即可\n" \ + "使用bot交互+spnode后 后续执行脚本命令请使用spnode否者无法使用logs/cookies.list的cookies执行脚本，定时任务也将自动替换为spnode命令执行\n" \ + "spnode功能概述示例\n\n" \ + "spnode conc /scripts/jd_bean_change.js 为每个cookie单独执行jd_bean_change脚本（伪并发\n" \ + "spnode 1 /scripts/jd_bean_change.js 为logs/cookies.list文件里面第一行cookie账户单独执行jd_bean_change脚本\n" \ + "spnode jd_XXXX /scripts/jd_bean_change.js 为logs/cookies.list文件里面pt_pin=jd_XXXX的cookie账户单独执行jd_bean_change脚本\n" \ + "spnode /scripts/jd_bean_change.js 为logs/cookies.list所有cookies账户一起执行jd_bean_change脚本\n" \ + "请仔细阅读并理解上面的内容，使用bot交互默认开启spnode指令功能功能。\n" \ + "如需____停用___请配置环境变量 -DISABLE_SPNODE=True" + update.message.reply_text(text="⚠️ 您输入了一个错误的指令，请参考说明使用\n" \ + "\n" \ + "/start 开始并获取指令说明\n" \ + "/node 获取可执行脚本的列表，选择对应的按钮执行。(拓展使用：运行指定路径脚本，例：/node /scripts/jd_818.js) \n" \ + "/git 获取可执行git指令列表，选择对应的按钮执行。(拓展使用：运行指定路径脚本，例：/git -C /scripts/ pull)\n" \ + "/logs 获取logs下的日志文件列表，选择对应名字可以下载日志文件\n" \ + "/env 获取系统环境变量列表。(拓展使用：设置系统环境变量，例：/env export JD_DEBUG=true，环境变量只针对当前bot进程生效) \n" \ + "/cmd 执行执行命令。参考：/cmd ls -l 涉及目录文件操作请使用绝对路径,部分shell命令开放使用\n" \ + "/gen_long_code 长期活动互助码提交消息生成\n" \ + "/gen_temp_code 短期临时活动互助码提交消息生成\n" \ + "/gen_daily_code 每天变化互助码活动提交消息生成\n\n%s" % spnode_readme, + parse_mode=ParseMode.HTML) + else: + update.message.reply_text(text='此为私人使用bot,不能执行您的指令！') - if 'TG_BOT_TOKEN' in os.environ: - bot_token = os.getenv('TG_BOT_TOKEN') + def error(update, context): + """Log Errors caused by Updates.""" + logger.warning('Update "%s" caused error "%s"', update, context.error) + context.bot.send_message( + 'Update "%s" caused error "%s"', update, context.error) - if 'TG_USER_ID' in os.environ: - admin_id = os.getenv('TG_USER_ID') + def main(): + global admin_id, bot_token, crontab_list_file - if 'CRONTAB_LIST_FILE' in os.environ: - crontab_list_file = os.getenv('CRONTAB_LIST_FILE') - else: - crontab_list_file = 'crontab_list.sh' + if 'TG_BOT_TOKEN' in os.environ: + bot_token = os.getenv('TG_BOT_TOKEN') - logger.info('CRONTAB_LIST_FILE=' + crontab_list_file) + if 'TG_USER_ID' in os.environ: + admin_id = os.getenv('TG_USER_ID') - # 创建更新程序并参数为你Bot的TOKEN。 - updater = Updater(bot_token, use_context=True) + if 'CRONTAB_LIST_FILE' in os.environ: + crontab_list_file = os.getenv('CRONTAB_LIST_FILE') + else: + crontab_list_file = 'crontab_list.sh' - # 获取调度程序来注册处理程序 - dp = updater.dispatcher + logger.info('CRONTAB_LIST_FILE=' + crontab_list_file) - # 通过 start 函数 响应 '/start' 命令 - dp.add_handler(CommandHandler('start', start)) + # 创建更新程序并参数为你Bot的TOKEN。 + updater = Updater(bot_token, use_context=True) - # 通过 start 函数 响应 '/help' 命令 - dp.add_handler(CommandHandler('help', start)) + # 获取调度程序来注册处理程序 + dp = updater.dispatcher - # 通过 node 函数 响应 '/node' 命令 - dp.add_handler(CommandHandler('node', node)) + # 通过 start 函数 响应 '/start' 命令 + dp.add_handler(CommandHandler('start', start)) - # 通过 node 函数 响应 '/spnode' 命令 - dp.add_handler(CommandHandler('spnode', spnode)) + # 通过 start 函数 响应 '/help' 命令 + dp.add_handler(CommandHandler('help', start)) - # 通过 git 函数 响应 '/git' 命令 - dp.add_handler(CommandHandler('git', git)) + # 通过 node 函数 响应 '/node' 命令 + dp.add_handler(CommandHandler('node', node)) - # 通过 crontab 函数 响应 '/crontab' 命令 - dp.add_handler(CommandHandler('crontab', crontab)) + # 通过 node 函数 响应 '/spnode' 命令 + dp.add_handler(CommandHandler('spnode', spnode)) - # 通过 logs 函数 响应 '/logs' 命令 - dp.add_handler(CommandHandler('logs', logs)) + # 通过 git 函数 响应 '/git' 命令 + dp.add_handler(CommandHandler('git', git)) - # 通过 cmd 函数 响应 '/cmd' 命令 - dp.add_handler(CommandHandler('cmd', shcmd)) + # 通过 crontab 函数 响应 '/crontab' 命令 + dp.add_handler(CommandHandler('crontab', crontab)) - # 通过 callback_run 函数 响应相关按钮命令 - dp.add_handler(CallbackQueryHandler(callback_run)) + # 通过 logs 函数 响应 '/logs' 命令 + dp.add_handler(CommandHandler('logs', logs)) - # 通过 env 函数 响应 '/env' 命令 - dp.add_handler(CommandHandler('env', env)) + # 通过 cmd 函数 响应 '/cmd' 命令 + dp.add_handler(CommandHandler('cmd', shcmd)) - # 通过 gen_long_code 函数 响应 '/gen_long_code' 命令 - dp.add_handler(CommandHandler('gen_long_code', gen_long_code)) + # 通过 callback_run 函数 响应相关按钮命令 + dp.add_handler(CallbackQueryHandler(callback_run)) - # 通过 gen_temp_code 函数 响应 '/gen_temp_code' 命令 - dp.add_handler(CommandHandler('gen_temp_code', gen_temp_code)) + # 通过 env 函数 响应 '/env' 命令 + dp.add_handler(CommandHandler('env', env)) - # 通过 gen_daily_code 函数 响应 '/gen_daily_code' 命令 - dp.add_handler(CommandHandler('gen_daily_code', gen_daily_code)) + # 通过 gen_long_code 函数 响应 '/gen_long_code' 命令 + dp.add_handler(CommandHandler('gen_long_code', gen_long_code)) - # 通过 get_jd_cookie 函数 响应 '/eikooc_dj_teg' 命令 #别问为啥这么写，有意为之的 - dp.add_handler(CommandHandler('eikooc_dj_teg', get_jd_cookie)) + # 通过 gen_temp_code 函数 响应 '/gen_temp_code' 命令 + dp.add_handler(CommandHandler('gen_temp_code', gen_temp_code)) - # 没找到对应指令 - dp.add_handler(MessageHandler(Filters.command, unknown)) + # 通过 gen_daily_code 函数 响应 '/gen_daily_code' 命令 + dp.add_handler(CommandHandler('gen_daily_code', gen_daily_code)) - # 响应普通文本消息 - # dp.add_handler(MessageHandler(Filters.text, resp_text)) + # 通过 get_jd_cookie 函数 响应 '/eikooc_dj_teg' 命令 #别问为啥这么写，有意为之的 + dp.add_handler(CommandHandler('eikooc_dj_teg', get_jd_cookie)) - dp.add_error_handler(error) + # 文件监听 + dp.add_handler(MessageHandler(Filters.document, saveFile)) - updater.start_polling() - updater.idle() + # 没找到对应指令 + dp.add_handler(MessageHandler(Filters.command, unknown)) + # 响应普通文本消息 + # dp.add_handler(MessageHandler(Filters.text, resp_text)) -# 生成依赖安装列表 -# pip3 freeze > requirements.txt -# 或者使用pipreqs -# pip3 install pipreqs -# 在当前目录生成 -# pipreqs . --encoding=utf8 --force -# 使用requirements.txt安装依赖 -# pip3 install -r requirements.txt -if __name__ == '__main__': - main() + dp.add_error_handler(error) + + updater.start_polling() + updater.idle() + + # 生成依赖安装列表 + # pip3 freeze > requirements.txt + # 或者使用pipreqs + # pip3 install pipreqs + # 在当前目录生成 + # pipreqs . --encoding=utf8 --force + # 使用requirements.txt安装依赖 + # pip3 install -r requirements.txt + if __name__ == '__main__': + main()