import sys
from ckan.lib.base import request
from ckan.lib.base import c, g, h
from ckan.lib.base import model
from ckan.lib.base import render
from ckan.lib.base import _
 
from ckan.controllers.user import UserController
 
 
class CustomUserController(UserController):
    """This controller is an example to show how you might extend or
    override core CKAN behaviour from an extension package.
 
    It duplicates functionality in the core CKAN UserController's
    register function, but extends it to make an email address
    mandatory.
    """
    def custom_register(self):
        if request.method == 'POST':
            # custom validation that requires an email address
            error = False
            c.email = request.params.getone('email')
            c.login = request.params.getone('login')
            if not model.User.check_name_available(c.login):
                error = True
                h.flash_error(_("That username is not available."))
            if not c.email:
                error = True
                h.flash_error(_("You must supply an email address."))
            try:
                self._get_form_password()
            except ValueError, ve:
                h.flash_error(ve)
                error = True
            if error:
                return render('user/register.html')
        # now delegate to core CKAN register method
        return self.register()