summaryrefslogtreecommitdiff
path: root/ATRI/plugins
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI/plugins')
-rw-r--r--ATRI/plugins/anime_search.py29
-rw-r--r--ATRI/plugins/manege/data_source.py66
-rw-r--r--ATRI/plugins/saucenao/data_source.py34
3 files changed, 54 insertions, 75 deletions
diff --git a/ATRI/plugins/anime_search.py b/ATRI/plugins/anime_search.py
index 6e73640..535dec2 100644
--- a/ATRI/plugins/anime_search.py
+++ b/ATRI/plugins/anime_search.py
@@ -1,5 +1,4 @@
import re
-from aiohttp import FormData
from random import choice
from nonebot.typing import T_State
@@ -24,9 +23,10 @@ __doc__ = """
class Anime(Service):
+
def __init__(self):
Service.__init__(self, "以图搜番", __doc__, rule=is_in_service("以图搜番"))
-
+
@staticmethod
async def _request(url: str) -> dict:
aim = URL + url
@@ -36,12 +36,12 @@ class Anime(Service):
raise RequestError("Request failed!")
result = await res.json()
return result
-
+
@classmethod
async def search(cls, url: str) -> str:
data = await cls._request(url)
data = data["result"]
-
+
d = dict()
for i in range(len(data)):
if data[i]["anilist"]["title"]["native"] in d.keys():
@@ -49,7 +49,7 @@ class Anime(Service):
else:
from_m = data[i]["from"] / 60
from_s = data[i]["from"] % 60
-
+
to_m = data[i]["to"] / 60
to_s = data[i]["to"] % 60
@@ -63,7 +63,7 @@ class Anime(Service):
f"第{n}集",
f"约{int(from_m)}min{int(from_s)}s至{int(to_m)}min{int(to_s)}s处",
]
-
+
result = sorted(d.items(), key=lambda x: x[1], reverse=True)
t = 0
msg0 = str()
@@ -76,23 +76,16 @@ class Anime(Service):
f"Name: {i[0]}\n"
f"Time: {i[1][1]} {i[1][2]}"
)
-
+
if len(result) == 2:
return msg0
else:
- data = FormData()
- data.add_field("poster", "ATRI running log")
- data.add_field("syntax", "text")
- data.add_field("expiration", "day")
- data.add_field("content", msg0)
-
- repo = f"详细请移步此处~\n{await UbuntuPaste(data).paste()}"
+ repo = f"详细请移步此处~\n{await UbuntuPaste(content=msg0).paste()}"
return repo
anime_search = Anime().on_command("以图搜番", "发送一张图以搜索可能的番剧")
-
@anime_search.args_parser # type: ignore
async def _get_anime(bot: Bot, event: MessageEvent, state: T_State):
msg = str(event.message).strip()
@@ -104,18 +97,16 @@ async def _get_anime(bot: Bot, event: MessageEvent, state: T_State):
else:
state["anime"] = msg
-
@anime_search.handle()
async def _ready_sear(bot: Bot, event: MessageEvent, state: T_State):
user_id = event.get_user_id()
if not _anime_flmt.check(user_id):
await anime_search.finish(_anime_flmt_notice)
-
+
msg = str(event.message).strip()
if msg:
state["anime"] = msg
-
@anime_search.got("anime", "图呢？")
async def _deal_sear(bot: Bot, event: MessageEvent, state: T_State):
user_id = event.get_user_id()
@@ -123,7 +114,7 @@ async def _deal_sear(bot: Bot, event: MessageEvent, state: T_State):
img = re.findall(r"url=(.*?)]", msg)
if not img:
await anime_search.reject("请发送图片而不是其它东西！！")
-
+
await bot.send(event, "别急，在找了")
a = await Anime().search(img[0])
result = f"> {MessageSegment.at(user_id)}\n" + a
diff --git a/ATRI/plugins/manege/data_source.py b/ATRI/plugins/manege/data_source.py
index de9dfdb..6ce9c60 100644
--- a/ATRI/plugins/manege/data_source.py
+++ b/ATRI/plugins/manege/data_source.py
@@ -1,7 +1,6 @@
import os
import json
from pathlib import Path
-from aiohttp import FormData
from datetime import datetime
from ATRI.service import Service, ServiceTools
@@ -29,6 +28,7 @@ __doc__ = """
class Manege(Service):
+
def __init__(self):
Service.__init__(self, "管理", __doc__, True)
@@ -53,7 +53,8 @@ class Manege(Service):
except BaseException:
data = dict()
return data
-
+
+
@staticmethod
def _save_block_user_list(data: dict) -> None:
file_name = "block_user.json"
@@ -61,10 +62,10 @@ class Manege(Service):
if not path.is_file():
with open(path, "w", encoding="utf-8") as w:
w.write(json.dumps({}))
-
+
with open(path, "w", encoding="utf-8") as w:
w.write(json.dumps(data, indent=4))
-
+
@staticmethod
def _load_block_group_list() -> dict:
"""
@@ -81,13 +82,13 @@ class Manege(Service):
with open(path, "w", encoding="utf-8") as w:
w.write(json.dumps({}))
return dict()
-
+
try:
data = json.loads(path.read_bytes())
except BaseException:
data = dict()
return data
-
+
@staticmethod
def _save_block_group_list(data: dict) -> None:
file_name = "block_group.json"
@@ -95,51 +96,55 @@ class Manege(Service):
if not path.is_file():
with open(path, "w", encoding="utf-8") as w:
w.write(json.dumps({}))
-
+
with open(path, "w", encoding="utf-8") as w:
w.write(json.dumps(data, indent=4))
-
+
@classmethod
def block_user(cls, user_id: str) -> bool:
data = cls._load_block_user_list()
now_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
- data[user_id] = {"time": now_time}
+ data[user_id] = {
+ "time": now_time
+ }
try:
cls._save_block_user_list(data)
return True
except BaseException:
return False
-
+
@classmethod
def unblock_user(cls, user_id: str) -> bool:
data: dict = cls._load_block_user_list()
if user_id not in data:
return False
-
+
try:
data.pop(user_id)
cls._save_block_user_list(data)
return True
except BaseException:
return False
-
+
@classmethod
def block_group(cls, group_id: str) -> bool:
data = cls._load_block_group_list()
now_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
- data[group_id] = {"time": now_time}
+ data[group_id] = {
+ "time": now_time
+ }
try:
cls._save_block_group_list(data)
return True
except BaseException:
return False
-
+
@classmethod
def unblock_group(cls, group_id: str) -> bool:
data: dict = cls._load_block_group_list()
if group_id not in data:
return False
-
+
try:
data.pop(group_id)
cls._save_block_group_list(data)
@@ -159,7 +164,7 @@ class Manege(Service):
data["enabled"] = is_enabled
ServiceTools().save_service(data, service)
return True
-
+
@staticmethod
def control_user_service(service: str, user_id: str, is_enabled: bool) -> bool:
"""
@@ -170,7 +175,7 @@ class Manege(Service):
except BaseException:
return False
temp_list: list = data.get("disable_user", list())
-
+
if is_enabled:
try:
temp_list.remove(user_id)
@@ -181,7 +186,7 @@ class Manege(Service):
data["disable_user"] = temp_list
ServiceTools().save_service(data, service)
return True
-
+
@staticmethod
def control_group_service(service: str, group_id: str, is_enabled: bool) -> bool:
"""
@@ -193,7 +198,7 @@ class Manege(Service):
except BaseException:
return False
temp_list: list = data.get("disable_group", list())
-
+
if is_enabled:
try:
temp_list.remove(group_id)
@@ -219,7 +224,7 @@ class Manege(Service):
except BaseException:
data = dict()
return data
-
+
@staticmethod
def save_friend_apply_list(data: dict) -> None:
file_name = "friend_add.json"
@@ -230,7 +235,7 @@ class Manege(Service):
with open(path, "w", encoding="utf-8") as w:
w.write(json.dumps(data, indent=4))
-
+
@staticmethod
def load_invite_apply_list() -> dict:
file_name = "group_invite.json"
@@ -245,7 +250,7 @@ class Manege(Service):
except BaseException:
data = dict()
return data
-
+
@staticmethod
def save_invite_apply_list(data: dict) -> None:
file_name = "group_invite.json"
@@ -263,19 +268,16 @@ class Manege(Service):
data = load_error(track_id)
except ReadFileError:
return "请检查ID是否正确..."
-
+
prompt = data.get("prompt", "ignore")
time = data.get("time", "ignore")
content = data.get("content", "ignore")
-
+
msg0 = TRACK_BACK_FORMAT.format(
- track_id=track_id, prompt=prompt, time=time, content=content
+ track_id=track_id,
+ prompt=prompt,
+ time=time,
+ content=content
)
- f_data = FormData()
- f_data.add_field("poster", "ATRI running log")
- f_data.add_field("syntax", "text")
- f_data.add_field("expiration", "day")
- f_data.add_field("content", msg0)
-
- repo = f"详细请移步此处~\n{await UbuntuPaste(f_data).paste()}"
+ repo = f"详细请移步此处~\n{await UbuntuPaste(content=msg0).paste()}"
return repo
diff --git a/ATRI/plugins/saucenao/data_source.py b/ATRI/plugins/saucenao/data_source.py
index 6e72c98..58b031b 100644
--- a/ATRI/plugins/saucenao/data_source.py
+++ b/ATRI/plugins/saucenao/data_source.py
@@ -1,5 +1,4 @@
from random import choice
-from aiohttp import FormData
from ATRI.service import Service
from ATRI.rule import is_in_service
@@ -16,15 +15,8 @@ __doc__ = """
class SaouceNao(Service):
- def __init__(
- self,
- api_key: str = None,
- output_type=2,
- testmode=1,
- dbmaski=32768,
- db=5,
- numres=5,
- ):
+
+ def __init__(self, api_key: str = None, output_type=2, testmode=1, dbmaski=32768, db=5, numres=5):
Service.__init__(self, "以图搜图", __doc__, rule=is_in_service("以图搜图"))
params = dict()
@@ -35,31 +27,31 @@ class SaouceNao(Service):
params["db"] = db
params["numres"] = numres
self.params = params
-
+
async def _request(self, url: str):
self.params["url"] = url
-
+
try:
res = await request.post(URL, params=self.params)
except RequestError:
raise RequestError("Request failed!")
data = await res.json()
return data
-
+
async def search(self, url: str) -> str:
data = await self._request(url)
res = data["results"]
-
+
result = list()
for i in range(len(res)):
data = res[i]
-
+
_result = dict()
_result["similarity"] = data["header"]["similarity"]
_result["index_name"] = data["header"]["index_name"]
_result["url"] = choice(data["data"].get("ext_urls", ["None"]))
result.append(_result)
-
+
msg0 = str()
for i in result:
msg0 += (
@@ -68,15 +60,9 @@ class SaouceNao(Service):
f"Name: {i['index_name']}\n"
f"URL: {i['url'].replace('https://', '')}"
)
-
+
if len(res) <= 3:
return msg0
else:
- data = FormData()
- data.add_field("poster", "ATRI running log")
- data.add_field("syntax", "text")
- data.add_field("expiration", "day")
- data.add_field("content", msg0)
-
- repo = f"\n详细请移步此处~\n{await UbuntuPaste(data).paste()}"
+ repo = f"\n详细请移步此处~\n{await UbuntuPaste(content=msg0).paste()}"
return repo
Powered by cgit, Theme cgit-dark created by Ted Yin, updated by Mole Shang.