-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
🐛 fix:修复了下载卫星数据的bug ✨ feat: 支持自定义截图，命令`/s`
- Loading branch information
Showing
7 changed files
with
257 additions
and
76 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,128 @@ | ||
| # python3 | ||
| # -*- coding: utf-8 -*- | ||
| # @Time : 2023-09-29 16:19 | ||
| # @Author : yzyyz | ||
| # @Email : youzyyz1384@qq.com | ||
| # @File : shoot_handle.py | ||
| # @Software: PyCharm | ||
| import re | ||
| from datetime import datetime, timedelta | ||
|
|
||
| from nonebot import on_command, require | ||
| from nonebot.adapters import Message | ||
| from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent, MessageSegment, MessageEvent | ||
| from nonebot.adapters.onebot.v11.exception import ActionFailed | ||
| from nonebot.internal.params import ArgStr | ||
| from nonebot.params import CommandArg | ||
| from nonebot.permission import SUPERUSER | ||
| from nonebot.typing import T_State | ||
|
|
||
| from .calculate_sat import download_ham_sat, data2Tle, calculate, get_tian_gong | ||
| from .sat_shoot_url_bank import * | ||
| from ..log import log | ||
| from ..path import * | ||
| from ..config import * | ||
| from ..utils import * | ||
|
|
||
| if plugin_config.sat_proxy_url: | ||
| logger.info(plugin_config.sat_proxy_url) | ||
|
|
||
| shoot_web = on_command("/s", aliases={"/S"}, block=True) | ||
|
|
||
| url_bank = UrlBank() | ||
|
|
||
|
|
||
| @shoot_web.handle() | ||
| async def query_tevel_(bot: Bot, event: MessageEvent, state: T_State, args: Message = CommandArg()): | ||
| not_found = True | ||
| Iter_msgs = [] | ||
| for item in (url_bank.get()): | ||
| tips = "" | ||
| for k, v in item.items(): | ||
| if k in ['url', 'cmd']: | ||
| tips += f"{k}:{v}\n" | ||
| Iter_msgs.append(tips) | ||
| if args: | ||
| args = str(args).strip().split(" ") | ||
| for item in (url_bank.get()): | ||
| for arg in args: | ||
| if arg in item["cmd"]: | ||
| not_found = False | ||
| url = item["url"] | ||
| path = SHOOTS_OUT_PATH / item["path"] | ||
| locator = item.get("locator", "html") | ||
| time_out = item.get("time_out", 0) | ||
| proxy = item.get("proxy", None) | ||
| timeout = item.get("timeout", 30000) | ||
| if path.exists(): | ||
| last_time = datetime.fromtimestamp(path.stat().st_mtime) | ||
| TIME_NOW = (datetime.now()) | ||
| TIME_DIFF = TIME_NOW - last_time | ||
| if TIME_DIFF > timedelta(minutes=float(time_out)): | ||
| logger.info("超时，重新截图") | ||
| await shoot_scr(url, locator=locator, img_output=path, proxy=proxy, timeout=timeout) | ||
| else: | ||
| await shoot_scr(url, locator=locator, img_output=path, proxy=proxy, timeout=timeout) | ||
| try: | ||
| await shoot_web.send(MessageSegment.image(f"file:///{Path(path).resolve()}")) | ||
| except Exception as e: | ||
| await shoot_web.finish(f"图片发送失败，哪里出错了？\n{e}") | ||
|
|
||
| if not args or not_found: | ||
| await shoot_web.send(f"请输入参数：\n例如：/s t a css\n支持的参数：") | ||
| await send_forward_msg(bot, event, "收录截图列表", Iter_msgs) | ||
|
|
||
|
|
||
| bank_handle = on_command("/截图", aliases={"/shoot"}, block=True, permission=SUPERUSER) | ||
|
|
||
|
|
||
| @bank_handle.handle() | ||
| async def _(bot: Bot, event: MessageEvent, state: T_State, args: Message = CommandArg()): | ||
| if args: | ||
|
|
||
| # args: add url=xxx cmd=xxx path=xxx locator=xxx | ||
| # 检查是否有中文、中文符号 | ||
| if re.findall(r'[\u4e00-\u9fa5，。？！：；“”‘’（）【】、]', str(args)): | ||
| await bank_handle.finish( | ||
| "参数错误：不得包含中文字符\n/截图add url=xxx cmd=xxx path=xxx locator=xxx time_out=123 ...") | ||
| args = str(args).strip().split(" ") | ||
| keyword_ = {} | ||
| if args[0] == "add": | ||
| for arg in args: | ||
| if "=" in arg: | ||
| arg = arg.split("=") | ||
| keyword_[arg[0]] = (arg[1]).replace("]", "]").replace("[", "[").replace("=", "=") | ||
| if keyword_: | ||
| if not keyword_.get("url", None) or not keyword_.get("cmd", None) or not keyword_.get("path", None): | ||
| await bank_handle.finish("参数错误：\n/截图add url=xxx cmd=xxx path=xxx locator=xxx time_out=123 " | ||
| "...\n必须包含url、cmd、path") | ||
| url_bank.add(**keyword_) | ||
| await bank_handle.finish("添加成功") | ||
| else: | ||
| await bank_handle.finish("参数错误：\n/截图add url=xxx cmd=xxx path=xxx locator=xxx time_out=123 " | ||
| "...\n必须包含url、cmd、path") | ||
| elif args[0] == "del": | ||
| for arg in args: | ||
| if "=" in arg: | ||
| arg = arg.split("=") | ||
| keyword_[arg[0]] = arg[1].replace("]", "]").replace("[", "[").replace("=", "=") | ||
| if keyword_: | ||
| url_bank.remove(**keyword_) | ||
| await bank_handle.finish("删除成功") | ||
| else: | ||
| await bank_handle.finish("参数错误：\n/截图del url=xxx cmd=xxx path=xxx" | ||
| "...\n必须包含url、cmd、path其中之一") | ||
| elif args[0] == "get": | ||
| data = url_bank.get() | ||
| messages = [] | ||
| for i in data: | ||
| reply = "" | ||
| for k, v in i.items(): | ||
| if k in ['cmd', 'url', 'path']: | ||
| reply += f"{k}:{v}\n" | ||
| messages.append(reply) | ||
|
|
||
| await send_forward_msg(bot, event, "收录截图列表", messages) | ||
| elif args[0] == "default": | ||
| url_bank.default() | ||
| await bank_handle.finish("恢复默认成功") |
Oops, something went wrong.