podcast-dl-server.py

from __future__ import unicode_literals
import json
import os
import subprocess
from queue import Queue
from bottle import route, run, Bottle, request, static_file
from threading import Thread
from pathlib import Path
from collections import ChainMap

app = Bottle()


app_defaults = {
    # 'YDL_FORMAT': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]',
    # 'YDL_EXTRACT_AUDIO_FORMAT': None,
    # 'YDL_EXTRACT_AUDIO_QUALITY': '192',
    # 'YDL_RECODE_VIDEO_FORMAT': None,
    # 'YDL_OUTPUT_TEMPLATE': '/podcast-dl/%(title)s [%(id)s].%(ext)s',
    # 'YDL_ARCHIVE_FILE': None,
    'PDL_SERVER_HOST': '0.0.0.0',
    'PDL_SERVER_PORT': 8567,
}


@app.route('/podcast-dl')
def dl_queue_list():
    return static_file('index.html', root='./')


@app.route('/podcast-dl/static/:filename#.*#')
def server_static(filename):
    return static_file(filename, root='./static')


@app.route('/podcast-dl/q', method='GET')
def q_size():
    return {"success": True, "size": json.dumps(list(dl_q.queue))}


@app.route('/podcast-dl/q', method='POST')
def q_put():
    url = request.forms.get("url")
    options = {
        'format': request.forms.get("format")
    }

    if not url:
        return {"success": False, "error": "/q called without a 'url' query param"}

    dl_q.put((url, options))
    print("Added url " + url + " to the download queue")
    return {"success": True, "url": url, "options": options}

@app.route("/podcast-dl/update", method="GET")
def update():
    command = ["npm", "update", "-g", "podcast-dl"]
    proc = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

    output, error = proc.communicate()
    return {
        "output": output.decode('ascii'),
        "error":  error.decode('ascii')
    }

def dl_worker():
    while not done:
        url, options = dl_q.get()
        download(url, options)
        dl_q.task_done()


# def get_ydl_options(request_options):
#     request_vars = {
#         'YDL_EXTRACT_AUDIO_FORMAT': None,
#         'YDL_RECODE_VIDEO_FORMAT': None,
#     }

#     requested_format = request_options.get('format', 'bestvideo')

#     if requested_format in ['aac', 'flac', 'mp3', 'm4a', 'opus', 'vorbis', 'wav']:
#         request_vars['YDL_EXTRACT_AUDIO_FORMAT'] = requested_format
#     elif requested_format == 'bestaudio':
#         request_vars['YDL_EXTRACT_AUDIO_FORMAT'] = 'best'
#     elif requested_format in ['mp4', 'flv', 'webm', 'ogg', 'mkv', 'avi']:
#         request_vars['YDL_RECODE_VIDEO_FORMAT'] = requested_format

#     ydl_vars = ChainMap(request_vars, os.environ, app_defaults)

#     postprocessors = []

#     if(ydl_vars['YDL_EXTRACT_AUDIO_FORMAT']):
#         postprocessors.append({
#             'key': 'FFmpegExtractAudio',
#             'preferredcodec': ydl_vars['YDL_EXTRACT_AUDIO_FORMAT'],
#             'preferredquality': ydl_vars['YDL_EXTRACT_AUDIO_QUALITY'],
#         })

#     if(ydl_vars['YDL_RECODE_VIDEO_FORMAT']):
#         postprocessors.append({
#             'key': 'FFmpegVideoConvertor',
#             'preferedformat': ydl_vars['YDL_RECODE_VIDEO_FORMAT'],
#         })

#     return {
#         'format': ydl_vars['YDL_FORMAT'],
#         'postprocessors': postprocessors,
#         'outtmpl': ydl_vars['YDL_OUTPUT_TEMPLATE'],
#         'download_archive': ydl_vars['YDL_ARCHIVE_FILE']
#     }


def download(url, request_options):
    os.system('podcast-dl --archive archive.log --out-dir /downloads --url '+url)

dl_q = Queue()
done = False
dl_thread = Thread(target=dl_worker)
dl_thread.start()

print("Updating podcast-dl to the newest version")
updateResult = update()
print(updateResult["output"])
print(updateResult["error"])

print("Started download thread")

app_vars = ChainMap(os.environ, app_defaults)

app.run(host=app_vars['PDL_SERVER_HOST'], port=app_vars['PDL_SERVER_PORT'], debug=True)
done = True
dl_thread.join()