CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-07-25 03:20:57 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add simple path isolation (#992)

Browse Source
This commit is contained in:
Henry Ruhs
2025-11-30 19:05:10 +01:00
committed by henryruhs
parent 0625e84e78
commit 38a7da5a26
10 changed files with 59 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_ffmpeg.py
+14 -14
View File
@@ -82,12 +82,12 @@ def test_extract_frames() -> None:
]
for target_path, output_path, trim_frame_start, trim_frame_end, frame_total in test_set:
create_temp_directory(state_manager.get_item('temp_path'), output_path)
create_temp_directory(state_manager.get_temp_path(), output_path)
assert extract_frames(target_path, output_path, (452, 240), 30.0, trim_frame_start, trim_frame_end) is True
assert len(resolve_temp_frame_paths(state_manager.get_item('temp_path'), output_path, state_manager.get_item('temp_frame_format'))) == frame_total
assert len(resolve_temp_frame_paths(state_manager.get_temp_path(), output_path, state_manager.get_item('temp_frame_format'))) == frame_total
clear_temp_directory(state_manager.get_item('temp_path'), output_path)
clear_temp_directory(state_manager.get_temp_path(), output_path)
def test_spawn_frames() -> None:
@@ -101,12 +101,12 @@ def test_spawn_frames() -> None:
]
for target_path, output_path, trim_frame_start, trim_frame_end, temp_video_fps, frame_total in test_set:
create_temp_directory(state_manager.get_item('temp_path'), output_path)
create_temp_directory(state_manager.get_temp_path(), output_path)
assert spawn_frames(target_path, output_path, (452, 240), temp_video_fps, trim_frame_start, trim_frame_end) is True
assert len(resolve_temp_frame_paths(state_manager.get_item('temp_path'), output_path, state_manager.get_item('temp_frame_format'))) == frame_total
assert len(resolve_temp_frame_paths(state_manager.get_temp_path(), output_path, state_manager.get_item('temp_frame_format'))) == frame_total
clear_temp_directory(state_manager.get_item('temp_path'), output_path)
clear_temp_directory(state_manager.get_temp_path(), output_path)
def test_merge_video() -> None:
@@ -127,12 +127,12 @@ def test_merge_video() -> None:
state_manager.init_item('output_path', target_path)
state_manager.init_item('output_video_fps', 25.0)
state_manager.init_item('output_video_encoder', output_video_encoder)
create_temp_directory(state_manager.get_item('temp_path'), output_path)
create_temp_directory(state_manager.get_temp_path(), output_path)
extract_frames(target_path, output_path, (452, 240), 25.0, 0, 1)
assert merge_video(target_path, output_path, 25.0, (452, 240), 0, 1) is True
clear_temp_directory(state_manager.get_item('temp_path'), output_path)
clear_temp_directory(state_manager.get_temp_path(), output_path)
state_manager.init_item('output_video_encoder', 'libx264')
@@ -169,15 +169,15 @@ def test_restore_audio() -> None:
output_audio_encoders = get_available_encoder_set().get('audio')
for target_path, output_path in test_set:
create_temp_directory(state_manager.get_item('temp_path'), output_path)
create_temp_directory(state_manager.get_temp_path(), output_path)
for output_audio_encoder in output_audio_encoders:
state_manager.init_item('output_audio_encoder', output_audio_encoder)
copy_file(target_path, get_temp_file_path(state_manager.get_item('temp_path'), output_path))
copy_file(target_path, get_temp_file_path(state_manager.get_temp_path(), output_path))
assert restore_audio(target_path, output_path, 0, 270) is True
clear_temp_directory(state_manager.get_item('temp_path'), output_path)
clear_temp_directory(state_manager.get_temp_path(), output_path)
state_manager.init_item('output_audio_encoder', 'aac')
@@ -196,15 +196,15 @@ def test_replace_audio() -> None:
output_audio_encoders = get_available_encoder_set().get('audio')
for target_path, output_path in test_set:
create_temp_directory(state_manager.get_item('temp_path'), output_path)
create_temp_directory(state_manager.get_temp_path(), output_path)
for output_audio_encoder in output_audio_encoders:
state_manager.init_item('output_audio_encoder', output_audio_encoder)
copy_file(target_path, get_temp_file_path(state_manager.get_item('temp_path'), output_path))
copy_file(target_path, get_temp_file_path(state_manager.get_temp_path(), output_path))
assert replace_audio(get_test_example_file('source.mp3'), output_path) is True
assert replace_audio(get_test_example_file('source.wav'), output_path) is True
clear_temp_directory(state_manager.get_item('temp_path'), output_path)
clear_temp_directory(state_manager.get_temp_path(), output_path)
state_manager.init_item('output_audio_encoder', 'aac')
tests/test_temp_helper.py
+3 -3
View File
@@ -21,12 +21,12 @@ def before_all() -> None:
def test_get_temp_file_path() -> None:
state_manager.init_item('output_path', 'temp.mp4')
assert get_temp_file_path(state_manager.get_item('temp_path'), get_test_example_file('target-240p.mp4')) == os.path.join(state_manager.get_item('temp_path'), 'facefusion', 'target-240p', 'temp.mp4')
assert get_temp_file_path(state_manager.get_temp_path(), get_test_example_file('target-240p.mp4')) == os.path.join(state_manager.get_temp_path(), 'facefusion', 'target-240p', 'temp.mp4')
def test_get_temp_directory_path() -> None:
assert get_temp_directory_path(state_manager.get_item('temp_path'), get_test_example_file('target-240p.mp4')) == os.path.join(state_manager.get_item('temp_path'), 'facefusion', 'target-240p')
assert get_temp_directory_path(state_manager.get_temp_path(), get_test_example_file('target-240p.mp4')) == os.path.join(state_manager.get_temp_path(), 'facefusion', 'target-240p')
def test_get_temp_frames_pattern() -> None:
assert get_temp_frames_pattern(state_manager.get_item('temp_path'), get_test_example_file('target-240p.mp4'), state_manager.get_item('temp_frame_format'), '%04d') == os.path.join(state_manager.get_item('temp_path'), 'facefusion', 'target-240p', '%04d.png')
assert get_temp_frames_pattern(state_manager.get_temp_path(), get_test_example_file('target-240p.mp4'), state_manager.get_item('temp_frame_format'), '%04d') == os.path.join(state_manager.get_temp_path(), 'facefusion', 'target-240p', '%04d.png')