Projet

Général

Profil

0001-Use-get_or_create-to-create-KeyValue-instances-so-th.patch

Benjamin Dauvergne, 01 avril 2015 18:03

Télécharger (1,15 ko)

Voir les différences: 

Subject: [PATCH] Use get_or_create() to create KeyValue instances so that
 AuthnRequest replay does not fail (#6883)


 src/authentic2/saml/models.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)
src/authentic2/saml/models.py
815 815 
    def __unicode__(self):
816 816 
        return self.key
817 817 

  		




  818
  818
  
    
    class Meta:

  		




  819
  819
  
    
        verbose_name = _("key value association")

  		




  820
  820
  
    
        verbose_name_plural = _("key value associations")

  		




  821
  821
  
    

  		




  822
  822
  
    
def save_key_values(key, *values):

  		




  823
  
  
    
    KeyValue(key = key, value = values).save()

  		




  
  823
  
    
    # never update an existing key, key are nonces

  		




  
  824
  
    
    KeyValue.objects.get_or_create(key=key, defaults={'value': values})

  		




  824
  825
  
    

  		




  825
  826
  
    
def get_and_delete_key_values(key):

  		




  826
  827
  
    
    try:

  		




  827
  828
  
    
        kv = KeyValue.objects.get(key=key)

  		




  828
  829
  
    
        return kv.value

  		




  829
  830
  
    
    except ObjectDoesNotExist:

  		




  830
  831
  
    
        raise KeyError

  		




  831
  
  
    
-