diff --git a/aax2mp3 b/aax2mp3 index 039e8f7..b4dd043 100755 --- a/aax2mp3 +++ b/aax2mp3 @@ -5,6 +5,10 @@ import subprocess import glob import os import multiprocessing +import re +import nltk +from nltk.corpus import words +from nltk.corpus import wordnet # Different ways to get CPU count logical_cores = os.cpu_count() # All cores including hyperthreading @@ -24,6 +28,59 @@ def convert_file(aax_file): print(f"Converting {aax_file} to {mp3_file}") subprocess.run(['ffmpeg', '-activation_bytes', os.getenv('AUDIBLE_ACTIVATION_BYTES'), '-i', aax_file, mp3_file], check=True) + return mp3_file + +def rename_file(mp3_file): + print(f"\n{'='*50}") + print(f"Original: {mp3_file}") + base, ext = os.path.splitext(mp3_file) + new_base = rename_base(base) + new_name = new_base + ext + print(f"New name: {new_name}") + + if new_name != mp3_file: + try: + os.rename(mp3_file, new_name) + print(f"Renamed: {mp3_file} -> {new_name}") + except OSError as e: + print(f"Error renaming {mp3_file}: {e}") + +def rename_base(base): + words = re.split('(?