This commit is contained in:
parent
c04cf18d11
commit
87d47fb293
@ -0,0 +1,198 @@
|
|||||||
|
from operator import attrgetter, itemgetter
|
||||||
|
|
||||||
|
from flask import redirect, render_template, request, url_for, abort
|
||||||
|
|
||||||
|
from pynyaata.config import app, auth, ADMIN_USERNAME, ADMIN_PASSWORD
|
||||||
|
from pynyaata.connectors import *
|
||||||
|
from pynyaata.forms import SearchForm, DeleteForm, EditForm
|
||||||
|
|
||||||
|
if MYSQL_ENABLED:
|
||||||
|
from pynyaata.config import db
|
||||||
|
from pynyaata.models import AnimeFolder, AnimeTitle, AnimeLink
|
||||||
|
|
||||||
|
|
||||||
|
def clean_model(obj):
|
||||||
|
for attr in dir(obj):
|
||||||
|
if not attr.startswith('_') and getattr(obj, attr) is None:
|
||||||
|
try:
|
||||||
|
setattr(obj, attr, '')
|
||||||
|
except AttributeError:
|
||||||
|
pass
|
||||||
|
return obj
|
||||||
|
|
||||||
|
|
||||||
|
def mysql_required(f):
|
||||||
|
@wraps(f)
|
||||||
|
def decorated_function(*args, **kwargs):
|
||||||
|
if not MYSQL_ENABLED:
|
||||||
|
return abort(404)
|
||||||
|
return f(*args, **kwargs)
|
||||||
|
|
||||||
|
return decorated_function
|
||||||
|
|
||||||
|
|
||||||
|
@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)
|
||||||
|
if MYSQL_ENABLED:
|
||||||
|
for keyword in db.session.query(AnimeTitle.keyword.distinct()).all():
|
||||||
|
if keyword[0].lower() != query.lower():
|
||||||
|
name = Connector.boldify(name, keyword[0])
|
||||||
|
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',
|
||||||
|
mysql_disabled=not MYSQL_ENABLED)
|
||||||
|
|
||||||
|
|
||||||
|
@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,
|
||||||
|
mysql_disabled=not MYSQL_ENABLED)
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/latest')
|
||||||
|
@app.route('/latest/<int:page>')
|
||||||
|
def latest(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,
|
||||||
|
mysql_disabled=not MYSQL_ENABLED)
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/list')
|
||||||
|
@app.route('/list/<url_filters>')
|
||||||
|
@mysql_required
|
||||||
|
def list_animes(url_filters='nyaa,yggtorrent'):
|
||||||
|
filters = None
|
||||||
|
for i, to_filter in enumerate(url_filters.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', methods=['GET', 'POST'])
|
||||||
|
@mysql_required
|
||||||
|
@auth.login_required
|
||||||
|
def admin():
|
||||||
|
folders = AnimeFolder.query.all()
|
||||||
|
for folder in folders:
|
||||||
|
for title in folder.titles:
|
||||||
|
title.links.sort(key=attrgetter('season'))
|
||||||
|
folder.titles.sort(key=attrgetter('name'))
|
||||||
|
form = DeleteForm(request.form)
|
||||||
|
|
||||||
|
if form.validate_on_submit():
|
||||||
|
link = AnimeLink.query.filter_by(id=form.id.data).first()
|
||||||
|
if link:
|
||||||
|
title = link.title
|
||||||
|
folder = title.folder
|
||||||
|
db.session.delete(link)
|
||||||
|
db.session.commit()
|
||||||
|
if not len(title.links):
|
||||||
|
db.session.delete(title)
|
||||||
|
db.session.commit()
|
||||||
|
if not len(folder.titles):
|
||||||
|
db.session.delete(folder)
|
||||||
|
db.session.commit()
|
||||||
|
form.message = '%s (%s) has been successfully deleted' % (title.name, link.season)
|
||||||
|
else:
|
||||||
|
form._errors = {'id': ['Id %s was not found in the database' % form.id.data]}
|
||||||
|
|
||||||
|
return render_template('admin/list.html', search_form=SearchForm(), folders=folders, action_form=form)
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/admin/edit', methods=['GET', 'POST'])
|
||||||
|
@app.route('/admin/edit/<int:link_id>', methods=['GET', 'POST'])
|
||||||
|
@mysql_required
|
||||||
|
@auth.login_required
|
||||||
|
def admin_edit(link_id=None):
|
||||||
|
folders = AnimeFolder.query.all()
|
||||||
|
form = EditForm(request.form)
|
||||||
|
|
||||||
|
if form.validate_on_submit():
|
||||||
|
folder = AnimeFolder.query.filter_by(name=form.folder.data).first()
|
||||||
|
folder = folder if folder else AnimeFolder()
|
||||||
|
folder.name = form.folder.data
|
||||||
|
db.session.add(folder)
|
||||||
|
title = AnimeTitle.query.filter_by(name=form.name.data).first()
|
||||||
|
title = title if title else AnimeTitle()
|
||||||
|
title.folder_id = folder.id
|
||||||
|
title.name = form.name.data
|
||||||
|
title.keyword = form.keyword.data.lower() if form.keyword.data else title.keyword
|
||||||
|
db.session.add(title)
|
||||||
|
link = AnimeLink.query.filter_by(id=form.id.data).first()
|
||||||
|
link = link if link else AnimeLink()
|
||||||
|
link.title_id = title.id
|
||||||
|
link.link = form.link.data
|
||||||
|
link.season = form.season.data
|
||||||
|
link.comment = form.comment.data
|
||||||
|
link.vf = form.is_vf.data
|
||||||
|
db.session.add(link)
|
||||||
|
db.session.commit()
|
||||||
|
return redirect(url_for('admin'))
|
||||||
|
|
||||||
|
if link_id:
|
||||||
|
link = AnimeLink.query.filter_by(id=link_id).first()
|
||||||
|
form.folder.data = link.title.folder.id
|
||||||
|
else:
|
||||||
|
link = clean_model(AnimeLink())
|
||||||
|
link.title = clean_model(AnimeTitle())
|
||||||
|
link.vf = False
|
||||||
|
|
||||||
|
return render_template('admin/edit.html', search_form=SearchForm(), link=link, folders=folders, action_form=form)
|
||||||
198
pynyaata/app.py
198
pynyaata/app.py
@ -1,198 +0,0 @@
|
|||||||
from operator import attrgetter, itemgetter
|
|
||||||
|
|
||||||
from flask import redirect, render_template, request, url_for, abort
|
|
||||||
|
|
||||||
from pynyaata.config import app, auth, ADMIN_USERNAME, ADMIN_PASSWORD
|
|
||||||
from pynyaata.connectors import *
|
|
||||||
from pynyaata.forms import SearchForm, DeleteForm, EditForm
|
|
||||||
|
|
||||||
if MYSQL_ENABLED:
|
|
||||||
from pynyaata.config import db
|
|
||||||
from pynyaata.models import AnimeFolder, AnimeTitle, AnimeLink
|
|
||||||
|
|
||||||
|
|
||||||
def clean_model(obj):
|
|
||||||
for attr in dir(obj):
|
|
||||||
if not attr.startswith('_') and getattr(obj, attr) is None:
|
|
||||||
try:
|
|
||||||
setattr(obj, attr, '')
|
|
||||||
except AttributeError:
|
|
||||||
pass
|
|
||||||
return obj
|
|
||||||
|
|
||||||
|
|
||||||
def mysql_required(f):
|
|
||||||
@wraps(f)
|
|
||||||
def decorated_function(*args, **kwargs):
|
|
||||||
if not MYSQL_ENABLED:
|
|
||||||
return abort(404)
|
|
||||||
return f(*args, **kwargs)
|
|
||||||
|
|
||||||
return decorated_function
|
|
||||||
|
|
||||||
|
|
||||||
@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)
|
|
||||||
if MYSQL_ENABLED:
|
|
||||||
for keyword in db.session.query(AnimeTitle.keyword.distinct()).all():
|
|
||||||
if keyword[0].lower() != query.lower():
|
|
||||||
name = Connector.boldify(name, keyword[0])
|
|
||||||
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',
|
|
||||||
mysql_disabled=not MYSQL_ENABLED)
|
|
||||||
|
|
||||||
|
|
||||||
@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,
|
|
||||||
mysql_disabled=not MYSQL_ENABLED)
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/latest')
|
|
||||||
@app.route('/latest/<int:page>')
|
|
||||||
def latest(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,
|
|
||||||
mysql_disabled=not MYSQL_ENABLED)
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/list')
|
|
||||||
@app.route('/list/<url_filters>')
|
|
||||||
@mysql_required
|
|
||||||
def list_animes(url_filters='nyaa,yggtorrent'):
|
|
||||||
filters = None
|
|
||||||
for i, to_filter in enumerate(url_filters.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', methods=['GET', 'POST'])
|
|
||||||
@mysql_required
|
|
||||||
@auth.login_required
|
|
||||||
def admin():
|
|
||||||
folders = AnimeFolder.query.all()
|
|
||||||
for folder in folders:
|
|
||||||
for title in folder.titles:
|
|
||||||
title.links.sort(key=attrgetter('season'))
|
|
||||||
folder.titles.sort(key=attrgetter('name'))
|
|
||||||
form = DeleteForm(request.form)
|
|
||||||
|
|
||||||
if form.validate_on_submit():
|
|
||||||
link = AnimeLink.query.filter_by(id=form.id.data).first()
|
|
||||||
if link:
|
|
||||||
title = link.title
|
|
||||||
folder = title.folder
|
|
||||||
db.session.delete(link)
|
|
||||||
db.session.commit()
|
|
||||||
if not len(title.links):
|
|
||||||
db.session.delete(title)
|
|
||||||
db.session.commit()
|
|
||||||
if not len(folder.titles):
|
|
||||||
db.session.delete(folder)
|
|
||||||
db.session.commit()
|
|
||||||
form.message = '%s (%s) has been successfully deleted' % (title.name, link.season)
|
|
||||||
else:
|
|
||||||
form._errors = {'id': ['Id %s was not found in the database' % form.id.data]}
|
|
||||||
|
|
||||||
return render_template('admin/list.html', search_form=SearchForm(), folders=folders, action_form=form)
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/admin/edit', methods=['GET', 'POST'])
|
|
||||||
@app.route('/admin/edit/<int:link_id>', methods=['GET', 'POST'])
|
|
||||||
@mysql_required
|
|
||||||
@auth.login_required
|
|
||||||
def admin_edit(link_id=None):
|
|
||||||
folders = AnimeFolder.query.all()
|
|
||||||
form = EditForm(request.form)
|
|
||||||
|
|
||||||
if form.validate_on_submit():
|
|
||||||
folder = AnimeFolder.query.filter_by(name=form.folder.data).first()
|
|
||||||
folder = folder if folder else AnimeFolder()
|
|
||||||
folder.name = form.folder.data
|
|
||||||
db.session.add(folder)
|
|
||||||
title = AnimeTitle.query.filter_by(name=form.name.data).first()
|
|
||||||
title = title if title else AnimeTitle()
|
|
||||||
title.folder_id = folder.id
|
|
||||||
title.name = form.name.data
|
|
||||||
title.keyword = form.keyword.data.lower() if form.keyword.data else title.keyword
|
|
||||||
db.session.add(title)
|
|
||||||
link = AnimeLink.query.filter_by(id=form.id.data).first()
|
|
||||||
link = link if link else AnimeLink()
|
|
||||||
link.title_id = title.id
|
|
||||||
link.link = form.link.data
|
|
||||||
link.season = form.season.data
|
|
||||||
link.comment = form.comment.data
|
|
||||||
link.vf = form.is_vf.data
|
|
||||||
db.session.add(link)
|
|
||||||
db.session.commit()
|
|
||||||
return redirect(url_for('admin'))
|
|
||||||
|
|
||||||
if link_id:
|
|
||||||
link = AnimeLink.query.filter_by(id=link_id).first()
|
|
||||||
form.folder.data = link.title.folder.id
|
|
||||||
else:
|
|
||||||
link = clean_model(AnimeLink())
|
|
||||||
link.title = clean_model(AnimeTitle())
|
|
||||||
link.vf = False
|
|
||||||
|
|
||||||
return render_template('admin/edit.html', search_form=SearchForm(), link=link, folders=folders, action_form=form)
|
|
||||||
Reference in New Issue
Block a user