Come mettere un CookieJar sottaceto?

Question

Ho un oggetto con un CookieJar che voglio mettere sottaceto.

Tuttavia, come probabilmente tutti sapete, il pickle si strozza sugli oggetti che contengono oggetti di blocco. E per qualche ragione orribile, un CookieJar ha un oggetto di blocco.

from cPickle import dumps 
from cookielib import CookieJar 

class Person(object): 
    def __init__(self, name): 
     self.name = name 
     self.cookies = CookieJar() 

bob = Person("bob") 
dumps(bob) 

# Traceback (most recent call last): 
# File "<stdin>", line 1, in <module> 
# cPickle.UnpickleableError: Cannot pickle <type 'thread.lock'> objects 

Come faccio persistono questo?

L'unica soluzione che posso pensare è utilizzare FileCookieJar.save e FileCookieJar.load in un oggetto stringIO. Ma c'è un modo migliore?

Answer 1

Ecco un tentativo, derivando una classe da CookieJar, che sostituisce getstate/setstate utilizzato da pickle. Non ho usato cookieJar, quindi non so se è utilizzabile ma puoi scaricare la classe derivata

from cPickle import dumps 
from cookielib import CookieJar 
import threading 

class MyCookieJar(CookieJar): 
    def __getstate__(self): 
     state = self.__dict__.copy() 
     del state['_cookies_lock'] 
     return state 

    def __setstate__(self, state): 
     self.__dict__ = state 
     self._cookies_lock = threading.RLock() 

class Person(object): 
    def __init__(self, name): 
     self.name = name 
     self.cookies = MyCookieJar() 

bob = Person("bob") 
print dumps(bob) 

Answer 2

CookieJar non è particolarmente ben progettato per la persistenza (che è quello che i FileCookieJar sottoclassi sono per lo più su -!), Ma si può iterare su un'istanza CookieJar per ottenere tutti i cookie (e persistono un elenco di quelli, per esempio), e, per ricostruire il barattolo dati i cookie, utilizzare set_cookie su ciascuno. Ecco come impostare i cookie persistenti e non persistenti, usando il metodo copy_reg per registrare le funzioni appropriate se dovessi usarle spesso.