From 88a4105e7cbfbcd7952b3d47be3279f021f7ad6f Mon Sep 17 00:00:00 2001 From: Kyomotoi Date: Thu, 9 Jun 2022 22:32:12 +0800 Subject: =?UTF-8?q?=F0=9F=92=A9=20=E4=BC=98=E5=8C=96=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/plugins/manage/data_source.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'ATRI/plugins/manage') diff --git a/ATRI/plugins/manage/data_source.py b/ATRI/plugins/manage/data_source.py index 395a165..1d870e6 100644 --- a/ATRI/plugins/manage/data_source.py +++ b/ATRI/plugins/manage/data_source.py @@ -1,4 +1,3 @@ -import os import json from pathlib import Path from datetime import datetime @@ -9,8 +8,9 @@ from ATRI.exceptions import ReadFileError, load_error MANAGE_DIR = Path(".") / "data" / "database" / "manege" ESSENTIAL_DIR = Path(".") / "data" / "database" / "essential" -os.makedirs(MANAGE_DIR, exist_ok=True) -os.makedirs(ESSENTIAL_DIR, exist_ok=True) +MANAGE_DIR.mkdir(parents=True, exist_ok=True) +ESSENTIAL_DIR.mkdir(parents=True, exist_ok=True) + TRACK_BACK_FORMAT = """Track ID：{track_id} Prompt: {prompt} -- cgit v1.2.3