Source code for tool_shed.webapp.controllers.hg

import logging

from mercurial.hgweb.hgwebdir_mod import hgwebdir

from galaxy import web
from galaxy.exceptions import ObjectNotFound
from galaxy.model.base import transaction
from galaxy.webapps.base.controller import BaseUIController
from tool_shed.util.repository_util import get_repository_by_name_and_owner

log = logging.getLogger(__name__)


[docs]class PortAsStringMiddleware:
[docs] def __init__(self, application): self.application = application
def __call__(self, environ, start_response): environ["SERVER_PORT"] = str(environ["SERVER_PORT"]) return self.application(environ, start_response)
[docs]class HgController(BaseUIController):
[docs] @web.expose def handle_request(self, trans, **kwd): # The os command that results in this method being called will look something like: # hg clone http://test@127.0.0.1:9009/repos/test/convert_characters1 hgweb_config = trans.app.hgweb_config_manager.hgweb_config def make_web_app(): hgwebapp = hgwebdir(hgweb_config.encode("utf-8")) return hgwebapp wsgi_app = make_web_app() repository = None path_info = kwd.get("path_info", None) if path_info and len(path_info.split("/")) == 2: owner, name = path_info.split("/") repository = get_repository_by_name_and_owner(trans.app, name, owner) if repository: if repository.deprecated: raise ObjectNotFound("Requested repository not found or deprecated.") cmd = kwd.get("cmd", None) if cmd == "getbundle": times_downloaded = repository.times_downloaded times_downloaded += 1 repository.times_downloaded = times_downloaded trans.sa_session.add(repository) with transaction(trans.sa_session): trans.sa_session.commit() return PortAsStringMiddleware(wsgi_app)