youtube-dl/plugin.py

# -*- coding: utf-8 -*-
#########################################################
# python
import os
import traceback

# third-party
from flask import Blueprint, request, render_template, redirect, jsonify, abort
from flask_login import login_required

# sjva 공용
from framework.logger import get_logger
from framework import db
from framework.util import Util

# 로그
package_name = __name__.split('.')[0]
logger = get_logger(package_name)

# 패키지
from .logic import Logic
from .model import ModelSetting
from .my_youtube_dl import Youtube_dl

#########################################################

blueprint = Blueprint(package_name, package_name, url_prefix='/%s' % package_name, template_folder=os.path.join(os.path.dirname(__file__), 'templates'))

def plugin_load():
	Logic.plugin_load()

def plugin_unload():
	Logic.plugin_unload()

plugin_info = {
	'version': '1.2.3',
	'name': 'youtube-dl',
	'category_name': 'vod',
	'icon': '',
	'developer': 'joyfuI',
	'description': '유튜브, 네이버TV 등 동영상 사이트에서 동영상 다운로드',
	'home': 'https://github.com/joyfuI/youtube-dl',
	'more': ''
}

# 메뉴 구성
menu = {
	'main': [package_name, 'youtube-dl'],
	'sub': [
		['setting', '설정'], ['download', '다운로드'], ['list', '목록'], ['log', '로그']
	],
	'category': 'vod'
}

#########################################################
# WEB Menu
#########################################################
@blueprint.route('/')
def home():
	return redirect('/%s/list' % package_name)

@blueprint.route('/<sub>')
@login_required
def detail(sub):
	try:
		if sub == 'setting':
			setting_list = db.session.query(ModelSetting).all()
			arg = Util.db_list_to_dict(setting_list)
			arg['package_name'] = package_name
			arg['youtube_dl_version'] = Youtube_dl.get_version()
			return render_template('%s_setting.html' % package_name, arg=arg)

		elif sub == 'download':
			arg = { }
			arg['package_name'] = package_name
			arg['file_name'] = '%(title)s-%(id)s.%(ext)s'
			arg['preset_list'] = Logic.get_preset_list()
			return render_template('%s_download.html' % package_name, arg=arg)

		elif sub == 'list':
			arg = { }
			arg['package_name'] = package_name
			return render_template('%s_list.html' % package_name, arg=arg)

		elif sub == 'log':
			return render_template('log.html', package=package_name)
	except Exception as e:
		logger.error('Exception:%s', e)
		logger.error(traceback.format_exc())
	return render_template('sample.html', title='%s - %s' % (package_name, sub))

#########################################################
# For UI
#########################################################
@blueprint.route('/ajax/<sub>', methods=['POST'])
@login_required
def ajax(sub):
	logger.debug('AJAX %s %s', package_name, sub)
	try:
		if sub == 'setting_save':
			ret = Logic.setting_save(request)
			return jsonify(ret)

		elif sub == 'download':
			url = request.form['url']
			filename = request.form['filename']
			temp_path = Logic.get_setting_value('temp_path')
			save_path = Logic.get_setting_value('save_path')
			format_code = request.form['format'] if request.form['format'] else None
			youtube_dl = Youtube_dl(package_name, url, filename, temp_path, save_path, format_code)
			Logic.youtube_dl_list.append(youtube_dl)	# 리스트 추가
			youtube_dl.start()
			return jsonify([])

		elif sub == 'list':
			ret = []
			for i in Logic.youtube_dl_list:
				data = Logic.get_data(i)
				if data is not None:
					ret.append(data)
			return jsonify(ret)

		elif sub == 'stop':
			index = int(request.form['index'])
			Logic.youtube_dl_list[index].stop()
			return jsonify([])
	except Exception as e:
		logger.error('Exception:%s', e)
		logger.error(traceback.format_exc())

