From 783191e036ccceca78b8e41ad27e7cd349f7958c Mon Sep 17 00:00:00 2001
From: Chris Pomeroy <chris.pomeroy@hotmail.com>
Date: Mon, 05 Mar 2018 03:24:30 +0000
Subject: [PATCH] Updated to python 3
---
aaxConvert.py | 203 +++++++++++++++++++++++++++++++++++++++++++-------
1 files changed, 175 insertions(+), 28 deletions(-)
diff --git a/aaxConvert.py b/aaxConvert.py
index 01c91bb..ae9567d 100755
--- a/aaxConvert.py
+++ b/aaxConvert.py
@@ -1,28 +1,175 @@
-#!/usr/bin/python
-import argparse
-import os
-import glob
-
-# arguments
-# activation_key, file name, codec(default to mp3)
-parser = argparse.ArgumentParser()
-parser.add_argument("--code", help="Activation Code from audible to decrypt files")
-parser.add_argument("filename", help="Filename to convert")
-
-args = parser.parse_args()
-
-act_byte = ""
-if args.code:
- act_byte = args.code
-elif os.path.isfile('./.authcode'):
- act_byte = open('./.authcode').readline()
-
-if act_byte == "":
- print "Please provide an activation code from audible. Either at the command line or the .authcode file"
-else:
- print act_byte
-
-for rfile in glob.glob(args.filename):
- if (rfile.find("aax") != -1):
- if os.path.isfile(rfile):
- print os.path.abspath(rfile)
+#!/usr/bin/python
+import argparse
+import os
+import glob
+import subprocess
+import shlex
+import json
+import requests
+from querysubsonic import findalbumbyname
+
+
+# arguments
+# activation_key, file name, codec(default to mp3)
+
+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 = ""
+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"
+
+
+def getmetadata(aaxfile):
+ ret = subprocess.check_output(["ffprobe", "-v", "info", "-hide_banner", "-show_format", "-show_chapters", "-print_format", "json", os.path.abspath(aaxfile)])
+ mdata = json.loads(ret)
+ return mdata
+
+
+def getmetabitrate():
+ bit_rate = metadata['format']['bit_rate']
+ return bit_rate[:2]
+
+
+def getmetadatatags(key):
+ # get specific data
+ tag = metadata['format']['tags'][key]
+ return tag
+
+
+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'),
+ getmetadatatags('copyright'),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
+ else:
+ process = subprocess.call(shlex.split(command))
+ return
+
+
+def getchaptercount():
+ # Get the number of chapters
+ ccount = metadata['chapters']
+ return len(ccount)
+
+
+def getchaptermetadata(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)
+ 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))
+ return
+
+
+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
+
+def filechecksum(aaxfile):
+ 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.strip('\n')
+ return hashsum
+
+def getcorrectkey(aaxfile):
+ hex = filechecksum(aaxfile)
+ payload = hex
+ r = requests.post('http://x86-docker01:8080/function/checkkey', payload, verify=False)
+ if r.status_code == 200:
+ key = r.text
+ else:
+ print('Checkkey function is down please wait')
+ return key
+
+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(rfile)
+ ddir = "%s/%s/%s" % (path, artist, title)
+ single_file_path = "%s/%s.mp3" % (ddir, 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 = 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)
+
\ No newline at end of file
--
Gitblit v1.10.0