Scrittura di un decoratore CherryPy per autorizzazione

Question

Ho un'applicazione cherrypy e su alcune delle viste che desidero iniziare solo consentendo ad alcuni utenti di visualizzarle e inviare qualcun altro a una pagina richiesta di autorizzazione.

C'è un modo per farlo con un decoratore personalizzato? Penso che sarebbe l'opzione più elegante.

Ecco un esempio di base di ciò che voglio fare:

class MyApp: 
    @authorization_required 
    def view_page1(self,appID): 
     ... do some stuff ... 
     return html 

def authorization_required(func): 
    #what do I put here? 

inoltre possibile la funzione authorization_required quando viene chiamato come decoratore accettare parametri come allow_group1, allow_group2? O ho bisogno di un decoratore separato per ogni gruppo?

Answer 1

Ok, in questo caso il vostro decoratore sarebbe simile a questa:

# without any parameters 
def authentication_required(f): 
    @functools.wraps(f) 
    def _authentication_required(*args, **kwargs): 
     # Do you login stuff here 
     return f(*args, **kwargs) 
    return _authentication_required 

# With parameters 
def authentication_required(*allowed_groups): 
    def _authentication_required(f): 
     @functools.wraps(f) 
     def __authentication_required(*args, **kwargs): 
      # Do you login stuff here 
      return f(*args, **kwargs) 
     return __authentication_required 
    return _authentication_required 

Answer 2

Davvero non vuole essere la scrittura decoratori personalizzati per CherryPy. Invece, si vuole scrivere un nuovo strumento:

def myauth(allowed_groups=None, debug=False): 
    # Do your auth here... 
    authlib.auth(...) 
cherrypy.tools.myauth = cherrypy.Tool("on_start_resource", myauth) 

Vedi http://docs.cherrypy.org/en/latest/extend.html#tools per ulteriori discussioni. Questo ha diversi vantaggi rispetto a scrivere un decoratore personalizzato:

1. Si ottiene il decoratore gratuitamente dallo strumento: @cherrypy.tools.myauth(allowed_groups=['me']), e sa già come fare a non clobber cherrypy.exposed sulla stessa funzione.
2. È possibile applicare Strumenti per gestore (con il decoratore), albero per controller (tramite _cp_config) o albero per URI (nei file di configurazione o dicts). Puoi persino mescolarli e fornire una funzione di base tramite i decoratori e quindi eseguire l'override del loro comportamento nei file di configurazione.
3. Se un file di configurazione disattiva la funzione, non si paga la penalità delle prestazioni per chiamare la funzione decoratore solo per vedere se è disattivata.
4. Ti ricorderai di aggiungere un argomento di debug come tutti gli strumenti incorporati. ;)
5. La tua funzione può essere eseguita prima (o più tardi, se è ciò di cui hai bisogno) rispetto a un decoratore personalizzato, selezionando un "punto" diverso.
6. La funzione può essere eseguita su più punti di aggancio, se necessario.