import datetime import flask; from flask import request import itertools import traceback import ycdl from .. import common from .. import jsonify site = common.site @site.route('/channels') def get_channels(): channels = common.ycdldb.get_channels() return flask.render_template('channels.html', channels=channels) @site.route('/videos') @site.route('/watch') @site.route('/videos/') @site.route('/channel/') @site.route('/channel//') def get_channel(channel_id=None, download_filter=None): if channel_id is not None: try: common.ycdldb.add_channel(channel_id) except Exception: traceback.print_exc() try: channel = common.ycdldb.get_channel(channel_id) except ycdl.exceptions.NoSuchChannel: flask.abort(404) else: channel = None orderby = request.args.get('orderby', None) video_id = request.args.get('v', '') if video_id: common.ycdldb.insert_video(video_id) videos = [common.ycdldb.get_video(video_id)] else: videos = common.ycdldb.get_videos( channel_id=channel_id, download_filter=download_filter, orderby=orderby, ) search_terms = request.args.get('q', '').lower().strip().replace('+', ' ').split() if search_terms: videos = (v for v in videos if all(term in v.title.lower() for term in search_terms)) limit = request.args.get('limit', None) if limit is not None: try: limit = int(limit) videos = itertools.islice(videos, limit) except ValueError: pass videos = list(videos) for video in videos: published = video.published published = datetime.datetime.utcfromtimestamp(published) published = published.strftime('%Y %m %d') video._published_str = published all_states = common.ycdldb.get_all_states() return flask.render_template( 'channel.html', all_states=all_states, channel=channel, download_filter=download_filter, query_string='?' + request.query_string.decode('utf-8'), videos=videos, ) @site.route('/add_channel', methods=['POST']) def post_add_channel(): channel_id = request.form.get('channel_id', '') 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 jsonify.make_json_response({}, status=404) channel = common.ycdldb.add_channel(channel_id, get_videos=True) return jsonify.make_json_response(ycdl.jsonify.channel(channel)) @site.route('/channel//delete', methods=['POST']) def post_delete_channel(channel_id): try: channel = common.ycdldb.get_channel(channel_id) except ycdl.exceptions.NoSuchChannel as exc: return jsonify.make_json_response(ycdl.jsonify.exception(exc), status=404) channel.delete() return jsonify.make_json_response({}) @site.route('/channel//refresh', methods=['POST']) def post_refresh_channel(channel_id): force = request.form.get('force', False) force = ycdl.helpers.truthystring(force) try: channel = common.ycdldb.get_channel(channel_id) except ycdl.exceptions.NoSuchChannel as exc: return jsonify.make_json_response(ycdl.jsonify.exception(exc), status=404) channel.refresh(force=force) return jsonify.make_json_response(ycdl.jsonify.channel(channel)) @site.route('/refresh_all_channels', methods=['POST']) def post_refresh_all_channels(): force = request.form.get('force', False) force = ycdl.helpers.truthystring(force) common.ycdldb.refresh_all_channels(force=force) return jsonify.make_json_response({}) @site.route('/channel//set_automark', methods=['POST']) def post_set_automark(channel_id): state = request.form['state'] channel = common.ycdldb.get_channel(channel_id) try: channel.set_automark(state) except ycdl.exceptions.InvalidVideoState: flask.abort(400) return jsonify.make_json_response({})