#########################################################
# API
#########################################################
# API 명세는 https://github.com/joyfuI/youtube-dl#api
@blueprint.route('/api/<sub>', methods=['GET', 'POST'])
def api(sub):
	plugin = request.form.get('plugin')
	logger.debug('API %s %s: %s', package_name, sub, plugin)
	if not plugin:	# 요청한 플러그인명이 빈문자열이거나 None면
		abort(403)	# 403 에러(거부)
	try:
		# 동영상 정보를 반환하는 API
		if sub == 'info_dict':
			url = request.form.get('url')
			ret = {
				'errorCode': 0,
				'info_dict': None
			}
			if None == url:
				return Logic.abort(ret, 1)	# 필수 요청 변수가 없음
			if not url.startswith('http'):
				return Logic.abort(ret, 2)	# 잘못된 동영상 주소
			info_dict = Youtube_dl.get_info_dict(url)
			if info_dict is None:
				return Logic.abort(ret, 10)	# 실패
			ret['info_dict'] = info_dict
			return jsonify(ret)

		# 다운로드 준비를 요청하는 API
		elif sub == 'download':
			key = request.form.get('key')
			url = request.form.get('url')
			filename = request.form.get('filename')
			temp_path = request.form.get('temp_path')
			save_path = request.form.get('save_path')
			format_code = request.form.get('format_code', None)
			start = request.form.get('start', False)
			ret = {
				'errorCode': 0,
				'index': None
			}
			if None in (key, url, filename, temp_path, save_path):
				return Logic.abort(ret, 1)	# 필수 요청 변수가 없음
			if not url.startswith('http'):
				return Logic.abort(ret, 2)	# 잘못된 동영상 주소
			youtube_dl = Youtube_dl(plugin, url, filename, temp_path, save_path, format_code)
			youtube_dl._key = key
			Logic.youtube_dl_list.append(youtube_dl)	# 리스트 추가
			ret['index'] = youtube_dl.index
			if start:
				youtube_dl.start()
			return jsonify(ret)

		# 다운로드 시작을 요청하는 API
		elif sub == 'start':
			index = request.form.get('index')
			key = request.form.get('key')
			ret = {
				'errorCode': 0,
				'status': None
			}
			if None in (index, key):
				return Logic.abort(ret, 1)	# 필수 요청 변수가 없음
			index = int(index)
			if not (0 <= index and index < Youtube_dl._index):
				return Logic.abort(ret, 3)	# 인덱스 범위를 벗어남
			youtube_dl = Logic.youtube_dl_list[index]
			if youtube_dl._key != key:
				return Logic.abort(ret, 4)	# 키가 일치하지 않음
			ret['status'] = youtube_dl.status.name
			if not youtube_dl.start():
				return Logic.abort(ret, 10)	# 실패
			return jsonify(ret)

		# 다운로드 중지를 요청하는 API
		elif sub == 'stop':
			index = request.form.get('index')
			key = request.form.get('key')
			ret = {
				'errorCode': 0,
				'status': None
			}
			if None in (index, key):
				return Logic.abort(ret, 1)	# 필수 요청 변수가 없음
			index = int(index)
			if not (0 <= index and index < Youtube_dl._index):
				return Logic.abort(ret, 3)	# 인덱스 범위를 벗어남
			youtube_dl = Logic.youtube_dl_list[index]
			if youtube_dl._key != key:
				return Logic.abort(ret, 4)	# 키가 일치하지 않음
			ret['status'] = youtube_dl.status.name
			if not youtube_dl.stop():
				return Logic.abort(ret, 10)	# 실패
			return jsonify(ret)

		# 현재 상태를 반환하는 API
		elif sub == 'status':
			index = request.form.get('index')
			key = request.form.get('key')
			ret = {
				'errorCode': 0,
				'status': None,
				'start_time': None,
				'end_time': None,
				'temp_path': None,
				'save_path': None
			}
			if None in (index, key):
				return Logic.abort(ret, 1)	# 필수 요청 변수가 없음
			index = int(index)
			if not (0 <= index and index < Youtube_dl._index):
				return Logic.abort(ret, 3)	# 인덱스 범위를 벗어남
			youtube_dl = Logic.youtube_dl_list[index]
			if youtube_dl._key != key:
				return Logic.abort(ret, 4)	# 키가 일치하지 않음
			ret['status'] = youtube_dl.status.name
			ret['start_time'] = youtube_dl.start_time.strftime('%Y %m %d %H %M %S') if youtube_dl.start_time is not None else None
			ret['end_time'] = youtube_dl.end_time.strftime('%Y %m %d %H %M %S') if youtube_dl.end_time is not None else None
			ret['temp_path'] = youtube_dl.temp_path
			ret['save_path'] = youtube_dl.save_path
			return jsonify(ret)
	except Exception as e:
		logger.error('Exception:%s', e)
		logger.error(traceback.format_exc())
		abort(500)	# 500 에러(서버 오류)
	abort(404)	# 404 에러(페이지 없음)