103 lines
2.9 KiB
Python
103 lines
2.9 KiB
Python
from operator import itemgetter
|
|
|
|
from flask import redirect, render_template, url_for, request
|
|
|
|
from config import app, auth, ADMIN_USERNAME, ADMIN_PASSWORD, IS_DEBUG, APP_PORT, db
|
|
from connectors import *
|
|
from models import SearchForm, AnimeTitle
|
|
|
|
|
|
@auth.verify_password
|
|
def verify_password(username, password):
|
|
return username is ADMIN_USERNAME and password is 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('shorten')
|
|
def shorten(str_to_replace):
|
|
return str_to_replace[:30] + '...' if len(str_to_replace) > 30 else str_to_replace
|
|
|
|
|
|
@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, connector=Connector, flags=ConnectorLang)
|
|
|
|
|
|
@app.route('/admin')
|
|
@auth.login_required
|
|
def admin():
|
|
return 'Hello!'
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run('0.0.0.0', APP_PORT, IS_DEBUG)
|