1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
#!/usr/bin/env python3
import argparse
import os
import shutil
import json
from flask import Flask, request, Response, render_template, send_file, templating
from flask.globals import _app_ctx_stack
from lib.books import Books
from lib.util import check_auth, escape, generate_rss, read_cache
abs_path = os.path.dirname(os.path.abspath(__file__))
app = Flask(__name__)
config_path = os.path.join(abs_path, 'app.cfg')
config_exists = os.path.exists(config_path)
if config_exists or __name__.startswith('uwsgi'):
app.config.from_pyfile(config_path)
cache_path = os.path.join(abs_path, 'cache')
json_path = os.path.join(cache_path, 'audiobooks.json')
@app.route('/')
def list_books():
'''
Book listing and audiobook RSS/file download
:a: audiobook hash; if provided without :f: (file) return RSS
:f: file hash; requires associated audiobook (:a:) to download
Listing of audiobooks returned if no params provided
'''
books = read_cache(json_path)
a = request.args.get('a') # audiobook hash
f = request.args.get('f') # file hash
# audiobook and file parameters provided: serve up file
if a and f:
if not books.get(a) or not books[a]['files'].get(f):
return 'book or file not found', 404
f_path = books[a]['files'][f]['path']
return send_file(f_path, conditional=True)
# serve up audiobook RSS feed; only audiobook hash provided
elif a:
if not books.get(a):
return 'book not found', 404
rss = generate_rss(request.base_url, a, books)
return Response(rss, mimetype='text/xml')
else:
auth = request.authorization
if not auth or not check_auth(app, auth.username, auth.password):
form = {'WWW-Authenticate': 'Basic realm="o/"'}
return Response('unauthorized', 401, form)
return render_template('index.html', books=books,
show_path=app.config.get('SHOW_PATH', True))
def generate(static_path, base_url, audiobook_dirs):
static_index_path = os.path.join(static_path, 'index.html')
books = Books()
books.scan_books(audiobook_dirs)
books.write_cache()
books = read_cache(json_path)
# A bit of a hack, but push to the app context stack so we can render a
# template outside of a Flask request
_app_ctx_stack.push(app.app_context())
index = render_template('index.html', books=books, static=True)
_app_ctx_stack.pop()
os.makedirs(static_path, exist_ok=True)
indexfile = open(static_index_path, 'w')
indexfile.write(index)
indexfile.close()
for b_key, book in books.items():
rss = generate_rss(base_url, b_key, books, static=True)
rss_path = os.path.join(static_path, b_key + '.xml')
rssfile = open(rss_path, 'w')
rssfile.write(rss.decode('utf-8'))
rssfile.close()
book_dir = os.path.join(static_path, b_key)
os.makedirs(book_dir, exist_ok=True)
for f_key, file in book['files'].items():
f_path = file['path']
copy_path = os.path.join(book_dir, f_key + '.mp3')
if not os.path.exists(copy_path):
shutil.copyfile(f_path, copy_path)
if __name__ == '__main__':
desc = 'roka: listen to audiobooks with podcast apps via RSS'
parser = argparse.ArgumentParser(description=desc)
parser.add_argument('--scan', dest='scan', action='store_true',
help='scan audiobooks directory for new books',
required=False)
parser.add_argument('--generate', dest='static_path', type=str, action='store',
help='Output directory to generate static files',
required=False)
parser.add_argument('--config', dest='config', type=str, action='store',
help='Json configuration instead of app.cfg',
required=False)
args = parser.parse_args()
if args.config:
class objectview(object):
def __init__(self, d):
self.__dict__ = d
config = objectview(json.loads(args.config))
# override app.cfg
app.config.from_object(config)
elif not config_exists:
raise Exception(f"Config file '{config_path}' doesn't exist")
root_path = os.path.expanduser(app.config['ROOT_PATH'])
if args.scan:
books = Books()
books.scan_books(root_path)
books.write_cache()
elif args.static_path:
generate(args.static_path, app.config['BASE_URL'], root_path)
else:
app.run(host='127.0.0.1', port='8085', threaded=True)
|
