Skip to content

Commit 900541b

Browse files
committed
feat(config): Add configurable workers and host settings
- Add "workers" configuration option to DEFAULT_CONFIG with default value of 1 - Add "host" configuration option to DEFAULT_CONFIG with default value of "::" - Update uvicorn.run() to use settings.host and settings.workers instead of hardcoded values - Reorganize imports in main.py for better readability and consistency - Add blank line after time import in core/utils.py for PEP 8 compliance - Allow dynamic configuration of server workers and binding host address
1 parent 8b813a2 commit 900541b

File tree

3 files changed

+8
-7
lines changed

3 files changed

+8
-7
lines changed

core/settings.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,8 @@
6666
"themesSelect": "themes/2024",
6767
"errorMinute": 1,
6868
"errorCount": 1,
69+
"workers": 1,
70+
"host": "::",
6971
"port": 12345,
7072
"showAdminAddr": 0,
7173
"robotsText": "User-agent: *\nDisallow: /",

core/utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
import re
1010
import string
1111
import time
12+
1213
from core.settings import settings
1314

1415

main.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,26 +4,24 @@
44
# @Software: PyCharm
55
import asyncio
66
import time
7+
from contextlib import asynccontextmanager
78

89
from fastapi import FastAPI
9-
1010
from fastapi.middleware.cors import CORSMiddleware
1111
from fastapi.responses import HTMLResponse
1212
from fastapi.staticfiles import StaticFiles
13+
from tortoise import Tortoise
1314
from tortoise.contrib.fastapi import register_tortoise
1415

16+
from apps.admin.views import admin_api
1517
from apps.base.models import KeyValue
1618
from apps.base.utils import ip_limit
1719
from apps.base.views import share_api, chunk_api
18-
from apps.admin.views import admin_api
1920
from core.database import init_db
21+
from core.logger import logger
2022
from core.response import APIResponse
2123
from core.settings import data_root, settings, BASE_DIR, DEFAULT_CONFIG
2224
from core.tasks import delete_expire_files, clean_incomplete_uploads
23-
from core.logger import logger
24-
25-
from contextlib import asynccontextmanager
26-
from tortoise import Tortoise
2725

2826

2927
@asynccontextmanager
@@ -150,5 +148,5 @@ async def get_config():
150148
import uvicorn
151149

152150
uvicorn.run(
153-
app="main:app", host="0.0.0.0", port=settings.port, reload=False, workers=1
151+
app="main:app", host=settings.host, port=settings.port, reload=False, workers=settings.workers
154152
)

0 commit comments

Comments
 (0)