132 lines
3.7 KiB
Python
132 lines
3.7 KiB
Python
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, 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(request.form)
|
|
if form.validate_on_submit():
|
|
link = AnimeLink.query.filter_by(id=form.id.data).first()
|
|
title = link.title
|
|
db.session.delete(link)
|
|
if not len(title.links):
|
|
db.session.delete(title)
|
|
db.session.commit()
|
|
else:
|
|
print(form.errors)
|
|
return redirect(url_for('admin'))
|
|
|
|
|
|
@app.route('/admin/edit/<id>')
|
|
@auth.login_required
|
|
def admin_edit(id):
|
|
return True
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run('0.0.0.0', APP_PORT, IS_DEBUG)
|