from youtube import util from .get_app_version import app_version import flask from flask import request import settings import traceback import re from sys import exc_info yt_app = flask.Flask(__name__) yt_app.config['TEMPLATES_AUTO_RELOAD'] = True yt_app.url_map.strict_slashes = False # yt_app.jinja_env.trim_blocks = True # yt_app.jinja_env.lstrip_blocks = True yt_app.add_url_rule('/settings', 'settings_page', settings.settings_page, methods=['POST', 'GET']) @yt_app.route('/') def homepage(): return flask.render_template('home.html', title="YT Local") @yt_app.route('/licenses') def licensepage(): return flask.render_template( 'licenses.html', title="Licenses - YT Local" ) theme_names = { 0: 'light_theme', 1: 'gray_theme', 2: 'dark_theme', } @yt_app.context_processor def inject_theme_preference(): return { 'theme_path': '/youtube.com/static/' + theme_names[settings.theme] + '.css', 'settings': settings, # Detect version 'current_version': app_version()['version'], 'current_branch': app_version()['branch'], 'current_commit': app_version()['commit'], } @yt_app.template_filter('commatize') def commatize(num): if num is None: return '' if isinstance(num, str): num = int(num) return '{:,}'.format(num) def timestamp_replacement(match): time_seconds = 0 for part in match.group(0).split(':'): time_seconds = 60*time_seconds + int(part) return ( """ %s """ % ( str(time_seconds), match.group(0), str(time_seconds), str(time_seconds) ) ) TIMESTAMP_RE = re.compile(r'\b(\d?\d:)?\d?\d:\d\d\b') @yt_app.template_filter('timestamps') def timestamps(text): return TIMESTAMP_RE.sub(timestamp_replacement, text) @yt_app.errorhandler(500) def error_page(e): slim = request.args.get('slim', False) # whether it was an ajax request if (exc_info()[0] == util.FetchError and exc_info()[1].code == '429' and settings.route_tor ): error_message = ('Error: YouTube blocked the request because the Tor' ' exit node is overutilized. Try getting a new exit node by' ' using the New Identity button in the Tor Browser.') if exc_info()[1].error_message: error_message += '\n\n' + exc_info()[1].error_message if exc_info()[1].ip: error_message += '\n\nExit node IP address: ' + exc_info()[1].ip return flask.render_template('error.html', error_message=error_message, slim=slim), 502 elif exc_info()[0] == util.FetchError and exc_info()[1].error_message: return (flask.render_template( 'error.html', error_message=exc_info()[1].error_message, slim=slim ), 502) return flask.render_template('error.html', traceback=traceback.format_exc(), slim=slim), 500 font_choices = { 0: 'initial', 1: '"liberation serif", "times new roman", calibri, carlito, serif', 2: 'arial, "liberation sans", sans-serif', 3: 'verdana, sans-serif', 4: 'tahoma, sans-serif', } @yt_app.route('/shared.css') def get_css(): return flask.Response( flask.render_template( 'shared.css', font_family=font_choices[settings.font] ), mimetype='text/css', )