from operator import itemgetter from flask import redirect, render_template, request, url_for from config import app, auth, db, ADMIN_USERNAME, ADMIN_PASSWORD, APP_PORT, IS_DEBUG from connectors import * from models import AnimeFolder, AnimeTitle, DeleteForm, SearchForm @auth.verify_password def verify_password(username, password): return username == ADMIN_USERNAME and password == ADMIN_PASSWORD @app.template_filter('boldify') def boldify(name): query = request.args.get('q') name = Connector.boldify(name, query) for title in ConnectorCache.get_keywords(): if title.keyword is not query: name = Connector.boldify(name, title.keyword) return name @app.template_filter('flagify') def flagify(is_vf): return ConnectorLang.FR.value if is_vf else ConnectorLang.JP.value @app.template_filter('colorify') def colorify(model): return Connector.get_instance(model.link, model.title.keyword).color @app.route('/') def home(): return render_template('layout.html', form=SearchForm()) @app.route('/search') def search(): query = request.args.get('q') if not query: return redirect(url_for('home')) results = [ Nyaa(query).run(), Pantsu(query).run(), YggTorrent(query).run(), YggAnimation(query).run(), AnimeUltime(query).run(), ] return render_template('search.html', form=SearchForm(), connectors=results) @app.route('/latest') def latest(): page = request.args.get('page', 1) torrents = [ Nyaa('', return_type=ConnectorReturn.HISTORY, page=page).run(), Pantsu('', return_type=ConnectorReturn.HISTORY, page=page).run(), YggTorrent('', return_type=ConnectorReturn.HISTORY, page=page).run(), YggAnimation('', return_type=ConnectorReturn.HISTORY, page=page).run(), AnimeUltime('', return_type=ConnectorReturn.HISTORY, page=page).run(), ] results = [] for torrent in torrents: results = results + torrent.data for result in results: result['self'] = Connector.get_instance(result['href'], '') results.sort(key=itemgetter('date'), reverse=True) return render_template('latest.html', form=SearchForm(), torrents=results, page=page) @app.route('/list') def list_animes(): filters = None for i, to_filter in enumerate(request.args.get('s', 'nyaa,yggtorrent').split(',')): if not i: filters = AnimeLink.link.contains(to_filter) else: filters = filters | AnimeLink.link.contains(to_filter) titles = db.session.query(AnimeTitle, AnimeLink).join(AnimeLink).filter(filters).order_by(AnimeTitle.name).all() results = {} for title, link in titles: if title.id not in results: results[title.id] = [link] else: results[title.id].append(link) return render_template('list.html', form=SearchForm(), titles=results) @app.route('/admin') @auth.login_required def admin(): folders = AnimeFolder.query.all() return render_template('admin/list.html', form=SearchForm(), folders=folders, delete_form=DeleteForm()) @app.route('/admin/delete', methods=['POST']) @auth.login_required def admin_delete(): form = DeleteForm() form_id = request.form.id if form.validate_on_submit() and form_id: link = AnimeLink.query.filter_by(id=form_id).first() title = link.title db.session.delete(link) if not len(title.links): db.session.delete(title) return redirect(url_for('admin')) @app.route('/admin/edit/') @auth.login_required def admin_edit(id): return True if __name__ == '__main__': app.run('0.0.0.0', APP_PORT, IS_DEBUG)