64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
import re
|
|
import time
|
|
import os
|
|
|
|
from handlers.pixiv.comb import extract_pixiv_info, get_prefix_name_on_entry
|
|
from handlers.pixiv.Webdav import Webdav
|
|
from handlers.pixiv.request import downloadPic
|
|
from handlers.pixiv.file import save_file, remove_file, clear_dir
|
|
|
|
|
|
webdav = Webdav({
|
|
'webdav_hostname': os.getenv('webdav_hostname'),
|
|
'webdav_login': os.getenv('webdav_login'),
|
|
'webdav_password': os.getenv('webdav_password'),
|
|
'disable_check': True,
|
|
'default_upload_path': os.getenv('default_upload_path')
|
|
})
|
|
|
|
# 处理结果
|
|
def pixiv_result_handler(entries):
|
|
list = extract_pixiv_info(entries)
|
|
|
|
success_entries = []
|
|
failed_entries = []
|
|
|
|
for item in list:
|
|
prefix_name = get_prefix_name_on_entry(item)
|
|
|
|
result_flag = True
|
|
|
|
need_sync_files = []
|
|
|
|
for url in item['link']:
|
|
file_name_pattern = re.compile(r'\/(\w*\.(?:jpg|png))$')
|
|
file_name = ','.join(re.findall(file_name_pattern, url))
|
|
|
|
if file_name:
|
|
full_name = f'{prefix_name} {file_name}'
|
|
# 替换不符合文件名规范的字符
|
|
full_name = re.sub(r'[\/\\\:\*\?\"\<\>\|]', "_", full_name)
|
|
(status, data) = downloadPic(url)
|
|
if status:
|
|
save_file(full_name, data)
|
|
need_sync_files.append({ 'id': item['id'], 'file_name': full_name })
|
|
else:
|
|
result_flag = False
|
|
time.sleep(10)
|
|
|
|
if result_flag:
|
|
(success_ids, failed_ids) = webdav.batch_sync(need_sync_files)
|
|
|
|
for entry in entries:
|
|
if entry['id'] in success_ids:
|
|
success_entries.append(entry)
|
|
elif entry['id'] in failed_ids:
|
|
failed_entries.append(entry)
|
|
else:
|
|
for entry in entries:
|
|
if entry['id'] == item['id']:
|
|
failed_entries.append(entry)
|
|
clear_dir()
|
|
return (success_entries, failed_entries)
|
|
|
|
|