lywsvip/StreamCap
1
0
Fork 0
mirror of https://github.com/ihmily/StreamCap.git synced 2026-05-07 22:15:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add record segment for flv

Browse Source
This commit is contained in:
ihmily
2025-07-15 17:41:14 +08:00
parent 794e0d3784
commit 39a3ccfb8e
2 changed files with 24 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
app/core/ffmpeg_builders/video/flv.py
View File

@@ -4,13 +4,26 @@ from ..base import FFmpegCommandBuilder
class FLVCommandBuilder(FFmpegCommandBuilder):
def build_command(self) -> list[str]:
command = self._get_basic_ffmpeg_command()
additional_commands = [
"-map", "0",
"-c:v", "copy",
"-c:a", "copy",
"-bsf:a", "aac_adtstoasc",
"-f", "flv",
self.full_path,
]
if self.segment_record:
additional_commands = [
"-map", "0",
"-c:v", "copy",
"-c:a", "copy",
"-bsf:a", "aac_adtstoasc",
"-f", "segment",
"-segment_time", str(self.segment_time),
"-segment_format", "flv",
"-reset_timestamps", "1",
self.full_path
]
else:
additional_commands = [
"-map", "0",
"-c:v", "copy",
"-c:a", "copy",
"-bsf:a", "aac_adtstoasc",
"-f", "flv",
self.full_path
]
command.extend(additional_commands)
return command

6
app/core/stream_manager.py
View File

@@ -127,9 +127,9 @@ class LiveStreamRecorder:
self.app.page.run_task(self.app.record_manager.persist_recordings)
return output_dir
def _get_save_path(self, filename: str) -> str:
def _get_save_path(self, filename: str, use_direct_download: bool = False) -> str:
suffix = self.save_format
suffix = "_%03d." + suffix if self.segment_record and self.save_format != "flv" else "." + suffix
suffix = "_%03d." + suffix if self.segment_record and not use_direct_download else "." + suffix
save_file_path = os.path.join(self.output_dir, filename + suffix).replace(" ", "_")
return save_file_path.replace("\\", "/")
@@ -207,7 +207,7 @@ class LiveStreamRecorder:
self.save_format, use_direct_download = self._get_record_format(stream_info)
filename = self._get_filename(stream_info)
self.output_dir = self._get_output_dir(stream_info)
save_path = self._get_save_path(filename)
save_path = self._get_save_path(filename, use_direct_download)
logger.info(f"Save Path: {save_path}")
self.recording.recording_dir = os.path.dirname(save_path)
os.makedirs(self.recording.recording_dir, exist_ok=True)