From 7148940a2fd0e259d10b75e52f1a138babb5f4e0 Mon Sep 17 00:00:00 2001
From: HibiKier <45528451+HibiKier@users.noreply.github.com>
Date: Mon, 9 Aug 2021 07:50:04 +0800
Subject: [PATCH] Update data_source.py

---
 plugins/send_setu/data_source.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/plugins/send_setu/data_source.py b/plugins/send_setu/data_source.py
index 08252aca..beaf4e54 100644
--- a/plugins/send_setu/data_source.py
+++ b/plugins/send_setu/data_source.py
@@ -90,6 +90,8 @@ async def search_online_setu(
                         index = random.randint(1, 100000) if id_ is None else id_
                         path_ = "temp" if not path_ else path_
                         file = f"{index}_temp_setu.jpg" if not path_ else f"{index}.jpg"
+                        if not os.path.exists(f'{IMAGE_PATH}/path_'):
+                            os.mkdir(f'{IMAGE_PATH}/path_')
                         async with aiofiles.open(
                             f"{IMAGE_PATH}/{path_}/{file}", "wb"
                         ) as f: