Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-15 14:22:55 +08:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

Update data_source.py

Browse Source
This commit is contained in:
HibiKier 2021-05-20 21:04:57 +08:00 committed by GitHub
parent 9fbc3b9d02
commit bcc6df9d4d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
plugins/send_setu/data_source.py
View File

@ -63,9 +63,8 @@ async def get_setu_urls(keyword: str, num: int = 1, r18: int = 0):
text_list.append(f'title：{title}\nauthor：{author}\nPID：{pid}')
img_url = str(img_url).replace('img-master', 'img-original').replace('_master1200', '')
txt_data += img_url + ','
if DOWNLOAD_SETU:
with open(TXT_PATH + file_name, 'w') as f:
f.write(txt_data)
with open(TXT_PATH + file_name, 'w') as f:
f.write(txt_data)
return urls, text_list, 200
else:
return "没找到符合条件的色图...", '', 401
@ -100,6 +99,8 @@ async def search_online_setu(url: str):
def get_setu(index: str):
length = len(os.listdir(IMAGE_PATH + path))
if length == 0:
return None, None
if not index:
index = random.randint(0, length - 1)
if is_number(index):