"""Solon decorators"""
import logging

import pylons
from decorator import decorator
from pylons.controllers.util import abort
from tw2.core import ValidationError

log = logging.getLogger(__name__)

def in_group(group):
    """Requires a user to be logged in, and the group specified"""
    def wrapper(func, *args, **kwargs):
        user = pylons.tmpl_context.user
        if not user:
            log.debug("No user logged in for permission restricted function")
            abort(401, "Not Authorized")
        if user.in_group(group):
            log.debug("User %s verified in group %s", user, group)
            return func(*args, **kwargs)
        else:
            log.debug("User %s not in group %s", user, group)
            abort(401, "Not Authorized")
    return decorator(wrapper)

@decorator
def logged_in(func, *args, **kwargs):
    if not pylons.tmpl_context.user:
        abort(401, "Not Authorized")
    else:
        return func(*args, **kwargs)

def validate(form, error_handler):
    """Validate a form with tw2"""
    @decorator
    def wrapper(func, self, *args, **kwargs):
        try:
            self.form_result=form.validate(pylons.request.params.mixed())
        except ValidationError, e:
            #Don't bother saving the widget, it's saved by tw2 on the request object anyway
            pylons.request.method = 'GET'
            return getattr(self, error_handler)(*args, **kwargs)
        return func(self, *args, **kwargs)
    return wrapper