From ee9deff6fecf1354ea3e529dda3b7b850a81050f Mon Sep 17 00:00:00 2001
From: Chris Pomeroy <chris.pomeroy@hotmail.com>
Date: Sat, 17 Jan 2026 20:24:48 +0000
Subject: [PATCH] Adding back to source control
---
aaxConvert.py | 387 +++++++++++++++++++++++++++++++++---------------------
1 files changed, 236 insertions(+), 151 deletions(-)
diff --git a/aaxConvert.py b/aaxConvert.py
index 8109422..d3386d0 100755
--- a/aaxConvert.py
+++ b/aaxConvert.py
@@ -1,190 +1,275 @@
-#!/usr/local/bin/python -u
+#!/usr/bin/env -S python -u
import argparse
import os
-import glob
import subprocess
-import shlex
import json
-import requests
-import unicodedata
-from querysubsonic import findalbumbyname
-from time import sleep
-
+import re
+import httpx
+import sys
# arguments
# activation_key, file name, codec(default to mp3)
+# Globals
-parser = argparse.ArgumentParser()
-parser.add_argument("-s","--single", help="Use this option to create a single file. This is false by default", action="store_true")
-parser.add_argument("-d","--dpath", help="Use this to set the destination path. Otherwise I will use the current directory")
-parser.add_argument("-v", "--verbose", help="Send output to stdout", action="store_true")
-parser.add_argument("filename", help="Filename to convert, or directory to look in")
-
-args = parser.parse_args()
-
-act_byte = ""
-metadata = ""
+_act_byte = ""
+_metadata = ""
mode = ""
stats = ""
-
-if args.dpath:
- path = args.dpath
-else:
- path = os.getcwd()
-
-if args.single:
- mode = 'single'
-else:
- mode = 'chapter'
-
-if args.verbose:
- stats = "-stats"
-else:
- stats = "-nostats"
+path = ""
-def getmetadata(aaxfile):
-# Returns the metadata from an aax file
- ret = subprocess.run(["ffprobe", "-v", "info", "-hide_banner", "-show_format", "-show_chapters", "-print_format", "json", os.path.abspath(aaxfile)], capture_output=True)
- mdata = json.loads(ret.stdout)
- aret = ret.stderr.decode().split('\n')[0]
- mdata["checksum"] = aret.split()[-1]
- return mdata
+def get_args():
+ parser = argparse.ArgumentParser()
+ parser.add_argument("-s", "--single", help="Use this option to create a\
+ single file. This is false by default",
+ action="store_true")
+ parser.add_argument("-d", "--dpath", help="Use this to set the destination\
+ path. Otherwise I will use the current directory")
+ parser.add_argument("-v", "--verbose", help="Send output to stdout",
+ action="store_true")
+ parser.add_argument("filename", help="Filename to convert, or directory to\
+ look in")
+ parser.add_argument("-a", "--actvation_bytes", help='The activate bytes\
+ provided by audible')
+
+ args = parser.parse_args()
+
+ if args.dpath:
+ path = args.dpath
+ else:
+ path = os.getcwd()
+
+ if args.single:
+ mode = 'single'
+ else:
+ mode = 'chapter'
+
+ if args.verbose:
+ stats = "-stats"
+ else:
+ stats = "-nostats"
+
+ return args
-def getmetabitrate():
-#Return the bitrate of the media
- bit_rate = metadata['format']['bit_rate']
+def sendtodiscord(message, audiofile):
+ '''
+ Send a message to the discord group
+
+ Arg:
+ message (str): Message to send.
+ audiofile (str): File name that was processed
+
+ Ret:
+ Boolean
+ '''
+ webhookurl = "https://discord.com/api/webhooks/764667082272145418/vorf2JdFG47WAmQP3yZhgHH12wW_qUXG0bS0SG8INLYVwU0HcDFajq9doaDgi_hnI00-" # noqa E501
+ data = {
+ "content": f"{message}, {audiofile}",
+ "username": "Captain Audio",
+ }
+
+ with httpx.Client() as r:
+ resp = r.post(webhookurl, json=data)
+ if resp.status_code == 204:
+ return True
+ else:
+ return False
+
+
+def get_metadata(aaxfile):
+ # Returns the _metadata from an aax file
+ try:
+ ret = subprocess.run(["ffprobe", "-v", "info", "-hide_banner",
+ "-show_format", "-show_chapters",
+ "-print_format", "json",
+ os.path.abspath(aaxfile)], capture_output=True)
+ if ret.returncode == 0:
+ mdata = json.loads(ret.stdout)
+ aret = ret.stderr.decode().split('\n')[0]
+ print(aret)
+ mdata["checksum"] = aret.split()[-1]
+ return mdata
+ else:
+ sys.stderr.write(ret.stderr)
+ return "None"
+ except Exception as err:
+ sys.stderr.write("Error processing _metadata: {}\n".format(err))
+ return "None"
+
+
+def get_bitrate():
+ # Return the bitrate of the media
+ bit_rate = _metadata['format']['bit_rate']
return bit_rate[:2]
-def getmetacopyright():
-# Return normalized copyright data
- copyright = unicodedata.normalize('NFKD', metadata['format']['tags']['copyright']).encode('ascii','ignore')
- return copyright.decode()
-def getmetadatatags(key):
-# get specific data
- tag = metadata['format']['tags'][key]
- return tag
+def get_copyright():
+ # Return normalized copyright data
+ copyright = normalize_data(_metadata['format']['tags']['copyright'])
+ return copyright
-def reencode(aaxfile, outpath):
-# decrypt and reencode to mp3
- command = ("ffmpeg -loglevel error {} -activation_bytes {} -i {} -vn -codec:a libmp3lame -ab {}k -map_metadata -1 "
- "-metadata \"title={}\" -metadata 'artist={}' -metadata 'album_artist={}' -metadata \"album={}\" -metadata 'date={}' "
- "-metadata track=1/1 -metadata 'genre={}' -metadata 'copyright={}' \"{}\" " ).format(stats, act_byte,aaxfile,getmetabitrate(),
- getmetadatatags('title'), getmetadatatags('artist'),
- getmetadatatags('album_artist'), getmetadatatags('album'),
- getmetadatatags('date'), getmetadatatags('genre'),
- getmetacopyright(),outpath)
- if args.verbose:
- print(command)
- process = subprocess.Popen(shlex.split(command), stdout=subprocess.PIPE)
- while True:
- output = process.stdout.readline()
- if output == '' and process.poll() is not None:
- break
- if output:
- print(output.strip())
- rc = process.poll()
- return rc
+def get_metadata_tags(key):
+ # get specific data
+ tag = normalize_data(_metadata['format']['tags'][key])
+ return " ".join(tag.split())
+
+
+def normalize_data(data):
+ # Return a normalized title
+ data = data.replace("'", "").replace('(', '').replace(')', '').replace(':', '').replace('/', '-').replace('\\', '-').replace('"', '').replace('?', '').replace('!', '').strip()
+ # pattern = re.compile('[^\p{Latin}]', data)
+ return data
+
+
+def reencode(aaxfile: str,
+ outpath: str,
+ act_byte: str = None,
+ key_iv: dict = None
+ ):
+ # decrypt and reencode to mp3
+ if get_metadata_tags('major_brand') == 'aaxc':
+ decrypt = f"-audible_key {key_iv['key']} -audible_iv {key_iv['iv']}"
else:
- process = subprocess.call(shlex.split(command))
- return
+ decrypt = f"-activation_bytes = {act_byte}"
+ command = (
+ "ffmpeg -loglevel error {} {} -i"
+ " {} -vn -codec:a libmp3lame -qscale:a 2 -map_metadata -1"
+ " -metadata \"title={}\" -metadata 'artist={}' -metadata"
+ " 'album_artist={}' -metadata \"album={}\" -metadata 'date={}' "
+ " -metadata track=1/1 -metadata 'genre={}' -metadata "
+ "'copyright={}' \"{}\" ").format(
+ stats, decrypt, aaxfile,
+ get_metadata_tags('title'), get_metadata_tags('artist'),
+ get_metadata_tags('album_artist'), get_metadata_tags('album'),
+ get_metadata_tags('date'), get_metadata_tags('genre'),
+ get_copyright(), outpath)
+ process = subprocess.run(command, shell=True)
def getchaptercount():
-# Get the number of chapters
- ccount = metadata['chapters']
+ # Get the number of chapters
+ ccount = _metadata['chapters']
return len(ccount)
-def getchaptermetadata(cid, key):
-# get the Chapter metadata
- for i in metadata['chapters']:
+def get_chapter_metadata(cid, key):
+ # get the Chapter _metadata
+ for i in _metadata['chapters']:
if i['id'] == cid:
return i[key]
-def movetochapters(path, outpath, chapter, title, start,end):
-# Creating individual chapters
- outfile = "{}/Ch-{}_{}.mp3".format(outpath, chapter, title.replace(' ', '_'))
+def movetochapters(path, outpath, chapter, title, start, end):
+ # Creating individual chapters
+
+ outfile = "{}/Ch-{} {}.mp3".format(
+ outpath, chapter, title)
command = "ffmpeg -loglevel error {} -i \"{}\" -ss {} -to {} -codec:a copy -metadata 'track={}' \"{}\"".format(stats, path,
- start, end,
- chapter, outfile)
- if args.verbose:
- print(command)
- process = subprocess.Popen(shlex.split(command), stdout=subprocess.PIPE)
- while True:
- output = process.stdout.readline()
- if output == '' and process.poll() is not None:
- break
- if output:
- print(output.strip())
- rc = process.poll()
- return rc
- else:
- process = subprocess.call(shlex.split(command))
+ start, end,
+ chapter, outfile)
+ subprocess.run(command, shell=True)
return
-def getcoverart(path,outpath):
-# Pull the coverart from the file
+def getcoverart(path, outpath):
+ # Pull the coverart from the file
command = "ffmpeg -loglevel error -activation_bytes {} -i \"{}\" -an -codec:v copy \"{}/cover.jpg\"".format(act_byte,
- path, outpath)
- if args.verbose:
- print(command)
- process = subprocess.call(shlex.split(command))
- return
+ path, outpath)
+ subprocess.run(command, shell=True)
-# def filechecksum(aaxfile):
-# #The the hex checksum from the file
-# ret = subprocess.Popen(["ffprobe", "-v", "info", "-hide_banner", os.path.abspath(aaxfile)], stderr=subprocess.PIPE)
-# grep = subprocess.Popen(["grep", "checksum"], stdin=ret.stderr, stdout=subprocess.PIPE)
-# awk = subprocess.Popen(["awk", " { print $8 } "], stdin=grep.stdout, stdout=subprocess.PIPE)
-# hashsum,out = awk.communicate()
-# hashsum = hashsum.decode().strip('\n')
-# return hashsum
-def getcorrectkey():
-#request the key for the checksum
- try:
- r = requests.post('http://prod-faas-gw.darkurthe.net:8080/function/checkkey', metadata['checksum'], verify=False, timeout=None)
- return r.text
- except requests.exceptions.HTTPError as err:
- raise error
- return None
+def convert_aax(
+ rfile: str,
+ activation_bytes: str = None,
+ key_iv: dict = None
+ ) -> bool:
+ '''
+ Do the work of convering the file to mp3.
-for rfile in glob.glob(args.filename):
- if rfile.find("aax") != -1 and os.path.isfile(rfile):
- metadata = getmetadata(rfile)
- album = getmetadatatags(album)
- #See if we got it already
- if (findalbumbyname(album) == False):
- artist = getmetadatatags('artist')
- title = getmetadatatags('title')
- act_byte = getcorrectkey()
- if act_byte == None:
- sys.exit("Can't continue with this file {rfile}")
- else:
- ddir = "%s/%s/%s" % (path, artist.replace(' ', '_'), title.replace(' ', '_'))
- single_file_path = "/processing/%s.mp3" % (title.replace(' ','_'))
- if not os.path.exists(ddir):
- os.makedirs(ddir)
- print(ddir)
- reencode(rfile, single_file_path)
- if mode == 'chapter':
- chapter = 0
- numchapters = getchaptercount()
- while (numchapters > 0 ):
- cstart = getchaptermetadata(chapter, 'start_time')
- cend = getchaptermetadata(chapter, 'end_time')
- chapter += 1
- numchapters -= 1
- schap = str(chapter).zfill(2)
- movetochapters(single_file_path, ddir, schap, title, cstart,cend)
- os.remove(single_file_path)
- getcoverart(rfile, ddir)
+ Args:
+ rfile (str): The file to convert
+ activation_bytes Optional(str): The unique activation bytes from audible. # noqa E501
+
+ Ret:
+ Boolean if successful
+ '''
+ chapter = 0
+ global _metadata
+ _metadata = get_metadata(rfile)
+ global act_byte
+ act_byte = activation_bytes
+ if _metadata == "None":
+ sendtodiscord("There was a problem processing ", rfile)
+ else:
+ # artist = normalize_data(_metadata['format']['tags']['artist'])
+ # title = normalize_data(_metadata['format']['tags']['title'])
+ artist = normalize_data(_metadata['format']['tags']['artist'])
+ title = normalize_data(_metadata['format']['tags']['title'])
+ if activation_bytes is None or activation_bytes == '':
+ sendtodiscord("We couldn't get the activation bytes", rfile)
+ return False
else:
- print('We have that book already')
-
\ No newline at end of file
+ dest_dir = f"/workspaces/audiobooks/{artist}/{title}"
+ processing_file = f"/tmp/{title}.mp3"
+ if not os.path.exists(dest_dir):
+ os.makedirs(dest_dir)
+ reencode(rfile, processing_file, activation_bytes, key_iv)
+ print(f"Splitting {title} into chapters")
+ numchapters = len(_metadata['chapters'])
+ while (numchapters > 0):
+ cstart = get_chapter_metadata(chapter, 'start_time')
+ cend = get_chapter_metadata(chapter, 'end_time')
+ chapter += 1
+ numchapters -= 1
+ schap = str(chapter).zfill(2)
+ movetochapters(processing_file, dest_dir, schap, title,
+ cstart, cend)
+ os.remove(processing_file)
+ getcoverart(rfile, dest_dir)
+ sendtodiscord(f"We have processed the book {title} ", rfile)
+ return True
+
+
+def main():
+ '''
+ Main excution
+ '''
+ args = get_args()
+ rfile = args.filename
+ _metadata = get_metadata(rfile)
+ if _metadata == "None":
+ sendtodiscord("There was a problem processing ", rfile)
+ else:
+ artist = normalize_data(get_metadata_tags('artist'))
+ title = normalize_data(get_metadata_tags('title'))
+ act_byte = args.activation_bytes
+ if act_byte is None or act_byte == '':
+ sendtodiscord("We couldn't get the activation bytes", rfile)
+ sys.exit(f"We couldn't get the activation bytes {rfile}")
+ else:
+ ddir = "%s/%s/%s" % (path, artist, title)
+ single_file_path = "/processing/%s.mp3" % (title)
+ if not os.path.exists(ddir):
+ os.makedirs(ddir)
+ print(ddir)
+ reencode(rfile, single_file_path)
+ if mode == 'chapter':
+ chapter = 0
+ numchapters = getchaptercount()
+ while (numchapters > 0):
+ cstart = get_chapter_metadata(chapter, 'start_time')
+ cend = get_chapter_metadata(chapter, 'end_time')
+ chapter += 1
+ numchapters -= 1
+ schap = str(chapter).zfill(2)
+ movetochapters(single_file_path, ddir, schap, title,
+ cstart, cend)
+ os.remove(single_file_path)
+ getcoverart(rfile, ddir)
+ sendtodiscord(f"We have added the book {title} ", rfile)
+
+
+if __name__ == "__main__":
+ main()
--
Gitblit v1.10.0