Remove old download pages

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

#!/usr/bin/python

import random
import tornado.web

from handlers_base import *

class DownloadFileHandler(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 DownloadCompatHandler(BaseHandler):
	def get(self, path, url):
		for filename in self.mirrors.get_all_files():
			if not filename.endswith("/%s" % url):
				continue

			self.redirect("/%s" % filename)
			return

		raise tornado.web.HTTPError(404)