Klagarge/Melies
Klagarge/Melies
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Activity

feat: log file generation optional

Browse Source 
Log files are now only
generated when --save-log flag is set.
This commit is contained in:
Rémi Heredero 2025-04-27 16:34:20 +02:00
parent 3588d9ad14
commit 63cefacbfa
Signed by: Klagarge
GPG Key ID: 735B36B074A65F0F
1 changed files with 36 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
encode_video.py
View File

@ -14,7 +14,7 @@ def get_duration(file_path):
], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
return float(result.stdout.strip())
def encode(input_file, codec, remove_source=False):
def encode(input_file, codec, remove_source=False, save_log=False):
if codec == "x265":
ffmpeg_codec = "libx265"
crf = 26
@ -46,7 +46,12 @@ def encode(input_file, codec, remove_source=False):
outdir = os.path.join(os.path.dirname(input_file), folder)
os.makedirs(outdir, exist_ok=True)
output_file = os.path.join(outdir, f"{name}.mkv")
log_file = os.path.join(outdir, f"{name}.log")
if save_log:
log_file = os.path.join(outdir, f"{name}.log")
log = open(log_file, "w")
else:
log = subprocess.DEVNULL # Pas de log
cmd = [
"ffmpeg",
@ -60,30 +65,35 @@ def encode(input_file, codec, remove_source=False):
]
print(f"\n🎬 Encoding {filename} → {folder}/{name}.mkv with codec [{ffmpeg_codec}]")
with open(log_file, "w") as log:
process = subprocess.Popen(cmd, stderr=subprocess.PIPE, stdout=subprocess.PIPE, text=True)
process = subprocess.Popen(cmd, stderr=subprocess.PIPE, stdout=subprocess.PIPE, text=True)
time_re = re.compile(r"time=(\d+):(\d+):([\d.]+)")
last_percent = -1
time_re = re.compile(r"time=(\d+):(\d+):([\d.]+)")
last_percent = -1
while True:
line = process.stderr.readline()
if not line:
break
while True:
line = process.stderr.readline()
if not line:
break
if save_log:
log.write(line)
match = time_re.search(line)
if match:
h, m, s = map(float, match.groups())
current = h * 3600 + m * 60 + s
percent = int((current / duration) * 100)
if percent != last_percent:
print(f"\r⏳ Progress: {percent}%", end='', flush=True)
last_percent = percent
match = time_re.search(line)
if match:
h, m, s = map(float, match.groups())
current = h * 3600 + m * 60 + s
percent = int((current / duration) * 100)
if percent != last_percent:
print(f"\r⏳ Progress: {percent}%", end='', flush=True)
last_percent = percent
process.wait()
process.wait()
if save_log:
log.close()
if process.returncode != 0:
print(f"\n❌ [Error] FFmpeg failed for {filename}. Check the log: {log_file}")
print(f"\n❌ [Error] FFmpeg failed for {filename}.")
if save_log:
print(f" ➔ Check the log file: {log_file}")
if os.path.exists(output_file):
os.remove(output_file)
else:
@ -91,11 +101,11 @@ def encode(input_file, codec, remove_source=False):
if remove_source:
try:
os.remove(input_file)
print(f"🗑️ Source file {input_file} removed.")
print(f"🗑️ Source file {input_file} removed.")
except Exception as e:
print(f"⚠️ Could not delete source file {input_file}: {e}")
def encode_batch(directory, codec, remove_source=False):
def encode_batch(directory, codec, remove_source=False, save_log=False):
if not os.path.isdir(directory):
print(f"❌ Not a valid directory: {directory}")
return
@ -105,7 +115,7 @@ def encode_batch(directory, codec, remove_source=False):
if file.lower().endswith(SUPPORTED_EXTENSIONS):
filepath = os.path.join(root, file)
try:
encode(filepath, codec, remove_source)
encode(filepath, codec, remove_source, save_log)
except Exception as e:
print(f"\n❌ Error with file {file}: {e}")
continue
@ -116,15 +126,16 @@ def main():
parser.add_argument("-d", "--directory", help="Path to a directory for batch encoding")
parser.add_argument("--codec", choices=["x265", "x265-amd", "av1"], default="x265", help="Codec to use")
parser.add_argument("--remove-source", action="store_true", help="Remove the source file after successful encoding")
parser.add_argument("--save-log", action="store_true", help="Save ffmpeg logs to a .log file")
args = parser.parse_args()
if args.directory:
encode_batch(args.directory, args.codec, args.remove_source)
encode_batch(args.directory, args.codec, args.remove_source, args.save_log)
elif args.input:
if not os.path.isfile(args.input):
print(f"❌ File not found: {args.input}")
sys.exit(1)
encode(args.input, args.codec, args.remove_source)
encode(args.input, args.codec, args.remove_source, args.save_log)
else:
print("❌ Please provide a file path or use -d for a directory.")
parser.print_help()