Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2026-09-29 00:32:06 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Update data_source.py

Browse Source
This commit is contained in:
HibiKier
2021-07-19 15:23:31 +08:00
committed by GitHub
parent e26da37c58
commit ae3a32589a
1 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
plugins/send_setu/data_source.py
+9 -4
View File
@@ -117,10 +117,15 @@ async def search_online_setu(url: str):
return '\n图片被小怪兽恰掉啦..!QAQ', -1
def get_setu(index: str, setu_data: dict, tag: str = None):
if not os.path.exists(IMAGE_PATH + path):
os.mkdir(IMAGE_PATH + path)
length = len(os.listdir(IMAGE_PATH + path))
def get_setu(index: str, setu_data: dict, num: int = 1, tag: str = None, r18: bool = False):
if r18:
if not os.path.exists(IMAGE_PATH + '_r18'):
os.mkdir(IMAGE_PATH + '_r18')
length = len(os.listdir(IMAGE_PATH + '_r18'))
else:
if not os.path.exists(IMAGE_PATH + path):
os.mkdir(IMAGE_PATH + path)
length = len(os.listdir(IMAGE_PATH + path))
if length == 0:
return None, None
if not index: