diff --git a/handlers/pixiv/file.py b/handlers/pixiv/file.py index 7b1ce78..388fd64 100644 --- a/handlers/pixiv/file.py +++ b/handlers/pixiv/file.py @@ -10,4 +10,12 @@ def remove_file(path): try: os.remove(path) except: - pass \ No newline at end of file + pass + +def clear_dir(): + temp_path = os.path.join(os.path.dirname(__file__), 'temp') + try: + os.rmdir(temp_path) + print('temp dir is cleared') + except: + print('temp clear failed') \ No newline at end of file diff --git a/handlers/pixiv/pixiv_handler.py b/handlers/pixiv/pixiv_handler.py index ef499b4..2071482 100644 --- a/handlers/pixiv/pixiv_handler.py +++ b/handlers/pixiv/pixiv_handler.py @@ -5,7 +5,7 @@ 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 +from handlers.pixiv.file import save_file, remove_file, clear_dir webdav = Webdav({ @@ -58,7 +58,7 @@ def pixiv_result_handler(entries): for entry in entries: if entry['id'] == item['id']: failed_entries.append(entry) - + clear_dir() return (success_entries, failed_entries) \ No newline at end of file