Produits Entr'ouvert » w.c.s. » Au quotidien

import time

from quixote import get_request, get_response, get_session, redirect

from quixote.directory import Directory, AccessControlled

import wcs

import wcs.admin.root

from wcs.backoffice.menu import *

from qommon import errors, misc

from qommon.form import *

from qommon.strftime import strftime

from strongbox import StrongboxType, StrongboxItem

class StrongboxTypeDirectory(Directory):

    _q_exports = ['', 'edit', 'delete']

    def __init__(self, strongboxtype):

        self.strongboxtype = strongboxtype

    def _q_index [html] (self):

        form = Form(enctype='multipart/form-data')

        form.add_submit('edit', _('Edit'))

        form.add_submit('delete', _('Delete'))

        form.add_submit('back', _('Back'))

        if form.get_submit() == 'edit':

            return redirect('edit')

        if form.get_submit() == 'delete':

            return redirect('delete')

        if form.get_submit() == 'back':

            return redirect('..')

        html_top('strongbox', title = _('Item Type: %s') % self.strongboxtype.label)

        '<h2>%s</h2>' % _('Item Type: %s') % self.strongboxtype.label

        get_session().display_message()

        form.render()

    def edit [html] (self):

        form = self.form()

        if form.get_submit() == 'cancel':

            return redirect('.')

        if form.is_submitted() and not form.has_errors():

            self.submit(form)

            return redirect('..')

        html_top('strongbox', title = _('Edit Item Type: %s') % self.strongboxtype.label)

        '<h2>%s</h2>' % _('Edit Item Type: %s') % self.strongboxtype.label

        form.render()

    def form(self):

        form = Form(enctype='multipart/form-data')

        form.add(StringWidget, 'label', title = _('Label'), required = True,

                value = self.strongboxtype.label)

        form.add(IntWidget, 'validation_months', title=_('Number of months of validity'),

                value=self.strongboxtype.validation_months,

                hint=_('Use 0 if there is no expiration'))

        form.add_submit('submit', _('Submit'))

        form.add_submit('cancel', _('Cancel'))

        return form

    def submit(self, form):

        for k in ('label', 'validation_months'):

            widget = form.get_widget(k)

            if widget:

                setattr(self.strongboxtype, k, widget.parse())

        self.strongboxtype.store()

    def delete [html] (self):

        form = Form(enctype='multipart/form-data')

        form.widgets.append(HtmlWidget('<p>%s</p>' % _(

                        'You are about to irrevocably delete this item type.')))

        form.add_submit('submit', _('Submit'))

        form.add_submit('cancel', _('Cancel'))

        if form.get_submit() == 'cancel':

            return redirect('..')

        if not form.is_submitted() or form.has_errors():

            get_response().breadcrumb.append(('delete', _('Delete')))

            html_top('strongbox', title = _('Delete Item Type'))

            '<h2>%s</h2>' % _('Deleting Item Type: %s') % self.strongboxtype.label

            form.render()

        else:

            self.strongboxtype.remove_self()

            return redirect('..')

class StrongboxTypesDirectory(Directory):

    _q_exports = ['', 'new']

    def _q_traverse(self, path):

        get_response().breadcrumb.append(('types/', _('Item Types')))

        return Directory._q_traverse(self, path)

    def _q_index [html] (self):

        return redirect('..')

    def new [html] (self):

        type_ui = StrongboxTypeDirectory(StrongboxType())

        form = type_ui.form()

        if form.get_submit() == 'cancel':

            return redirect('.')

        if form.is_submitted() and not form.has_errors():

            type_ui.submit(form)

            return redirect('%s/' % type_ui.strongboxtype.id)

        get_response().breadcrumb.append(('new', _('New Item Type')))

        html_top('strongbox', title = _('New Item Type'))

        '<h2>%s</h2>' % _('New Item Type')

        form.render()

    def _q_lookup(self, component):

        try:

            strongboxtype = StrongboxType.get(component)

        except KeyError:

            raise errors.TraversalError()

        get_response().breadcrumb.append((str(strongboxtype.id), strongboxtype.label))

        return StrongboxTypeDirectory(strongboxtype)

