Category:
Date:
diff --git a/nyaa/views/__init__.py b/nyaa/views/__init__.py
index 7eb4491..c6e7d4f 100644
--- a/nyaa/views/__init__.py
+++ b/nyaa/views/__init__.py
@@ -1,11 +1,13 @@
from nyaa.views import (
account,
admin,
+ main,
site,
users,
)
account_bp = account.bp
admin_bp = admin.bp
+main_bp = main.bp
site_bp = site.bp
users_bp = users.bp
diff --git a/nyaa/views/main.py b/nyaa/views/main.py
new file mode 100644
index 0000000..086bc4e
--- /dev/null
+++ b/nyaa/views/main.py
@@ -0,0 +1,163 @@
+import math
+import re
+
+import flask
+from flask_paginate import Pagination
+
+from nyaa import app, models
+from nyaa.search import (DEFAULT_MAX_SEARCH_RESULT, DEFAULT_PER_PAGE, SERACH_PAGINATE_DISPLAY_MSG,
+ _generate_query_string, search_db, search_elastic)
+from nyaa.utils import chain_get
+
+bp = flask.Blueprint('main', __name__)
+
+
+@bp.route('/rss', defaults={'rss': True})
+@bp.route('/', defaults={'rss': False})
+def home(rss):
+ render_as_rss = rss
+ req_args = flask.request.args
+ if req_args.get('page') == 'rss':
+ render_as_rss = True
+
+ search_term = chain_get(req_args, 'q', 'term')
+
+ sort_key = req_args.get('s')
+ sort_order = req_args.get('o')
+
+ category = chain_get(req_args, 'c', 'cats')
+ quality_filter = chain_get(req_args, 'f', 'filter')
+
+ user_name = chain_get(req_args, 'u', 'user')
+ page_number = chain_get(req_args, 'p', 'page', 'offset')
+ try:
+ page_number = max(1, int(page_number))
+ except (ValueError, TypeError):
+ page_number = 1
+
+ # Check simply if the key exists
+ use_magnet_links = 'magnets' in req_args or 'm' in req_args
+
+ results_per_page = app.config.get('RESULTS_PER_PAGE', DEFAULT_PER_PAGE)
+
+ user_id = None
+ if user_name:
+ user = models.User.by_username(user_name)
+ if not user:
+ flask.abort(404)
+ user_id = user.id
+
+ special_results = {
+ 'first_word_user': None,
+ 'query_sans_user': None,
+ 'infohash_torrent': None
+ }
+ # Add advanced features to searches (but not RSS or user searches)
+ if search_term and not render_as_rss and not user_id:
+ # Check if the first word of the search is an existing user
+ user_word_match = re.match(r'^([a-zA-Z0-9_-]+) *(.*|$)', search_term)
+ if user_word_match:
+ special_results['first_word_user'] = models.User.by_username(user_word_match.group(1))
+ special_results['query_sans_user'] = user_word_match.group(2)
+
+ # Check if search is a 40-char torrent hash
+ infohash_match = re.match(r'(?i)^([a-f0-9]{40})$', search_term)
+ if infohash_match:
+ # Check for info hash in database
+ matched_torrent = models.Torrent.by_info_hash_hex(infohash_match.group(1))
+ special_results['infohash_torrent'] = matched_torrent
+
+ query_args = {
+ 'user': user_id,
+ 'sort': sort_key or 'id',
+ 'order': sort_order or 'desc',
+ 'category': category or '0_0',
+ 'quality_filter': quality_filter or '0',
+ 'page': page_number,
+ 'rss': render_as_rss,
+ 'per_page': results_per_page
+ }
+
+ if flask.g.user:
+ query_args['logged_in_user'] = flask.g.user
+ if flask.g.user.is_moderator: # God mode
+ query_args['admin'] = True
+
+ infohash_torrent = special_results.get('infohash_torrent')
+ if infohash_torrent:
+ # infohash_torrent is only set if this is not RSS or userpage search
+ flask.flash(flask.Markup('You were redirected here because '
+ 'the given hash matched this torrent.'), 'info')
+ # Redirect user from search to the torrent if we found one with the specific info_hash
+ return flask.redirect(flask.url_for('view_torrent', torrent_id=infohash_torrent.id))
+
+ # If searching, we get results from elastic search
+ use_elastic = app.config.get('USE_ELASTIC_SEARCH')
+ if use_elastic and search_term:
+ query_args['term'] = search_term
+
+ max_search_results = app.config.get('ES_MAX_SEARCH_RESULT', DEFAULT_MAX_SEARCH_RESULT)
+
+ # Only allow up to (max_search_results / page) pages
+ max_page = min(query_args['page'], int(math.ceil(max_search_results / results_per_page)))
+
+ query_args['page'] = max_page
+ query_args['max_search_results'] = max_search_results
+
+ query_results = search_elastic(**query_args)
+
+ if render_as_rss:
+ return render_rss(
+ '"{}"'.format(search_term), query_results,
+ use_elastic=True, magnet_links=use_magnet_links)
+ else:
+ rss_query_string = _generate_query_string(
+ search_term, category, quality_filter, user_name)
+ max_results = min(max_search_results, query_results['hits']['total'])
+ # change p= argument to whatever you change page_parameter to or pagination breaks
+ pagination = Pagination(p=query_args['page'], per_page=results_per_page,
+ total=max_results, bs_version=3, page_parameter='p',
+ display_msg=SERACH_PAGINATE_DISPLAY_MSG)
+ return flask.render_template('home.html',
+ use_elastic=True,
+ pagination=pagination,
+ torrent_query=query_results,
+ search=query_args,
+ rss_filter=rss_query_string,
+ special_results=special_results)
+ else:
+ # If ES is enabled, default to db search for browsing
+ if use_elastic:
+ query_args['term'] = ''
+ else: # Otherwise, use db search for everything
+ query_args['term'] = search_term or ''
+
+ query = search_db(**query_args)
+ if render_as_rss:
+ return render_rss('Home', query, use_elastic=False, magnet_links=use_magnet_links)
+ else:
+ rss_query_string = _generate_query_string(
+ search_term, category, quality_filter, user_name)
+ # Use elastic is always false here because we only hit this section
+ # if we're browsing without a search term (which means we default to DB)
+ # or if ES is disabled
+ return flask.render_template('home.html',
+ use_elastic=False,
+ torrent_query=query,
+ search=query_args,
+ rss_filter=rss_query_string,
+ special_results=special_results)
+
+
+def render_rss(label, query, use_elastic, magnet_links=False):
+ rss_xml = flask.render_template('rss.xml',
+ use_elastic=use_elastic,
+ magnet_links=magnet_links,
+ term=label,
+ site_url=flask.request.url_root,
+ torrent_query=query)
+ response = flask.make_response(rss_xml)
+ response.headers['Content-Type'] = 'application/xml'
+ # Cache for an hour
+ response.headers['Cache-Control'] = 'max-age={}'.format(1 * 5 * 60)
+ return response