feat: add progress bar for video creation
This commit is contained in:
@@ -7,7 +7,8 @@ requires-python = ">=3.9"
|
|||||||
|
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"obsws-python",
|
"obsws-python",
|
||||||
"watchdog"
|
"watchdog",
|
||||||
|
"tdqm"
|
||||||
]
|
]
|
||||||
|
|
||||||
[project.scripts]
|
[project.scripts]
|
||||||
|
|||||||
@@ -1,3 +1,4 @@
|
|||||||
obsws-python==1.8.0
|
obsws-python==1.8.0
|
||||||
|
tqdm==4.67.2
|
||||||
watchdog==6.0.0
|
watchdog==6.0.0
|
||||||
websocket-client==1.9.0
|
websocket-client==1.9.0
|
||||||
|
|||||||
@@ -5,6 +5,7 @@ import subprocess
|
|||||||
import json
|
import json
|
||||||
|
|
||||||
from rewind.state import load_state
|
from rewind.state import load_state
|
||||||
|
from tqdm import tqdm
|
||||||
|
|
||||||
def clip(seconds_from_end: float) -> None:
|
def clip(seconds_from_end: float) -> None:
|
||||||
output_file_name = f"{datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}.mp4"
|
output_file_name = f"{datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}.mp4"
|
||||||
@@ -153,7 +154,7 @@ def _concat_ts_files(file_list: list[str], start_offset: float, end_offset: floa
|
|||||||
for file_path in file_list:
|
for file_path in file_list:
|
||||||
f.write(f"file '{file_path}'\n")
|
f.write(f"file '{file_path}'\n")
|
||||||
|
|
||||||
cmd = ["ffmpeg", "-y"]
|
cmd = ["ffmpeg", "-y", "-hide_banner", "-loglevel", "error", "-nostats", "-progress", "pipe:1"]
|
||||||
if start_offset > 0:
|
if start_offset > 0:
|
||||||
cmd += ["-ss", str(start_offset)]
|
cmd += ["-ss", str(start_offset)]
|
||||||
if end_offset > 0:
|
if end_offset > 0:
|
||||||
@@ -161,9 +162,39 @@ def _concat_ts_files(file_list: list[str], start_offset: float, end_offset: floa
|
|||||||
cmd += ["-f", "concat", "-safe", "0", "-i", "file_list.txt", "-c", "copy"]
|
cmd += ["-f", "concat", "-safe", "0", "-i", "file_list.txt", "-c", "copy"]
|
||||||
cmd.append(output_file)
|
cmd.append(output_file)
|
||||||
|
|
||||||
subprocess.run(cmd)
|
process = subprocess.Popen(
|
||||||
|
cmd,
|
||||||
|
stdout=subprocess.PIPE,
|
||||||
|
stderr=subprocess.PIPE,
|
||||||
|
text=True,
|
||||||
|
bufsize=1,
|
||||||
|
)
|
||||||
|
|
||||||
|
with tqdm(
|
||||||
|
total=length,
|
||||||
|
unit="s",
|
||||||
|
unit_scale=False,
|
||||||
|
desc="Processing",
|
||||||
|
leave=True,
|
||||||
|
) as pbar:
|
||||||
|
for line in process.stdout:
|
||||||
|
line = line.strip()
|
||||||
|
|
||||||
|
if line.startswith("out_time_ms="):
|
||||||
|
out_time_ms = int(line.split("=")[1])
|
||||||
|
seconds = out_time_ms / 1_000_000
|
||||||
|
pbar.n = min(seconds, length)
|
||||||
|
pbar.refresh()
|
||||||
|
|
||||||
|
elif line == "progress=end":
|
||||||
|
break
|
||||||
|
|
||||||
|
ret = process.wait()
|
||||||
os.remove("file_list.txt")
|
os.remove("file_list.txt")
|
||||||
|
|
||||||
|
if ret != 0:
|
||||||
|
raise RuntimeError("ffmpeg failed")
|
||||||
|
|
||||||
def get_duration(file_path: str) -> float:
|
def get_duration(file_path: str) -> float:
|
||||||
result = subprocess.run(
|
result = subprocess.run(
|
||||||
["ffprobe", "-v", "error", "-show_entries",
|
["ffprobe", "-v", "error", "-show_entries",
|
||||||
|
|||||||
Reference in New Issue
Block a user