Move download handlers into new module

[ipfire.org.git] / src / web / download.py

#!/usr/bin/python

import random
import tornado.web

from . import handlers_base as base

from . import ui_modules

class IndexHandler(base.BaseHandler):
	def get(self):
		release = self.backend.releases.get_latest()
		if not release:
			raise tornado.web.HTTPError(404)

		self.render("download/index.html", release=release)


class ReleaseHandler(base.BaseHandler):
	def get(self, slug):
		release = self.backend.releases.get_by_sname(slug)
		if not release:
			raise tornado.web.HTTPError(404)

		self.render("download/release.html", release=release)


class FileHandler(base.BaseHandler):
	def prepare(self):
		self.set_header("Pragma", "no-cache")

	def head(self, filename):
		self.redirect_to_mirror(filename)

	def get(self, filename):
		self.redirect_to_mirror(filename, log_download=True)

	def find_mirror(self, filename):
		exists = self.mirrors.file_exists(filename)
		if not exists:
			raise tornado.web.HTTPError(404, "File not found: %s" % filename)

		# Find mirrors located near to the user.
		# If we have not found any, we use a random one.
		remote_location = self.get_remote_location()

		if remote_location:
			mirrors = self.mirrors.get_for_location(remote_location, filename=filename)

			if mirrors:
				return random.choice(mirrors)

		return self.mirrors.get_random(filename=filename)

	def redirect_to_mirror(self, filename, log_download=False):
		# Find a random mirror.
		mirror = self.find_mirror(filename)

		# Construct the redirection URL.
		download_url = mirror.build_url(filename)

		# Redirect the request.
		self.redirect(download_url)

		if not log_download:
			return

		remote_location = self.get_remote_location()
		if remote_location:
			country_code = remote_location.country
		else:
			country_code = None

		self.db.execute("INSERT INTO log_download(filename, mirror, country_code) \
			VALUES(%s, %s, %s)", filename, mirror.id, country_code)


class ButtonModule(ui_modules.UIModule):
	def render(self, release):
		for arch in release.arches:
			for file in release.get_files_by_arch(arch):
				return self.render_string("download/modules/button.html", release=release, file=file)

		# return nothing
		return ""