152 lines
5.2 KiB
Python
152 lines
5.2 KiB
Python
import configparser
|
|
from requests import get
|
|
from feedgen.feed import FeedGenerator
|
|
import datetime
|
|
import os
|
|
import re
|
|
import json
|
|
import sys
|
|
|
|
# osSafeName() convertes a string to an OS safe name.
|
|
def osSafeName(name):
|
|
return re.sub(r'[^a-zA-Z0-9 -]', '', name).lower().replace(' ', '-').replace('--', '-')
|
|
|
|
class AudibleBook(object):
|
|
|
|
def __init__(self, filePath, activationBytes, targetDir):
|
|
self.filePath = filePath
|
|
self.activationBytes = activationBytes
|
|
self.targetDir = targetDir
|
|
self.convertedFilePath=''
|
|
self.metadata = json.loads(os.popen('ffprobe -i {} -show_format \
|
|
-print_format json'.format(filePath)).read())['format']['tags']
|
|
self.chapters = json.loads(os.popen('ffprobe -i {} -print_format json \
|
|
-show_chapters -loglevel error -sexagesimal'.format(self.filePath))
|
|
.read())['chapters']
|
|
|
|
# getPath() returns path to origonal audible file.
|
|
def getPath(self):
|
|
return self.filePath
|
|
|
|
# getMetadata() returns audio metadata for a given filepath.
|
|
def getMetadata(self):
|
|
return self.metadata
|
|
|
|
# getChapters() returns chapter metadata for a given filepath.
|
|
def getChapters(self):
|
|
return self.chapters
|
|
|
|
# getTitle() returns audiobook title.
|
|
def getTitle(self):
|
|
return self.getMetadata()['title']
|
|
|
|
def getAuthor(self):
|
|
return self.getMetadata()['artist']
|
|
|
|
def setTargetDir(self, targetDir):
|
|
self.targetDir = targetDir
|
|
|
|
def removeDRM(self, fileName):
|
|
filePath = '{}/{}.m4b'.format(self.targetDir, fileName)
|
|
if not os.path.exists(filePath):
|
|
os.system('ffmpeg -activation_bytes {} -i {} -c copy "{}"'
|
|
.format(self.activationBytes, self.filePath, filePath))
|
|
self.convertedFilePath = filePath
|
|
|
|
def splitChapters(self):
|
|
for c in self.getChapters():
|
|
start=c['start_time']
|
|
end=c['end_time']
|
|
title=c['tags']['title']
|
|
outFilePath = '{}-{}.m4a'.format(os.path.splitext(self.convertedFilePath)[0],
|
|
osSafeName(title))
|
|
if not os.path.exists(outFilePath):
|
|
os.system('ffmpeg -i {} -acodec copy -vcodec copy -ss {} -t {} {}'.
|
|
format(self.convertedFilePath, start, end, outFilePath))
|
|
def getCover(self):
|
|
outFilePath = '{}/cover.jpg'.format(self.targetDir)
|
|
if not os.path.exists(outFilePath):
|
|
os.system('ffmpeg -i {} -an -vcodec copy {}'.
|
|
format(self.convertedFilePath, outFilePath))
|
|
|
|
def deleteConverted(self):
|
|
if self.convertedFilePath is not '' and os.path.exists(self.configFilePath):
|
|
os.remove(self.convertedFilePath)
|
|
|
|
|
|
# getHash() returns the hash of a given file.
|
|
def getHash(filePath):
|
|
with open(filePath, 'rb') as f:
|
|
f.seek(653)
|
|
data = f.read(20)
|
|
f.close()
|
|
return data.hex()
|
|
|
|
# getActivationBytes returns the bytes needed to decrypt a given hash.
|
|
def getActivationBytes(filehash):
|
|
headers = {'User-Agent': 'audible-converter'}
|
|
response = get('https://aax.api.j-kit.me/api/v2/activation/{}'.format(filehash),
|
|
headers=headers)
|
|
return json.loads(response.text)['activationBytes']
|
|
|
|
# createConfig() creates a config file.
|
|
def createConfig(configFilePath, filePath):
|
|
activationBytes = getActivationBytes(getHash(filePath))
|
|
f = open(configFilePath, 'w')
|
|
f.write('activationBytes={}'.format(activationBytes))
|
|
f.close()
|
|
|
|
configFilePath='./settings.conf'
|
|
audibleBookPath=sys.argv[1]
|
|
serverURL=sys.argv[2]
|
|
targetDir='./audiobooks'
|
|
|
|
# Check if config file exists and creates one if needed.
|
|
if not os.path.exists(configFilePath):
|
|
print('Creating config file')
|
|
createConfig(configFilePath, audibleBookPath)
|
|
|
|
configString = '[Settings]\n' + open(configFilePath).read()
|
|
configParser = configparser.RawConfigParser()
|
|
configParser.read_string(configString)
|
|
activationBytes=configParser.get('Settings', 'activationBytes')
|
|
|
|
book = AudibleBook(audibleBookPath, activationBytes, targetDir)
|
|
|
|
name = osSafeName(book.getTitle())
|
|
outputDir = '{}/{}'.format(targetDir, name)
|
|
if not os.path.exists(outputDir):
|
|
os.makedirs(outputDir)
|
|
book.setTargetDir(outputDir)
|
|
|
|
rssFilePath='{}/rss.xml'.format(outputDir)
|
|
|
|
if not os.path.exists(rssFilePath):
|
|
book.removeDRM(name)
|
|
book.splitChapters()
|
|
book.getCover()
|
|
|
|
fg = FeedGenerator()
|
|
fg.id(name)
|
|
fg.title(book.getTitle())
|
|
fg.author( {'name': book.getAuthor()} )
|
|
fg.link( href='https://audible.com', rel='alternate' )
|
|
fg.logo('{}/{}/cover.jpg'.format(serverURL, name))
|
|
fg.subtitle(book.getMetadata()['copyright'])
|
|
fg.language('en')
|
|
|
|
count = 0
|
|
for c in book.getChapters():
|
|
key = osSafeName(c['tags']['title'])
|
|
fe = fg.add_entry()
|
|
fe.id(key)
|
|
fe.title(c['tags']['title'])
|
|
fe.link(href='{0}/{1}/{1}-{2}.m4a'.format(serverURL, name, key))
|
|
fe.enclosure('{0}/{1}/{1}-{2}.m4a'.format(serverURL, name, key), 0, 'audio')
|
|
fe.published(datetime.datetime(int(book.getMetadata()['date']), 6, 1, 0, count, tzinfo=datetime.timezone.utc))
|
|
count += 1
|
|
|
|
rssfeed = fg.rss_str(pretty=True)
|
|
fg.rss_file(rssFilePath)
|
|
book.deleteConverted()
|