simon987/nyaa
1
0
Fork 0
mirror of https://github.com/simon987/nyaa.git synced 2025-12-18 01:09:03 +00:00
Code Issues Projects Releases Wiki Activity

Move torrent magnet and torrent download into 'torrents' blueprint

Browse Source
This commit is contained in:
Kfir Hadas
2017-07-24 23:10:59 +03:00
parent 9fef343c1b
commit ef56e54521
5 changed files with 48 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
nyaa/views/torrents.py
View File

@@ -1,11 +1,13 @@
import json
import os.path
from urllib.parse import quote
import flask
from werkzeug.datastructures import CombinedMultiDict
from sqlalchemy.orm import joinedload
from nyaa import backend, db, forms, models
from nyaa import app, backend, db, forms, models, torrents
from nyaa.utils import cached_function
bp = flask.Blueprint('torrents', __name__)
@@ -146,6 +148,35 @@ def edit_torrent(torrent_id):
torrent=torrent)
@bp.route('/view/<int:torrent_id>/magnet')
def redirect_magnet(torrent_id):
torrent = models.Torrent.by_id(torrent_id)
if not torrent:
flask.abort(404)
return flask.redirect(torrents.create_magnet(torrent))
@bp.route('/view/<int:torrent_id>/torrent')
@bp.route('/download/<int:torrent_id>.torrent', endpoint='download')
def download_torrent(torrent_id):
torrent = models.Torrent.by_id(torrent_id)
if not torrent or not torrent.has_torrent:
flask.abort(404)
torrent_file, torrent_file_size = _get_cached_torrent_file(torrent)
disposition = 'attachment; filename="{0}"; filename*=UTF-8\'\'{0}'.format(
quote(torrent.torrent_name.encode('utf-8')))
resp = flask.Response(torrent_file)
resp.headers['Content-Type'] = 'application/x-bittorrent'
resp.headers['Content-Disposition'] = disposition
resp.headers['Content-Length'] = torrent_file_size
return resp
@bp.route('/upload', methods=['GET', 'POST'])
def upload():
upload_form = forms.UploadForm(CombinedMultiDict((flask.request.files, flask.request.form)))
@@ -175,3 +206,14 @@ def _create_upload_category_choices():
cat_name = ' - '.join(cat_names)
choices.append((key, cat_name, is_main_cat))
return choices
def _get_cached_torrent_file(torrent):
# Note: obviously temporary
cached_torrent = os.path.join(app.config['BASE_DIR'],
'torrent_cache', str(torrent.id) + '.torrent')
if not os.path.exists(cached_torrent):
with open(cached_torrent, 'wb') as out_file:
out_file.write(torrents.create_bencoded_torrent(torrent))
return open(cached_torrent, 'rb'), os.path.getsize(cached_torrent)