mirror of
https://github.com/facefusion/facefusion.git
synced 2026-08-18 23:07:14 +02:00
Ignore set state for non api scope (#1073)
* ignore set state for non api scope * replace 404 with 422 * remove two loops * use HTTP_422_UNPROCESSABLE_CONTENT
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
from starlette.requests import Request
|
||||
from starlette.responses import JSONResponse
|
||||
from starlette.status import HTTP_200_OK, HTTP_404_NOT_FOUND
|
||||
from starlette.status import HTTP_200_OK, HTTP_400_BAD_REQUEST, HTTP_404_NOT_FOUND, HTTP_422_UNPROCESSABLE_CONTENT
|
||||
|
||||
from facefusion import args_helper, capability_store, session_manager, state_manager, translator
|
||||
from facefusion.apis import asset_store
|
||||
@@ -13,6 +13,8 @@ async def get_state(request : Request) -> JSONResponse:
|
||||
|
||||
|
||||
async def set_state(request : Request) -> JSONResponse:
|
||||
__api_args__ = {}
|
||||
|
||||
action = request.query_params.get('action')
|
||||
asset_type = request.query_params.get('type')
|
||||
|
||||
@@ -26,11 +28,19 @@ async def set_state(request : Request) -> JSONResponse:
|
||||
api_args = capability_store.get_api_arguments()
|
||||
|
||||
for key, value in body.items():
|
||||
if key in api_args:
|
||||
state_manager.set_item(key, value)
|
||||
if key not in api_args:
|
||||
return JSONResponse(
|
||||
{
|
||||
'message': translator.get('invalid_state_key', 'facefusion.apis')
|
||||
}, status_code = HTTP_400_BAD_REQUEST)
|
||||
__api_args__[key] = value
|
||||
state_manager.set_item(key, value)
|
||||
|
||||
__api_args__ = args_helper.extract_api_args(state_manager.get_state())
|
||||
return JSONResponse(state_manager.collect_state(__api_args__), status_code = HTTP_200_OK)
|
||||
if __api_args__:
|
||||
__api_args__ = args_helper.extract_api_args(state_manager.get_state())
|
||||
return JSONResponse(state_manager.collect_state(__api_args__), status_code = HTTP_200_OK)
|
||||
|
||||
return JSONResponse({}, status_code = HTTP_422_UNPROCESSABLE_CONTENT)
|
||||
|
||||
|
||||
async def select_source(request : Request) -> JSONResponse:
|
||||
|
||||
@@ -9,6 +9,7 @@ LOCALES : Locales =\
|
||||
'invalid_access_token': 'invalid access token',
|
||||
'invalid_refresh_token': 'invalid refresh token',
|
||||
'source_asset_not_found': 'source asset not found',
|
||||
'target_asset_not_found': 'target asset not found'
|
||||
'target_asset_not_found': 'target asset not found',
|
||||
'invalid_state_key': 'invalid state key'
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user