class StrongboxDirectory(AccessControlled, Directory):

    _q_exports = ['', 'types', 'add', 'add_to']

    label = N_('Strongbox')

    types = StrongboxTypesDirectory()

    def _q_access(self):

        user = get_request().user

        if not user:

            raise errors.AccessUnauthorizedError()

        admin_role = get_cfg('aq-permissions', {}).get('strongbox', None)

        if not (user.is_admin or admin_role in (user.roles or [])):

            raise errors.AccessForbiddenError(

                    public_msg = _('You are not allowed to access Strongbox Management'),

                    location_hint = 'backoffice')

        get_response().breadcrumb.append(('strongbox/', _('Strongbox')))

    def _q_index [html] (self):

        html_top('strongbox', _('Strongbox'))

        '<ul id="main-actions">'

        '  <li><a class="new-item" href="types/new">%s</a></li>' % _('New Item Type')

        '</ul>'

        get_session().display_message()

        '<div class="splitcontent-left">'

        '<div class="bo-block">'

        '<h2>%s</h2>' % _('Propose a file to a user')

        form = Form(enctype='multipart/form-data')

        form.add(StringWidget, 'q', title = _('User'), required=True)

        form.add_submit('search', _('Search'))

        form.render()

        if form.is_submitted() and not form.has_errors():

            q = form.get_widget('q').parse()

            users = self.search_for_users(q)

            if users:

                if len(users) == 1:

                    return redirect('add_to?user_id=%s' % users[0].id)

                if len(users) < 50:

                    _('(first 50 users only)')

                '<ul>'

                for u in users:

                    '<li><a href="add_to?user_id=%s">%s</a></li>' % (u.id, u.display_name)

                '</ul>'

            else:

                _('No user found.')

        '</div>'

        '</div>'

        '<div class="splitcontent-right">'

        '<div class="bo-block">'

        types = StrongboxType.select()

        '<h2>%s</h2>' % _('Item Types')

        if not types:

            '<p>'

            _('There is no item types defined at the moment.')

            '</p>'

        '<ul class="biglist" id="strongbox-list">'

        for l in types:

            type_id = l.id

            '<li class="biglistitem" id="itemId_%s">' % type_id

            '<strong class="label"><a href="types/%s/">%s</a></strong>' % (type_id, l.label)

            '<p class="commands">'

            command_icon('types/%s/edit' % type_id, 'edit')

            command_icon('types/%s/delete' % type_id, 'remove')

            '</p></li>'

        '</ul>'

        '</div>'

        '</div>'

    def search_for_users(self, q):

        if hasattr(get_publisher().user_class, 'search'):

            return get_publisher().user_class.search(q)

        if q:

            users = [x for x in get_publisher().user_class.select()

                     if q in (x.name or '') or q in (x.email or '')]

            return users

        else:

            return []

    def get_form(self):

        types = [(x.id, x.label) for x in StrongboxType.select()]

        form = Form(action='add', enctype='multipart/form-data')

        form.add(StringWidget, 'description', title=_('Description'), size=60)

        form.add(FileWidget, 'file', title=_('File'), required=True)

        form.add(SingleSelectWidget, 'type_id', title=_('Document Type'),

                 options = [(None, _('Not specified'))] + types)

        form.add(DateWidget, 'date_time', title = _('Document Date'))

        form.add_submit('submit', _('Upload'))

        return form

    def add(self):

        form = self.get_form()

        form.add(StringWidget, 'user_id', title=_('User'))

        if not form.is_submitted():

           return redirect('.')

        sffile = StrongboxItem()

        sffile.user_id = form.get_widget('user_id').parse()

        sffile.description = form.get_widget('description').parse()

        sffile.proposed_time = time.localtime()

        sffile.proposed_id = get_request().user.id

        sffile.type_id = form.get_widget('type_id').parse()

        v = form.get_widget('date_time').parse()

        sffile.set_expiration_time_from_date(v)

        sffile.store()

        sffile.set_file(form.get_widget('file').parse())

        sffile.store()

        return redirect('.')

    def add_to [html] (self):

        form = Form(enctype='multipart/form-data', action='add_to')

        form.add(StringWidget, 'user_id', title = _('User'), required=True)

        try:

            user_id = form.get_widget('user_id').parse()

            user = get_publisher().user_class.get(user_id)

        except:

            return redirect('.')

        if not user:

            return redirect('.')

        get_request().form = {}

        get_request().environ['REQUEST_METHOD'] = 'GET'

        html_top('strongbox', _('Strongbox'))

        '<h2>%s %s</h2>' % (_('Propose a file to:'), user.display_name)

        form = self.get_form()

        form.add(HiddenWidget, 'user_id', title=_('User'), value=user.id)

        form.render()