feat: update user preferences and system settings management
This commit is contained in:
@@ -15,9 +15,12 @@ from db.crud import (
|
||||
list_users,
|
||||
create_user_by_admin,
|
||||
update_user,
|
||||
delete_user
|
||||
delete_user,
|
||||
get_system_setting,
|
||||
update_system_setting,
|
||||
is_local_model_enabled
|
||||
)
|
||||
from schemas.user import User, UserCreateByAdmin, UserUpdate, UserListResponse
|
||||
from schemas.user import User, UserCreateByAdmin, UserUpdate, UserListResponse, SystemSettingsUpdate, SystemSettingsResponse
|
||||
|
||||
router = APIRouter(prefix="/users", tags=["users"])
|
||||
limiter = Limiter(key_func=get_remote_address)
|
||||
@@ -167,3 +170,42 @@ async def delete_user_by_id(
|
||||
status_code=status.HTTP_404_NOT_FOUND,
|
||||
detail="User not found"
|
||||
)
|
||||
|
||||
@router.get("/system/settings", response_model=SystemSettingsResponse)
|
||||
async def get_system_settings(
|
||||
current_user: Annotated[User, Depends(require_superuser)],
|
||||
db: Session = Depends(get_db)
|
||||
):
|
||||
local_enabled = is_local_model_enabled(db)
|
||||
return {"local_model_enabled": local_enabled}
|
||||
|
||||
@router.put("/system/settings")
|
||||
async def update_system_settings(
|
||||
settings: SystemSettingsUpdate,
|
||||
current_user: Annotated[User, Depends(require_superuser)],
|
||||
db: Session = Depends(get_db)
|
||||
):
|
||||
from db.models import User
|
||||
from datetime import datetime
|
||||
|
||||
update_system_setting(db, "local_model_enabled", {"enabled": settings.local_model_enabled})
|
||||
|
||||
if not settings.local_model_enabled:
|
||||
users = db.query(User).filter(User.is_superuser == False).all()
|
||||
migrated_count = 0
|
||||
|
||||
for user in users:
|
||||
prefs = user.user_preferences or {}
|
||||
if prefs.get("default_backend") == "local":
|
||||
prefs["default_backend"] = "aliyun"
|
||||
user.user_preferences = prefs
|
||||
user.updated_at = datetime.utcnow()
|
||||
migrated_count += 1
|
||||
|
||||
db.commit()
|
||||
return {
|
||||
"message": "System settings updated",
|
||||
"users_migrated": migrated_count
|
||||
}
|
||||
|
||||
return {"message": "System settings updated", "users_migrated": 0}
|
||||
|
||||
Reference in New Issue
Block a user