125 lines
4.0 KiB
Python
125 lines
4.0 KiB
Python
#!/usr/bin/env python3
|
||
import json
|
||
import os
|
||
import subprocess
|
||
import time
|
||
import requests
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
PROM_URL = "http://127.0.0.1:9104/metrics"
|
||
|
||
def load_config(path):
|
||
"""Load JSON configuration file."""
|
||
with open(path, "r", encoding="utf-8") as f:
|
||
return json.load(f)
|
||
|
||
def wait_for_solar_power():
|
||
"""
|
||
Wait until the Prometheus metric `fronius_site_autonomy_ratio` equals 1.0.
|
||
Check every 5 minutes if not available.
|
||
"""
|
||
while True:
|
||
try:
|
||
r = requests.get(PROM_URL, timeout=5)
|
||
if r.status_code == 200:
|
||
for line in r.text.splitlines():
|
||
if line.startswith("fronius_site_autonomy_ratio"):
|
||
try:
|
||
value = float(line.split()[-1])
|
||
if value == 1.0:
|
||
print("[INFO] Solar power available – starting conversion.")
|
||
return
|
||
except ValueError:
|
||
pass
|
||
print("[INFO] No solar power – waiting 5 minutes...")
|
||
except requests.RequestException as e:
|
||
print(f"[WARN] Could not reach Prometheus: {e}")
|
||
time.sleep(300) # Wait 5 minutes
|
||
|
||
def analyze_codecs(oldfile, newfile, dst_folder):
|
||
"""
|
||
Run codecVis to compare old and new files.
|
||
Then rename output.png to newfile.png.
|
||
"""
|
||
cmd = ["codecVis", oldfile, newfile]
|
||
print(f"[CMD] {' '.join(cmd)}")
|
||
subprocess.run(cmd, cwd=dst_folder, check=True)
|
||
|
||
# Rename output.png to match the new file
|
||
output_png = Path(dst_folder) / "output.png"
|
||
new_png = Path(dst_folder) / (Path(newfile).name + ".png")
|
||
if output_png.exists():
|
||
output_png.rename(new_png)
|
||
print(f"[INFO] Analysis image saved as {new_png}")
|
||
else:
|
||
print("[WARN] output.png not found!")
|
||
|
||
def main():
|
||
if len(sys.argv) != 2:
|
||
print("Give path to config file as argument.")
|
||
sys.exit(1)
|
||
|
||
cfg = load_config(sys.argv[1])
|
||
dst_folder = Path(cfg["dst_folder"])
|
||
src_folder = Path(cfg["src_folder"])
|
||
dst_folder.mkdir(parents=True, exist_ok=True)
|
||
|
||
for job in cfg["jobs"]:
|
||
src_file = src_folder / job
|
||
tmp_movie_name = "tmp_" + Path(job).stem + ".mkv"
|
||
movie_name = Path(job).stem + ".mkv"
|
||
|
||
tmp_dst_file = dst_folder / tmp_movie_name
|
||
dst_file = dst_folder / movie_name
|
||
|
||
print(f"Source: {src_file}")
|
||
print(f"Temp name: {tmp_movie_name}")
|
||
print(f"Final name: {movie_name}")
|
||
|
||
# Remove leftover temporary file
|
||
if tmp_dst_file.exists():
|
||
print(f"[INFO] File {tmp_dst_file} already exists. --> Delete!")
|
||
tmp_dst_file.unlink()
|
||
|
||
# Skip if final file already exists
|
||
if dst_file.exists():
|
||
print(f"[INFO] Skip {dst_file}, already exists. --> Convert already done!")
|
||
continue
|
||
|
||
wait_for_solar_power()
|
||
|
||
try:
|
||
cmd = [
|
||
"taskset", "-c", "0,1,2,3", # limit to first 4 CPU cores
|
||
"ffmpeg", "-i", str(src_file),
|
||
"-c:v", "libaom-av1",
|
||
"-c:a", "libopus",
|
||
"-mapping_family", "1",
|
||
"-af", "aformat=channel_layouts=5.1",
|
||
"-c:s", "copy",
|
||
"-map", "0",
|
||
"-crf", "24",
|
||
"-b:v", "0",
|
||
"-b:a", "128k",
|
||
"-cpu-used", "4",
|
||
"-row-mt", "1",
|
||
"-tiles", "2x2",
|
||
str(tmp_dst_file)
|
||
]
|
||
print(f"[CMD] {' '.join(cmd)}")
|
||
subprocess.run(cmd, check=True)
|
||
|
||
# Rename temp file to final name
|
||
tmp_dst_file.rename(dst_file)
|
||
|
||
# Run codec analysis
|
||
analyze_codecs(str(src_file), str(dst_file), dst_folder)
|
||
|
||
except subprocess.CalledProcessError as e:
|
||
print(f"[ERROR] Processing failed for {job}: {e}")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|