84 lines
3.1 KiB
Python
84 lines
3.1 KiB
Python
#Confirmed working as of 1/22/23
|
|
|
|
import os
|
|
import shutil
|
|
from pydub import AudioSegment
|
|
from config import annex,circulation,archive
|
|
|
|
AudioSegment.converter = "/opt/homebrew/Cellar/ffmpeg/5.1.2_3/bin/ffmpeg"
|
|
AudioSegment.ffmpeg = "/opt/homebrew/Cellar/ffmpeg/5.1.2_3/bin/ffmpeg"
|
|
AudioSegment.ffprobe ="/opt/homebrew/Cellar/ffmpeg/5.1.2_3/bin/ffmpeg"
|
|
|
|
def export_audio_file(start, end, name):
|
|
start_ms = int(start)
|
|
end_ms = int(end)
|
|
export = combined[start_ms:end_ms]
|
|
folder_name = folder.split(" - ")[-1].strip()
|
|
destination_path = circulation
|
|
author_name = folder.split(" - ")[0].strip()
|
|
export_folder = os.path.join(destination_path, author_name, folder_name)
|
|
author_folder = os.path.join(destination_path, author_name)
|
|
if not os.path.exists(author_folder):
|
|
os.makedirs(author_folder)
|
|
export_folder = os.path.join(author_folder, folder_name)
|
|
|
|
if not os.path.exists(export_folder):
|
|
os.makedirs(export_folder)
|
|
export_path = os.path.join(export_folder, f"{name}.mp3")
|
|
export.export(export_path, format="mp3")
|
|
|
|
metadata_file = os.path.join(folder_path, "cleaned_metadata.json")
|
|
chapters_file = os.path.join(folder_path, "overdrive_chapters_ms.txt")
|
|
album_art = os.path.join(folder_path, "folder.jpg")
|
|
if os.path.exists(album_art):
|
|
shutil.copy(album_art, export_folder)
|
|
shutil.copy(metadata_file, export_folder)
|
|
|
|
main_directory = annex
|
|
|
|
for folder in sorted(os.listdir(main_directory)):
|
|
folder_path = os.path.join(main_directory, folder)
|
|
if os.path.isdir(folder_path):
|
|
try:
|
|
# Import MP3s in alphabetical order
|
|
mp3_files = sorted(os.listdir(folder_path))
|
|
mp3_files = [os.path.join(folder_path, file) for file in mp3_files if file.endswith(".mp3")]
|
|
|
|
combined = AudioSegment.empty()
|
|
for mp3_file in mp3_files:
|
|
print(mp3_file)
|
|
combined += AudioSegment.from_file(mp3_file)
|
|
|
|
# Align end to end
|
|
combined = combined.set_channels(1)
|
|
|
|
# Import labels
|
|
labels_file = os.path.join(folder_path, "overdrive_chapters_ms_spans.txt")
|
|
labels = []
|
|
with open(labels_file, "r") as f:
|
|
for line in f:
|
|
start, end, name = line.strip().split("\t")
|
|
labels.append((start, end, name))
|
|
|
|
# Initialize counter for duplicate label names
|
|
counter = {}
|
|
|
|
# Initialize counter for file export
|
|
file_count = 0
|
|
|
|
for i, label in enumerate(labels):
|
|
start, end, name = label
|
|
if end == "0" and i == len(labels) - 1:
|
|
end = combined.duration_seconds * 1000
|
|
|
|
file_count += 1
|
|
name = f"{file_count:03}_{name.replace('/', '_')}"
|
|
# format file_count as 3 digit number
|
|
export_audio_file(start, end, name)
|
|
|
|
except Exception as e:
|
|
print(f"Error processing {folder}: {e}")
|
|
|
|
#Archive everything except chapters_list.py
|
|
if folder_path != "chapters_list.py":
|
|
shutil.move(folder_path, archive/ + folder) |