ycdl/frontends/ycdl_flask/backend/endpoints/channel_endpoints.py

203 lines
6.6 KiB
Python
Raw Normal View History

2020-03-29 00:05:43 +00:00
import flask; from flask import request
import itertools
2020-03-29 00:05:43 +00:00
from voussoirkit import flasktools
2021-09-02 06:37:43 +00:00
from voussoirkit import pathclass
2021-09-05 08:22:21 +00:00
from voussoirkit import stringtools
2020-03-29 00:05:43 +00:00
import ycdl
from .. import common
2020-03-29 00:05:43 +00:00
site = common.site
def _get_or_insert_video(video_id):
try:
video = common.ycdldb.get_video(video_id)
except ycdl.exceptions.NoSuchVideo:
video = common.ycdldb.insert_video(video_id)['video']
return video
@site.route('/all_channels.json')
def get_all_channel_names():
all_channels = {channel.id: channel.name for channel in common.ycdldb.get_channels()}
response = {'channels': all_channels}
return flasktools.json_response(response)
2020-03-29 00:05:43 +00:00
@site.route('/channels')
def get_channels():
channels = common.ycdldb.get_channels()
return flask.render_template('channels.html', channels=channels)
def _render_videos_listing(videos, channel, state, orderby):
2020-03-29 00:05:43 +00:00
search_terms = request.args.get('q', '').lower().strip().replace('+', ' ').split()
if search_terms:
lowered = ((video, video.title.lower()) for video in videos)
videos = (
video for (video, title) in lowered
if all(term in title for term in search_terms)
)
2020-03-29 00:05:43 +00:00
limit = request.args.get('limit', None)
if limit is not None:
try:
limit = int(limit)
videos = itertools.islice(videos, limit)
2020-03-29 00:05:43 +00:00
except ValueError:
pass
if not isinstance(videos, list):
videos = list(videos)
2020-03-29 00:05:43 +00:00
all_states = common.ycdldb.get_all_states()
return flask.render_template(
'channel.html',
all_states=all_states,
channel=channel,
state=state,
orderby=orderby,
2020-03-29 00:05:43 +00:00
videos=videos,
)
@site.route('/channel/<channel_id>')
@site.route('/channel/<channel_id>/<state>')
def get_channel(channel_id, state=None):
try:
channel = common.ycdldb.add_channel(channel_id, commit=True)
except ycdl.ytapi.ChannelNotFound:
flask.abort(404)
orderby = request.args.get('orderby', None)
videos = common.ycdldb.get_videos(
channel_id=channel.id,
orderby=orderby,
state=state,
)
return _render_videos_listing(videos, channel=channel, state=state, orderby=orderby)
@site.route('/videos')
@site.route('/videos/<state>')
def get_videos(state=None):
orderby = request.args.get('orderby', None)
videos = common.ycdldb.get_videos(
orderby=orderby,
state=state,
)
return _render_videos_listing(videos, channel=None, state=state, orderby=orderby)
@site.route('/watch')
def get_watch():
video_id = request.args.get('v', '')
if not video_id:
return flask.redirect('/')
try:
video = _get_or_insert_video(video_id)
except ycdl.ytapi.VideoNotFound:
flask.abort(404)
videos = [video]
return _render_videos_listing(videos, channel=None, state=None, orderby=None)
@site.route('/add_channel', methods=['POST'])
def post_add_channel():
channel_id = request.form.get('channel_id', '')
2020-03-29 00:05:43 +00:00
channel_id = channel_id.strip()
if not channel_id:
flask.abort(400)
if not (len(channel_id) == 24 and channel_id.startswith('UC')):
# It seems they have given us a username instead.
try:
channel_id = common.ycdldb.youtube.get_user_id(username=channel_id)
except ycdl.ytapi.ChannelNotFound:
return flasktools.json_response({}, status=404)
2020-03-29 00:05:43 +00:00
2021-10-16 04:00:04 +00:00
channel = common.ycdldb.add_channel(channel_id, get_videos=True, commit=True)
return flasktools.json_response(channel.jsonify())
@site.route('/channel/<channel_id>/delete', methods=['POST'])
def post_delete_channel(channel_id):
try:
channel = common.ycdldb.get_channel(channel_id)
except ycdl.exceptions.NoSuchChannel as exc:
return flasktools.json_response(exc.jsonify(), status=404)
2021-10-16 04:00:04 +00:00
channel.delete(commit=True)
return flasktools.json_response({})
@site.route('/channel/<channel_id>/refresh', methods=['POST'])
def post_refresh_channel(channel_id):
2020-03-29 00:05:43 +00:00
force = request.form.get('force', False)
2021-09-05 08:22:21 +00:00
force = stringtools.truthystring(force, False)
try:
channel = common.ycdldb.get_channel(channel_id)
except ycdl.exceptions.NoSuchChannel as exc:
return flasktools.json_response(exc.jsonify(), status=404)
2021-10-16 04:00:04 +00:00
channel.refresh(force=force, commit=True)
return flasktools.json_response(channel.jsonify())
2020-03-29 00:05:43 +00:00
@site.route('/refresh_all_channels', methods=['POST'])
def post_refresh_all_channels():
force = request.form.get('force', False)
2021-09-05 08:22:21 +00:00
force = stringtools.truthystring(force, False)
2021-10-16 04:00:04 +00:00
common.ycdldb.refresh_all_channels(force=force, skip_failures=True, commit=True)
return flasktools.json_response({})
@site.route('/channel/<channel_id>/set_automark', methods=['POST'])
def post_set_automark(channel_id):
state = request.form['state']
channel = common.ycdldb.get_channel(channel_id)
try:
2021-10-16 04:00:04 +00:00
channel.set_automark(state, commit=True)
except ycdl.exceptions.InvalidVideoState:
flask.abort(400)
return flasktools.json_response({})
@flasktools.required_fields(['autorefresh'], forbid_whitespace=True)
@site.route('/channel/<channel_id>/set_autorefresh', methods=['POST'])
def post_set_autorefresh(channel_id):
autorefresh = request.form['autorefresh']
channel = common.ycdldb.get_channel(channel_id)
try:
autorefresh = stringtools.truthystring(autorefresh)
2021-10-16 04:00:04 +00:00
channel.set_autorefresh(autorefresh, commit=True)
except (ValueError, TypeError):
flask.abort(400)
return flasktools.json_response({})
2021-09-02 06:37:43 +00:00
@site.route('/channel/<channel_id>/set_download_directory', methods=['POST'])
def post_set_download_directory(channel_id):
download_directory = request.form['download_directory']
channel = common.ycdldb.get_channel(channel_id)
try:
2021-10-16 04:00:04 +00:00
channel.set_download_directory(download_directory, commit=True)
2021-09-02 06:37:43 +00:00
except pathclass.NotDirectory:
exc = {
'error_type': 'NOT_DIRECTORY',
'error_message': f'"{download_directory}" is not a directory.',
}
return flasktools.json_response(exc, status=400)
2021-09-02 06:37:43 +00:00
abspath = channel.download_directory.absolute_path if channel.download_directory else None
response = {'download_directory': abspath}
return flasktools.json_response(response)
2021-09-02 06:37:43 +00:00
@site.route('/channel/<channel_id>/set_queuefile_extension', methods=['POST'])
def post_set_queuefile_extension(channel_id):
extension = request.form['extension']
channel = common.ycdldb.get_channel(channel_id)
2021-10-16 04:00:04 +00:00
channel.set_queuefile_extension(extension, commit=True)
response = {'queuefile_extension': channel.queuefile_extension}
return flasktools.json_response(response)