chapters.py

import os
from mutagen.mp3 import MP3
from lxml import etree
import sys
import re


def chapterize(path, title, parts):
    if title:
        file = MP3(path + "/" + title + "-Part01.mp3")
    else:
        file = MP3(path + "/" + "Part01.mp3")
    chapters = etree.fromstring(file["TXXX:OverDrive MediaMarkers"][0])
    if chapters:
        length = 0.0
        f = open(f"{path}/chapters.txt", "w")
        for i in range(parts):
            part = "{:02d}".format(i + 1)
            if title:
                file = MP3(path + "/" + title + "-Part" + part + ".mp3")
            else:
                file = MP3(path + "/" + "Part" + part + ".mp3")
            chapters = etree.fromstring(file["TXXX:OverDrive MediaMarkers"][0])

            for chapter in chapters.getchildren():
                name = chapter[0].text

                time = chapter[1].text.split(":")
                time = length + float(time[0]) * 60 + float(time[1])

                h = "{:02d}".format(int(time / 3600))
                m = "{:02d}".format(int((time % 3600) / 60))
                s = "{:02d}".format(int(time % 60))
                ms = "{:03d}".format(int((time % 1) * 1000))

                duration = h + ":" + m + ":" + s + "." + ms
                # Filter out weird Overdrive parts of chapters
                if not re.findall("([(]\d|[cC]ontinued)", name):
                    f.write(duration + "    " + name + "\n")

            length += file.info.length


if __name__ == "__main__":
    try:
        mode = sys.argv[1]
    except IndexError:
        print(
            "usage: python3 chapters.py [-m Manual Mode, one directory at a time] [-a Automatic mode, all subdirectories] [-l List folders without chapters] <audiobook_directory>"
        )
        exit()
    if mode == "-m":
        try:
            folder = sys.argv[2]
        except IndexError:
            print("chapters.py -m <Audiobook directory>")
            exit()
        parts = 0
        for file in os.listdir(folder):
            if os.path.splitext(file)[1] == ".mp3":
                parts += 1
                try:
                    # Try in order to error if no dash
                    title = file.split("-")[1]
                    title = file.split("-")[0]
                except:
                    title = None
        chapterize(folder, title, parts)
    elif mode == "-a":
        for folder in os.listdir():
            if os.path.isdir(folder):
                parts = 0
                for file in os.listdir(folder):
                    if os.path.splitext(file)[1] == ".mp3":
                        parts += 1
                        try:
                            # Try in order to error if no dash
                            title = file.split("-")[1]
                            title = file.split("-")[0]
                        except:
                            title = None
                if parts > 0:
                    print("Extracting chapters from " + folder + "...")
                    try:
                        chapterize(folder, title, parts)
                    except:
                        print("Error!")
                else:
                    print("No mp3s, skipping...")
    elif mode == "-l":
        for folder in os.listdir():
            if os.path.isdir(folder):
                if "chapters.txt" in os.listdir(folder):
                    pass
                else:
                    print(folder + " is missing chapters!")
    else:
        print("Not sure what mode that is, but I don't do it.")