diff options
| author | Kyomotoi <1172294279@qq.com> | 2020-08-12 00:05:21 +0800 |
|---|---|---|
| committer | Kyomotoi <1172294279@qq.com> | 2020-08-12 00:05:21 +0800 |
| commit | b11b204043451334c33d827b776f230b6c8e84ce (patch) | |
| tree | 32a1e42a3871e34401775bf7843444f3eeaa904a /ATRI/plugins/pixiv.py | |
| parent | 0cd82d9166c38fa455895cc236f406902e8e6549 (diff) | |
| download | ATRI-b11b204043451334c33d827b776f230b6c8e84ce.tar.gz ATRI-b11b204043451334c33d827b776f230b6c8e84ce.tar.bz2 ATRI-b11b204043451334c33d827b776f230b6c8e84ce.zip | |
[Fix]
Diffstat (limited to 'ATRI/plugins/pixiv.py')
| -rw-r--r-- | ATRI/plugins/pixiv.py | 246 |
1 files changed, 132 insertions, 114 deletions
diff --git a/ATRI/plugins/pixiv.py b/ATRI/plugins/pixiv.py index 4be60dc..5ca7b20 100644 --- a/ATRI/plugins/pixiv.py +++ b/ATRI/plugins/pixiv.py @@ -31,145 +31,163 @@ Link: {user_link} @on_command('pixiv_seach_img', aliases = ['p站搜图', 'P站搜图', '批站搜图'], only_to_me = False) async def _(session: CommandSession): - with open('ATRI/plugins/switch/switch.json', 'r') as f: + user = session.event.user_id + with open('ATRI/plugins/noobList/noobList.json', 'r') as f: data = json.load(f) - if data["pixiv_seach_img"] == 0: - user = session.event.user_id - pid = session.current_arg.strip() - - if not pid: - pid = session.get('message', prompt = '请告诉ATRI需要查询的Pid码') - - start =time.perf_counter() - await session.send('开始P站搜图\n如搜索时间过长或许为图片过大上传较慢') - - URL = URL_1 + pid - - dc = json.loads(response.request_api(URL)) - - if not dc: - session.finish('ATRI在网络上走散了...请重试...') - - img = f'https://pixiv.cat/{pid}.jpg' - - - end = time.perf_counter() - - await session.send( - IMG_SEACH_REPLY.format( - user = user, - pid = pid, - title = dc["response"][0]["title"], - width = dc["response"][0]["width"], - height = dc["response"][0]["height"], - tags = dc["response"][0]["tags"], - account = dc["response"][0]["user"]["account"], - name = dc["response"][0]["user"]["name"], - user_link = f'https://www.pixiv.net/users/' + f'{dc["response"][0]["user"]["id"]}', - img = img, - time = round(end - start, 3) - ) - ) - + if data[f"{user}"] == str(user): + pass else: - await session.send('该功能已被禁用...') + with open('ATRI/plugins/switch/switch.json', 'r') as f: + data = json.load(f) + + if data["pixiv_seach_img"] == 0: + pid = session.current_arg.strip() + + if not pid: + pid = session.get('message', prompt = '请告诉ATRI需要查询的Pid码') + + start =time.perf_counter() + await session.send('开始P站搜图\n如搜索时间过长或许为图片过大上传较慢') + + URL = URL_1 + pid + + dc = json.loads(response.request_api(URL)) + + if not dc: + session.finish('ATRI在网络上走散了...请重试...') + + img = f'https://pixiv.cat/{pid}.jpg' + + + end = time.perf_counter() + + await session.send( + IMG_SEACH_REPLY.format( + user = user, + pid = pid, + title = dc["response"][0]["title"], + width = dc["response"][0]["width"], + height = dc["response"][0]["height"], + tags = dc["response"][0]["tags"], + account = dc["response"][0]["user"]["account"], + name = dc["response"][0]["user"]["name"], + user_link = f'https://www.pixiv.net/users/' + f'{dc["response"][0]["user"]["id"]}', + img = img, + time = round(end - start, 3) + ) + ) + + else: + await session.send('该功能已被禁用...') @on_command('pixiv_seach_author', aliases = ['画师', '搜索画师', '画师搜索'], only_to_me = False) async def _(session: CommandSession): - with open('ATRI/plugins/switch/switch.json', 'r') as f: + user = session.event.user_id + with open('ATRI/plugins/noobList/noobList.json', 'r') as f: data = json.load(f) - - if data["pixiv_seach_author"] == 0: - user = session.event.user_id - author_id = session.current_arg.strip() - - if not author_id: - author_id = session.get('message', prompt = '请告诉ATRI需要查询的画师ID') - - start =time.perf_counter() - await session.send(f'开始获取画师{author_id}的前三项作品\n如获取时间过长或许为图片过大上传较慢') - - URL = URL_2 + author_id - dc = json.loads(response.request_api(URL)) - - if not dc: - session.finish('ATRI在网络上走散了...请重试...') - - d ={} - - for i in range(0,3): - pid = dc["response"][i]["id"] - pidURL = f'https://pixiv.cat/{pid}.jpg' - d[i] = [f'{pid}',f'{pidURL}'] - - msg0 = (f'[CQ:at,qq={user}]\n画师id:{author_id}，接下来展示前三作品') - - result = sorted( - d.items(), - key = lambda x:x[1], - reverse = True - ) + if data[f"{user}"] == str(user): + pass + else: + with open('ATRI/plugins/switch/switch.json', 'r') as f: + data = json.load(f) + + if data["pixiv_seach_author"] == 0: + author_id = session.current_arg.strip() + + if not author_id: + author_id = session.get('message', prompt = '请告诉ATRI需要查询的画师ID') + + start =time.perf_counter() + await session.send(f'开始获取画师{author_id}的前三项作品\n如获取时间过长或许为图片过大上传较慢') + + URL = URL_2 + author_id + + dc = json.loads(response.request_api(URL)) + + if not dc: + session.finish('ATRI在网络上走散了...请重试...') + + d ={} + + for i in range(0,3): + pid = dc["response"][i]["id"] + pidURL = f'https://pixiv.cat/{pid}.jpg' + d[i] = [f'{pid}',f'{pidURL}'] + + msg0 = (f'[CQ:at,qq={user}]\n画师id:{author_id}，接下来展示前三作品') + + result = sorted( + d.items(), + key = lambda x:x[1], + reverse = True + ) - t = 0 + t = 0 - for i in result: - t += 1 - msg = (f'\n---------------\n({t})\nPid: {i[1][0]}\n{i[1][1]}') - msg0 += msg - end = time.perf_counter() + for i in result: + t += 1 + msg = (f'\n---------------\n({t})\nPid: {i[1][0]}\n{i[1][1]}') + msg0 += msg + end = time.perf_counter() - msg1 = (f'\n---------------\n完成时间:{round(end - start, 3)}s') - msg0 += msg1 + msg1 = (f'\n---------------\n完成时间:{round(end - start, 3)}s') + msg0 += msg1 + + await session.send(msg0) - await session.send(msg0) - - else: - await session.send('该功能已被禁用...') + else: + await session.send('该功能已被禁用...') @on_command('pixiv_daily_rank', aliases = ['P站排行榜', '批站排行榜', 'p站排行榜'], only_to_me = False) async def _(session: CommandSession): - with open('ATRI/plugins/switch/switch.json', 'r') as f: + user = session.event.user_id + with open('ATRI/plugins/noobList/noobList.json', 'r') as f: data = json.load(f) - - if data["pixiv_daily_rank"] == 0: - user = session.event.user_id - await session.send('ATRI正在获取P站每日排行榜前五作品...') + if data[f"{user}"] == str(user): + pass + else: + with open('ATRI/plugins/switch/switch.json', 'r') as f: + data = json.load(f) + + if data["pixiv_daily_rank"] == 0: - start =time.perf_counter() - dc = json.loads(response.request_api(URL_3)) + await session.send('ATRI正在获取P站每日排行榜前五作品...') - d = {} + start =time.perf_counter() + dc = json.loads(response.request_api(URL_3)) - for i in range(0,5): - pid = dc["response"][0]["works"][i]["work"]["id"] - pidURL = f'https://pixiv.cat/{pid}.jpg' - d[i] = [f'{pid}',f'{pidURL}'] + d = {} - msg0 = (f'[CQ:at,qq={user}]') + for i in range(0,5): + pid = dc["response"][0]["works"][i]["work"]["id"] + pidURL = f'https://pixiv.cat/{pid}.jpg' + d[i] = [f'{pid}',f'{pidURL}'] - result = sorted( - d.items(), - key = lambda x:x[1], - reverse = True - ) + msg0 = (f'[CQ:at,qq={user}]') - t = 0 + result = sorted( + d.items(), + key = lambda x:x[1], + reverse = True + ) - for i in result: - t += 1 - msg = (f'\n---------------\n({t})\nPid: {i[1][0]}\n{i[1][1]}') - msg0 += msg - end = time.perf_counter() + t = 0 - msg1 = (f'\n---------------\n完成时间:{round(end - start, 3)}s') - msg0 += msg1 + for i in result: + t += 1 + msg = (f'\n---------------\n({t})\nPid: {i[1][0]}\n{i[1][1]}') + msg0 += msg + end = time.perf_counter() - await session.send(msg0) - - else: - await session.send('该功能已被禁用...')
\ No newline at end of file + msg1 = (f'\n---------------\n完成时间:{round(end - start, 3)}s') + msg0 += msg1 + + await session.send(msg0) + + else: + await session.send('该功能已被禁用...')
\ No newline at end of file |