CalvinBackup/facefusion
1
0 Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-10-13 17:17:30 +02:00
Code Issues Packages Projects Releases Wiki Activity

state api updates

Browse Source
This commit is contained in:
harisreedhar authored and henryruhs committed 2026-05-11 16:34:28 +02:00
1 parent a01bed0bda
commit 98384a55df
2 files changed
+15 -16
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

facefusion/apis/endpoints/state.py
+6 -6
View File
@@ -30,8 +30,8 @@ async def set_state(request : Request) -> JSONResponse:
if key in api_args: if key in api_args:
state_manager.set_item(key, value) state_manager.set_item(key, value)
api_args = args_store.filter_api_args(state_manager.get_state()) #type:ignore[arg-type,assignment] __api_args__ = args_store.filter_api_args(state_manager.get_state()) #type:ignore[arg-type]
return JSONResponse(state_manager.collect_state(api_args), status_code = HTTP_200_OK) #type:ignore[arg-type] return JSONResponse(state_manager.collect_state(__api_args__), status_code = HTTP_200_OK) #type:ignore[arg-type]
async def select_source(request : Request) -> JSONResponse: async def select_source(request : Request) -> JSONResponse:
@@ -49,8 +49,8 @@ async def select_source(request : Request) -> JSONResponse:
state_manager.set_item('source_paths', source_paths) state_manager.set_item('source_paths', source_paths)
api_args = args_store.filter_api_args(state_manager.get_state()) #type:ignore[arg-type] __api_args__ = args_store.filter_api_args(state_manager.get_state()) #type:ignore[arg-type]
return JSONResponse(state_manager.collect_state(api_args), status_code = HTTP_200_OK) return JSONResponse(state_manager.collect_state(__api_args__), status_code = HTTP_200_OK)
return JSONResponse( return JSONResponse(
{ {
@@ -68,8 +68,8 @@ async def select_target(request : Request) -> JSONResponse:
if asset: if asset:
state_manager.set_item('target_path', asset.get('path')) state_manager.set_item('target_path', asset.get('path'))
api_args = args_store.filter_api_args(state_manager.get_state()) #type:ignore[arg-type] __api_args__ = args_store.filter_api_args(state_manager.get_state()) #type:ignore[arg-type]
return JSONResponse(state_manager.collect_state(api_args), status_code = HTTP_200_OK) return JSONResponse(state_manager.collect_state(__api_args__), status_code = HTTP_200_OK)
return JSONResponse( return JSONResponse(
{ {
tests/test_api_state.py
+9 -10
View File
@@ -85,12 +85,11 @@ def test_set_state(test_client : TestClient) -> None:
def test_select_source_assets(test_client : TestClient) -> None: def test_select_source_assets(test_client : TestClient) -> None:
asset_id_1 = asset_store.register_asset('/path/to/source1.jpg') asset_ids = [ asset_store.register_asset('/path/to/source1.jpg'), asset_store.register_asset('/path/to/source2.jpg') ]
asset_id_2 = asset_store.register_asset('/path/to/source2.jpg')
select_response = test_client.put('/state?action=select&type=source', json = select_response = test_client.put('/state?action=select&type=source', json =
{ {
'asset_ids': [ asset_id_1, asset_id_2 ] 'asset_ids': asset_ids
}) })
assert select_response.status_code == 401 assert select_response.status_code == 401
@@ -103,7 +102,7 @@ def test_select_source_assets(test_client : TestClient) -> None:
select_response = test_client.put('/state?action=select&type=source', json = select_response = test_client.put('/state?action=select&type=source', json =
{ {
'asset_ids': 'invalid_string_not_list' 'asset_ids': 'invalid'
}, headers = }, headers =
{ {
'Authorization': 'Bearer ' + create_session_body.get('access_token') 'Authorization': 'Bearer ' + create_session_body.get('access_token')
@@ -113,7 +112,7 @@ def test_select_source_assets(test_client : TestClient) -> None:
select_response = test_client.put('/state?action=select&type=source', json = select_response = test_client.put('/state?action=select&type=source', json =
{ {
'asset_ids': [ asset_id_1, asset_id_2 ] 'asset_ids': asset_ids
}, headers = }, headers =
{ {
'Authorization': 'Bearer ' + create_session_body.get('access_token') 'Authorization': 'Bearer ' + create_session_body.get('access_token')
@@ -125,11 +124,11 @@ def test_select_source_assets(test_client : TestClient) -> None:
def test_select_target_assets(test_client : TestClient) -> None: def test_select_target_assets(test_client : TestClient) -> None:
asset_id_1 = asset_store.register_asset('/path/to/target1.jpg') asset_id = asset_store.register_asset('/path/to/target.jpg')
select_response = test_client.put('/state?action=select&type=target', json = select_response = test_client.put('/state?action=select&type=target', json =
{ {
'asset_id': asset_id_1 'asset_id': asset_id
}) })
assert select_response.status_code == 401 assert select_response.status_code == 401
@@ -142,7 +141,7 @@ def test_select_target_assets(test_client : TestClient) -> None:
select_response = test_client.put('/state?action=select&type=target', json = select_response = test_client.put('/state?action=select&type=target', json =
{ {
'asset_id': 'invalid_asset_id' 'asset_id': 'invalid'
}, headers = }, headers =
{ {
'Authorization': 'Bearer ' + create_session_body.get('access_token') 'Authorization': 'Bearer ' + create_session_body.get('access_token')
@@ -152,12 +151,12 @@ def test_select_target_assets(test_client : TestClient) -> None:
select_response = test_client.put('/state?action=select&type=target', json = select_response = test_client.put('/state?action=select&type=target', json =
{ {
'asset_id': asset_id_1 'asset_id': asset_id
}, headers = }, headers =
{ {
'Authorization': 'Bearer ' + create_session_body.get('access_token') 'Authorization': 'Bearer ' + create_session_body.get('access_token')
}) })
select_body = select_response.json() select_body = select_response.json()
assert select_body.get('target_path') == '/path/to/target1.jpg' assert select_body.get('target_path') == '/path/to/target.jpg'
assert select_response.status_code == 200 assert select_response.status_code == 200