# -*- coding: utf-8 -*-
"""Main Controller"""
import logging, string, os
from datetime import datetime
from collections import defaultdict
import pkg_resources
from tg import expose, flash, redirect, session, config, response, request, config
from tg.decorators import with_trailing_slash, without_trailing_slash
from tg.flash import TGFlash
from pylons import c, g, cache
import ew
import ming
import allura
from allura.app import SitemapEntry
from allura.lib.base import WsgiDispatchController
from allura.lib import helpers as h
from allura.lib import plugin
from allura.controllers.error import ErrorController
from allura import model as M
from allura.lib.widgets import project_list as plw
from .auth import AuthController
from .search import SearchController, ProjectBrowseController
from .static import NewForgeController
from .site_admin import SiteAdminController
from .project import NeighborhoodController, HostNeighborhoodController
from .rest import RestController
__all__ = ['RootController']
log = logging.getLogger(__name__)
TGFlash.static_template = '''$('#messages').notify('%(message)s', {status: '%(status)s'});'''
class W:
project_summary = plw.ProjectSummary()
class RootController(WsgiDispatchController):
"""
The root controller for the allura application.
All the other controllers and WSGI applications should be mounted on this
controller. For example::
panel = ControlPanelController()
another_app = AnotherWSGIApplication()
Keep in mind that WSGI applications shouldn't be mounted directly: They
must be wrapped around with :class:`tg.controllers.WSGIAppController`.
"""
auth = AuthController()
error = ErrorController()
nf = NewForgeController()
nf.admin = SiteAdminController()
search = SearchController()
rest = RestController()
def __init__(self):
for n in M.Neighborhood.query.find():
if n.url_prefix.startswith('//'): continue
n.bind_controller(self)
self.browse = ProjectBrowseController()
self.allura_sitemap = SitemapIndexController()
super(RootController, self).__init__()
def _setup_request(self):
c.project = c.app = None
c.memoize_cache = {}
c.user = plugin.AuthenticationProvider.get(request).authenticate_request()
assert c.user is not None, 'c.user should always be at least User.anonymous()'
def _cleanup_request(self):
pass
@expose('jinja:allura:templates/project_list.html')
@with_trailing_slash
def index(self, **kw):
"""Handle the front-page."""
c.project_summary = W.project_summary
projects = M.Project.query.find(
dict(is_root=True,
shortname={'$ne':'--init--'},
deleted=False)).sort('shortname').all()
neighborhoods = M.Neighborhood.query.find().sort('name')
psort = [ (n, [ p for p in projects if p.neighborhood_id==n._id ])
for n in neighborhoods ]
categories = M.ProjectCategory.query.find({'parent_id':None}).sort('name').all()
c.custom_sidebar_menu = [
SitemapEntry(cat.label, '/browse/'+cat.name) for cat in categories
]
return dict(projects=psort,title="All Projects",text=None)
class SitemapIndexController(object):
projects_per_page = 1000
@expose('jinja:allura:templates/sitemap_index.xml')
def index(self, **kw):
base_url = config.get('base_url', 'sourceforge.net')
num_projects = M.Project.query.find().count()
now = datetime.utcnow().date()
return dict(
now=now,
sitemaps = [
'%s/sitemap/%d' % (base_url, offset)
for offset in range(0, num_projects, self.projects_per_page) ])
@expose()
def _lookup(self, offset, *remainder):
return SitemapController(int(offset), self.projects_per_page), remainder
class SitemapController(object):
def __init__(self, offset, limit):
self.offset, self.limit = offset, limit
@expose('jinja:allura:templates/sitemap.xml')
def index(self, **kw):
now = datetime.utcnow().date()
base_url = config.get('base_url', 'sourceforge.net')
locs = []
for p in M.Project.query.find().skip(self.offset).limit(self.limit):
c.project = p
locs += [ base_url + s.url
for s in p.sitemap() ]
return dict(
now=now,
locs=locs)