Move endpoints to new endpoints.py.
This commit is contained in:
parent
1b2778ca7a
commit
4effa64694
3 changed files with 154 additions and 153 deletions
|
|
@ -1,3 +1,4 @@
|
|||
from . import common
|
||||
from . import endpoints
|
||||
|
||||
site = common.site
|
||||
|
|
|
|||
|
|
@ -5,22 +5,18 @@ Use ycdl_launch.py to start the server with gevent.
|
|||
import logging
|
||||
logging.getLogger('googleapicliet.discovery_cache').setLevel(logging.ERROR)
|
||||
|
||||
import datetime
|
||||
import flask; from flask import request
|
||||
import json
|
||||
import mimetypes
|
||||
import os
|
||||
import threading
|
||||
import time
|
||||
import traceback
|
||||
|
||||
from voussoirkit import pathclass
|
||||
|
||||
import bot
|
||||
import ycdl
|
||||
|
||||
from voussoirkit import pathclass
|
||||
|
||||
from . import jinja_filters
|
||||
from . import jsonify
|
||||
|
||||
root_dir = pathclass.Path(__file__).parent.parent
|
||||
|
||||
|
|
@ -118,153 +114,6 @@ def send_file(filepath):
|
|||
####################################################################################################
|
||||
####################################################################################################
|
||||
|
||||
@site.route('/')
|
||||
def root():
|
||||
return flask.render_template('root.html')
|
||||
|
||||
@site.route('/favicon.ico')
|
||||
@site.route('/favicon.png')
|
||||
def favicon():
|
||||
return flask.send_file(FAVICON_PATH.absolute_path)
|
||||
|
||||
|
||||
@site.route('/channels')
|
||||
def get_channels():
|
||||
channels = ycdldb.get_channels()
|
||||
for channel in channels:
|
||||
channel['has_pending'] = ycdldb.channel_has_pending(channel['id'])
|
||||
return flask.render_template('channels.html', channels=channels)
|
||||
|
||||
@site.route('/videos')
|
||||
@site.route('/watch')
|
||||
@site.route('/videos/<download_filter>')
|
||||
@site.route('/channel/<channel_id>')
|
||||
@site.route('/channel/<channel_id>/<download_filter>')
|
||||
def get_channel(channel_id=None, download_filter=None):
|
||||
if channel_id is not None:
|
||||
try:
|
||||
ycdldb.add_channel(channel_id)
|
||||
except Exception:
|
||||
traceback.print_exc()
|
||||
channel = ycdldb.get_channel(channel_id)
|
||||
else:
|
||||
channel = None
|
||||
|
||||
orderby = request.args.get('orderby', None)
|
||||
|
||||
video_id = request.args.get('v', '')
|
||||
if video_id:
|
||||
ycdldb.insert_video(video_id)
|
||||
videos = [ycdldb.get_video(video_id)]
|
||||
else:
|
||||
videos = 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 = videos[:limit]
|
||||
except ValueError:
|
||||
pass
|
||||
|
||||
for video in videos:
|
||||
published = video['published']
|
||||
published = datetime.datetime.utcfromtimestamp(published)
|
||||
published = published.strftime('%Y %m %d')
|
||||
video['_published_str'] = published
|
||||
|
||||
all_states = 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('/mark_video_state', methods=['POST'])
|
||||
def post_mark_video_state():
|
||||
if 'video_ids' not in request.form or 'state' not in request.form:
|
||||
flask.abort(400)
|
||||
video_ids = request.form['video_ids']
|
||||
state = request.form['state']
|
||||
try:
|
||||
video_ids = video_ids.split(',')
|
||||
for video_id in video_ids:
|
||||
ycdldb.mark_video_state(video_id, state, commit=False)
|
||||
ycdldb.sql.commit()
|
||||
|
||||
except ycdl.exceptions.NoSuchVideo:
|
||||
ycdldb.rollback()
|
||||
traceback.print_exc()
|
||||
flask.abort(404)
|
||||
|
||||
except ycdl.exceptions.InvalidVideoState:
|
||||
ycdldb.rollback()
|
||||
flask.abort(400)
|
||||
|
||||
return jsonify.make_json_response({'video_ids': video_ids, 'state': state})
|
||||
|
||||
@site.route('/refresh_all_channels', methods=['POST'])
|
||||
def post_refresh_all_channels():
|
||||
force = request.form.get('force', False)
|
||||
force = ycdl.helpers.truthystring(force)
|
||||
ycdldb.refresh_all_channels(force=force)
|
||||
return jsonify.make_json_response({})
|
||||
|
||||
@site.route('/refresh_channel', methods=['POST'])
|
||||
def post_refresh_channel():
|
||||
if 'channel_id' not in request.form:
|
||||
flask.abort(400)
|
||||
channel_id = request.form['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 = ycdldb.youtube.get_user_id(username=channel_id)
|
||||
except IndexError:
|
||||
flask.abort(404)
|
||||
|
||||
force = request.form.get('force', False)
|
||||
force = ycdl.helpers.truthystring(force)
|
||||
ycdldb.add_channel(channel_id, commit=False)
|
||||
ycdldb.refresh_channel(channel_id, force=force)
|
||||
return jsonify.make_json_response({})
|
||||
|
||||
@site.route('/start_download', methods=['POST'])
|
||||
def post_start_download():
|
||||
if 'video_ids' not in request.form:
|
||||
flask.abort(400)
|
||||
video_ids = request.form['video_ids']
|
||||
try:
|
||||
video_ids = video_ids.split(',')
|
||||
for video_id in video_ids:
|
||||
ycdldb.download_video(video_id, commit=False)
|
||||
ycdldb.sql.commit()
|
||||
|
||||
except ycdl.ytapi.VideoNotFound:
|
||||
ycdldb.rollback()
|
||||
flask.abort(404)
|
||||
|
||||
return jsonify.make_json_response({'video_ids': video_ids, 'state': 'downloaded'})
|
||||
|
||||
####################################################################################################
|
||||
####################################################################################################
|
||||
####################################################################################################
|
||||
####################################################################################################
|
||||
|
||||
def refresher_thread(rate):
|
||||
while True:
|
||||
time.sleep(rate)
|
||||
|
|
|
|||
151
frontends/ycdl_flask/backend/endpoints.py
Normal file
151
frontends/ycdl_flask/backend/endpoints.py
Normal file
|
|
@ -0,0 +1,151 @@
|
|||
import datetime
|
||||
import flask; from flask import request
|
||||
import traceback
|
||||
|
||||
import ycdl
|
||||
|
||||
from . import common
|
||||
from . import jsonify
|
||||
|
||||
site = common.site
|
||||
|
||||
@site.route('/')
|
||||
def root():
|
||||
return flask.render_template('root.html')
|
||||
|
||||
@site.route('/favicon.ico')
|
||||
@site.route('/favicon.png')
|
||||
def favicon():
|
||||
return flask.send_file(common.FAVICON_PATH.absolute_path)
|
||||
|
||||
@site.route('/channels')
|
||||
def get_channels():
|
||||
channels = common.ycdldb.get_channels()
|
||||
for channel in channels:
|
||||
channel['has_pending'] = common.ycdldb.channel_has_pending(channel['id'])
|
||||
return flask.render_template('channels.html', channels=channels)
|
||||
|
||||
@site.route('/videos')
|
||||
@site.route('/watch')
|
||||
@site.route('/videos/<download_filter>')
|
||||
@site.route('/channel/<channel_id>')
|
||||
@site.route('/channel/<channel_id>/<download_filter>')
|
||||
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()
|
||||
channel = common.ycdldb.get_channel(channel_id)
|
||||
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 = videos[:limit]
|
||||
except ValueError:
|
||||
pass
|
||||
|
||||
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('/mark_video_state', methods=['POST'])
|
||||
def post_mark_video_state():
|
||||
if 'video_ids' not in request.form or 'state' not in request.form:
|
||||
flask.abort(400)
|
||||
video_ids = request.form['video_ids']
|
||||
state = request.form['state']
|
||||
try:
|
||||
video_ids = video_ids.split(',')
|
||||
for video_id in video_ids:
|
||||
common.ycdldb.mark_video_state(video_id, state, commit=False)
|
||||
common.ycdldb.sql.commit()
|
||||
|
||||
except ycdl.exceptions.NoSuchVideo:
|
||||
common.ycdldb.rollback()
|
||||
traceback.print_exc()
|
||||
flask.abort(404)
|
||||
|
||||
except ycdl.exceptions.InvalidVideoState:
|
||||
common.ycdldb.rollback()
|
||||
flask.abort(400)
|
||||
|
||||
return jsonify.make_json_response({'video_ids': video_ids, 'state': state})
|
||||
|
||||
@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('/refresh_channel', methods=['POST'])
|
||||
def post_refresh_channel():
|
||||
if 'channel_id' not in request.form:
|
||||
flask.abort(400)
|
||||
channel_id = request.form['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 IndexError:
|
||||
flask.abort(404)
|
||||
|
||||
force = request.form.get('force', False)
|
||||
force = ycdl.helpers.truthystring(force)
|
||||
common.ycdldb.add_channel(channel_id, commit=False)
|
||||
common.ycdldb.refresh_channel(channel_id, force=force)
|
||||
return jsonify.make_json_response({})
|
||||
|
||||
@site.route('/start_download', methods=['POST'])
|
||||
def post_start_download():
|
||||
if 'video_ids' not in request.form:
|
||||
flask.abort(400)
|
||||
video_ids = request.form['video_ids']
|
||||
try:
|
||||
video_ids = video_ids.split(',')
|
||||
for video_id in video_ids:
|
||||
common.ycdldb.download_video(video_id, commit=False)
|
||||
common.ycdldb.sql.commit()
|
||||
|
||||
except ycdl.ytapi.VideoNotFound:
|
||||
common.ycdldb.rollback()
|
||||
flask.abort(404)
|
||||
|
||||
return jsonify.make_json_response({'video_ids': video_ids, 'state': 'downloaded'})
|
||||
Loading…
Reference in a new issue