diff --git a/core/settings.py b/core/settings.py new file mode 100644 index 0000000..a8acc22 --- /dev/null +++ b/core/settings.py @@ -0,0 +1,9 @@ +# @Time : 2023/8/15 09:51 +# @Author : Lan +# @File : settings.py +# @Software: PyCharm +from pathlib import Path + +data_root = Path('./data') +if not data_root.exists(): + data_root.mkdir(parents=True, exist_ok=True) diff --git a/core/storage.py b/core/storage.py index 426197c..ac6c65c 100644 --- a/core/storage.py +++ b/core/storage.py @@ -8,15 +8,14 @@ import time import aioboto3 from fastapi import UploadFile -from pathlib import Path - +from core.settings import data_root from apps.base.models import FileCodes class SystemFileStorage: def __init__(self): self.chunk_size = 256 * 1024 - self.root_path = Path('./data') + self.root_path = data_root self.token = '123456' def _save(self, file, save_path): diff --git a/filecodebox.db-shm b/filecodebox.db-shm deleted file mode 100644 index fe9ac28..0000000 Binary files a/filecodebox.db-shm and /dev/null differ diff --git a/filecodebox.db-wal b/filecodebox.db-wal deleted file mode 100644 index e69de29..0000000 diff --git a/main.py b/main.py index bbd078c..a81dae1 100644 --- a/main.py +++ b/main.py @@ -9,9 +9,9 @@ from starlette.staticfiles import StaticFiles from tortoise.contrib.fastapi import register_tortoise from apps.base.views import share_api from apps.admin.views import admin_api +from core.settings import data_root app = FastAPI() - app.add_middleware( CORSMiddleware, allow_origins=["*"], @@ -19,7 +19,6 @@ app.add_middleware( allow_methods=["*"], allow_headers=["*"], ) - app.mount('/assets', StaticFiles(directory='./fcb-fronted/dist/assets'), name="assets") register_tortoise( @@ -28,7 +27,7 @@ register_tortoise( add_exception_handlers=True, config={ 'connections': { - 'default': 'sqlite://filecodebox.db' + 'default': f'sqlite://{data_root}/filecodebox.db' }, 'apps': { 'models': {