irc-books/bookworm/web.py

from flask import Flask, render_template, make_response, request
from bookworm import s3, constants
import pg_simple
import json
import redis
import os
import re

s3client = s3.client()
app = Flask(__name__)
pool = pg_simple.config_pool(dsn='dbname=david user=david')
db = pg_simple.PgSimple(pool)
r = redis.StrictRedis(host='localhost', port=6379, decode_responses=True)

@app.route('/books/status', methods=['GET'])
def status_books():
    keys = r.keys('book_*')
    ret = {}
    for key in keys:
        val = r.hgetall(key)
        ret[key] = val
    print(ret)
    return json.dumps(ret)

def clean_book_name(book):
    cbr = re.compile(r'epub|azw3|mobi|\(v[0-9.]+\)')
    return cbr.sub('', book).rstrip('() .-')

@app.route('/')
def index():
    objects = s3client.list_objects_v2(Bucket=constants.PROCESSED_FILE_BUCKET)['Contents']
    objects = sorted(objects, key=lambda x: x['LastModified'], reverse=True)
    books = [(obj['Key'], clean_book_name(obj['Key'])) for obj in objects]
    return render_template('kindle-index.j2', books=books)

@app.route('/book/<path:book>')
def serve_books(book):
    obj = s3client.get_object(Bucket=constants.PROCESSED_FILE_BUCKET, Key=book)
    data = obj['Body'].read()
    response = make_response(data)
    response.headers.set('Content-Disposition', f'attachment; filename="{book}"')
    response.headers.set('Content-Length', len(data))

    _, ext = os.path.splitext(book)
    if ext == '.mobi':
        content_type = 'application/x-mobipocket-ebook'
    elif ext == '.azw3':
        content_type = 'application/x-mobi8-ebook'
    else:
        content_type = 'text/plain'

    response.headers.set('Content-Type', content_type)
    return response

@app.route('/book/search', methods=['GET'])
def search_books():
    terms = request.args.get('terms').split()
    terms = [f'%{term}%' for term in terms]

    books = db.fetchall('books',
                        fields=['bot', 'book'],
                        where=('lower(book) like all(%s)', [terms]),
                        limit=100)
    return json.dumps([b._asdict() for b in books])

@app.route('/book/fetch', methods=['POST'])
def fetch_books():
    fetch = request.json
    data = {'bot': fetch['bot'],
            'book': fetch['book'],
            'meta': {
                'unpack_file_queue': constants.REDIS_UNPACK_FILE,
                'fetch_file_queue': constants.REDIS_FETCH_FILE,
                'raw_file_bucket': constants.RAW_FILE_BUCKET,
                'processed_file_bucket': constants.PROCESSED_FILE_BUCKET,
            }
           }
    r.rpush(constants.REDIS_BOOK_COMMANDS, json.dumps(data))
    return ''


def main():
    app.run(host='0.0.0.0', debug=True)

if __name__ == '__main__':
    main()