79 lines
2.1 KiB
Python
79 lines
2.1 KiB
Python
|
# Recursivly scan from the currenct directory and add every flac file to an array
|
||
|
|
|
||
|
|
import os
|
||
|
|
|
||
|
|
import json
|
||
|
|
import subprocess
|
||
|
|
import ffmpeg
|
||
|
|
|
||
|
|
# TODO: Rewrite using ffmpeg-python probe functionality
|
||
|
|
def _ffprobe(file_path):
|
||
|
|
command = (
|
||
|
|
'ffprobe -v quiet -print_format json -show_streams -select_streams a "'f'{file_path}"'
|
||
|
|
)
|
||
|
|
result = subprocess.run(command, capture_output=True)
|
||
|
|
if result.returncode == 0:
|
||
|
|
return json.loads(result.stdout)
|
||
|
|
raise RuntimeError(
|
||
|
|
f'FFProbe failed for {file_path}, output: {result.stderr}'
|
||
|
|
)
|
||
|
|
|
||
|
|
def scan_dir():
|
||
|
|
flac_files = []
|
||
|
|
for root, dirs, files in os.walk(os.getcwd()):
|
||
|
|
for file in files:
|
||
|
|
if file.endswith(".flac"):
|
||
|
|
flac_files.append(os.path.join(root, file))
|
||
|
|
return flac_files
|
||
|
|
|
||
|
|
|
||
|
|
# Filter the array of flac files to only include files with a bit depth higher than 16, and a sample rate higher than 44.1kHz
|
||
|
|
def filter_files(flac_files):
|
||
|
|
filtered_files = []
|
||
|
|
counter = 0
|
||
|
|
max_counter = len(flac_files)
|
||
|
|
for file in flac_files:
|
||
|
|
counter += 1
|
||
|
|
print("Progress: " + str(counter) + "/" + str(max_counter))
|
||
|
|
metadata = _ffprobe(file)
|
||
|
|
if int(metadata['streams'][0]['bits_per_raw_sample']) > 16 and int(
|
||
|
|
metadata['streams'][0]['sample_rate']) > 44100:
|
||
|
|
filtered_files.append(file)
|
||
|
|
return filtered_files
|
||
|
|
|
||
|
|
|
||
|
|
# Print the filtered files to the console
|
||
|
|
|
||
|
|
def print_files(filtered_files):
|
||
|
|
for file in filtered_files:
|
||
|
|
print(file)
|
||
|
|
|
||
|
|
|
||
|
|
def get_relative_path(file):
|
||
|
|
return os.path.relpath(file, os.getcwd())
|
||
|
|
|
||
|
|
def get_root_path(file):
|
||
|
|
return os.path.dirname(file)
|
||
|
|
|
||
|
|
|
||
|
|
def convert_files(filtered_files):
|
||
|
|
for file in filtered_files:
|
||
|
|
output = ".\\Output\\" + get_relative_path(file)
|
||
|
|
|
||
|
|
output_no_basename = os.path.dirname(output)
|
||
|
|
os.makedirs(output_no_basename, exist_ok=True)
|
||
|
|
|
||
|
|
stream = ffmpeg.input(file)
|
||
|
|
stream = ffmpeg.output(stream, output, sample_fmt="s16", ar='44100')
|
||
|
|
stream = ffmpeg.overwrite_output(stream)
|
||
|
|
ffmpeg.run(stream)
|
||
|
|
|
||
|
|
# Run the functions
|
||
|
|
|
||
|
|
flac_files = scan_dir()
|
||
|
|
print_files(flac_files)
|
||
|
|
filtered_files = filter_files(flac_files)
|
||
|
|
print_files(filtered_files)
|
||
|
|
convert_files(filtered_files)
|
||
|
|
|