This repository has been archived on 2023-10-01. You can view files and clone it, but cannot push or open issues or pull requests.
PyNyaaTa/app.py
2019-12-12 14:37:56 +01:00

178 lines
5.4 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, AnimeTitle, DeleteForm, SearchForm, EditForm
@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 AnimeTitle.query.all():
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', search_form=SearchForm(), title='Animes torrents search engine')
@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', search_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', search_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', search_form=SearchForm(), titles=results)
@app.route('/admin')
@auth.login_required
def admin():
folders = AnimeFolder.query.all()
return render_template('admin/list.html', search_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/<link_id>')
@auth.login_required
def admin_edit(link_id):
link = AnimeLink.query.filter_by(id=link_id).first()
edit_form = EditForm()
edit_form.folder.choices = [(query.id, query.name) for query in AnimeFolder.query.all()]
titles = AnimeTitle.query.all()
return render_template('admin/edit.html', search_form=SearchForm(), link=link, titles=titles, edit_form=edit_form)
@app.route('/admin/add')
@auth.login_required
def admin_add():
edit_form = EditForm()
edit_form.folder.choices = [('', '')] + [(query.id, query.name) for query in AnimeFolder.query.all()]
titles = AnimeTitle.query.all()
link = AnimeLink()
for attr in dir(link):
if not attr.startswith('_') and getattr(link, attr) is None:
try:
setattr(link, attr, '')
except:
pass
return render_template('admin/edit.html', search_form=SearchForm(), link=link, titles=titles, edit_form=edit_form)
@app.route('/admin/save', methods=['POST'])
@auth.login_required
def admin_save():
form = EditForm(request.form)
if form.validate_on_submit():
folder = AnimeFolder.query.filter_by(id=form.folder).first()
title = AnimeTitle.query.filter_by(name=form.name).first()
title.folder = folder
title.name = form.name
title.keyword = form.keyword.lower() if form.keyword else title.keyword
db.session.add(title)
link = AnimeLink.query.filter_by(id=form.id)
link.title = title
link.link = form.link
link.season = form.season
link.comment = form.comment
link.vf = form.is_vf
db.session.add(link)
db.session.commit()
else:
print(form.errors)
return redirect(url_for('admin'))
if __name__ == '__main__':
app.run('0.0.0.0', APP_PORT, IS_DEBUG)