From a9fd99706266f2b7cb8d58e6d3b14c534e52a9b3 Mon Sep 17 00:00:00 2001 From: lan-air Date: Mon, 12 Dec 2022 15:15:17 +0800 Subject: [PATCH] add setting port --- main.py | 2 +- settings.py | 2 ++ storage.py | 4 ++-- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/main.py b/main.py index bd9471c..6be2abe 100644 --- a/main.py +++ b/main.py @@ -207,4 +207,4 @@ async def share(text: str = Form(default=None), style: str = Form(default='2'), if __name__ == '__main__': import uvicorn - uvicorn.run('main:app', host='0.0.0.0', port=12345) + uvicorn.run('main:app', host='0.0.0.0', port=settings.PORT, debug=settings.DEBUG) diff --git a/settings.py b/settings.py index 9959f14..8405216 100644 --- a/settings.py +++ b/settings.py @@ -4,6 +4,8 @@ config = Config(".env") DEBUG = config('DEBUG', cast=bool, default=False) +PORT = config('PORT', cast=int, default=12345) + DATABASE_URL = config('DATABASE_URL', cast=str, default="sqlite+aiosqlite:///database.db") DATA_ROOT = config('DATA_ROOT', cast=str, default="./static") diff --git a/storage.py b/storage.py index 76c9d69..3b43d24 100644 --- a/storage.py +++ b/storage.py @@ -33,7 +33,7 @@ class FileSystemStorage: async def delete_file(self, file): filepath = self.DATA_ROOT / file['text'].lstrip(self.STATIC_URL + '/') await asyncio.to_thread(os.remove, filepath) - + async def delete_files(self, files): for file in files: if file['type'] != 'text': @@ -42,4 +42,4 @@ class FileSystemStorage: STORAGE_ENGINE = { "filesystem": FileSystemStorage -} \ No newline at end of file +}