~/code/projects/amnesiabbpf/ cat ../amnesiacms/amnesia/resources/root.py
# -*- coding: utf-8 -*-

from amnesia.modules.content import Content
from amnesia.modules.content import EntityManager
from amnesia.modules.content import SessionResource
from amnesia.modules.folder import FolderResource
from amnesia.modules.document import DocumentResource
from amnesia.modules.event import EventResource
from amnesia.modules.file import FileResource
from amnesia.modules.account import DatabaseAuthResource
from amnesia.modules.account import RoleResource
from amnesia.modules.search import SearchResource
from amnesia.modules.tag import TagResource

from amnesia.resources import Resource


_tree = {
    'auth': DatabaseAuthResource,
    'entities': EntityManager,
    'session': SessionResource,
    'event': EventResource,
    'document': DocumentResource,
    'folder': FolderResource,
    'file': FileResource,
    'search': SearchResource,
    'tag': TagResource,
    'roles': RoleResource
}


class Root(Resource):

    __name__ = ''
    __parent__ = None

    def __str__(self):
        return self.__class__.__name__

    def __getitem__(self, path):
        # Access to a specific resource through it's id, ex: /123
        if path.isdigit():
            entity = self.dbsession.query(Content).get(path)
            resource = self.request.cms_get_resource(entity)
            return resource(self.request, entity)

        return _tree[path](self.request, self)


def get_root(request=None):
    return Root(request)