U Z¨+dèã@sŠdZddlmZddlmZddlmZddlmZddl m Z z ddl Z Wne k rddZ YnXe e ƒZdZd ZGd d „d eƒZdS) zŽConsul result store backend. - :class:`ConsulBackend` implements KeyValueStoreBackend to store results in the key-value store of Consul. é)Ú bytes_to_str)Ú parse_url)ÚKeyValueStoreBackend)ÚImproperlyConfigured)Ú get_loggerN)Ú ConsulBackendz^You need to install the python-consul library in order to use the Consul result store backend.csheZdZdZeZdZdZdZ‡fdd„Zdd„Z d d „Z d d „Z d d„Z dd„Z dd„Zdd„Z‡ZS)rz'Consul.io K/V store backend for Celery.TZ consistentNcs<tƒj||Ž|jdkr ttƒ‚d|_|jft|jƒŽdS©N) ÚsuperÚ__init__ÚconsulrÚCONSUL_MISSINGÚ one_clientÚ_init_from_paramsrÚurl)ÚselfÚargsÚkwargs©Ú __class__©ú:/tmp/pip-unpacked-wheel-ucduq0nd/celery/backends/consul.pyr %s  zConsulBackend.__init__cKs:t d||¡||_||_||_| dd¡r6| ¡|_dS)Nz,Setting on Consul client to connect to %s:%dr )ÚloggerÚdebugÚpathÚhostnameÚportÚgetÚclientr )rrrZ virtual_hostÚparamsrrrr3sÿ zConsulBackend._init_from_paramscCs|jptj|j|j|jdS)N)ÚhostrÚ consistency)r r ZConsulrrr )rrrrrAsþzConsulBackend.clientcCs&t|ƒ}|jdkr|S|j›d|›S)Nú/)rr©rÚkeyrrrÚ_key_to_consul_keyFsz ConsulBackend._key_to_consul_keycCsN| |¡}t d|¡z| ¡j |¡\}}|dWStk rHYnXdS)Nz"Trying to fetch key %s from ConsulÚValue)r$rrrÚkvrÚ TypeError)rr#Ú_ÚdatarrrrJs   zConsulBackend.getccs|D]}| |¡VqdSr)r)rÚkeysr#rrrÚmgetSszConsulBackend.mgetcCsht|ƒ}| |¡}t d||j¡| ¡}|jj|d|jd}t d|¡t d|¡|jj |||dS)aMSet a key in Consul. Before creating the key it will create a session inside Consul where it creates a session with a TTL The key created afterwards will reference to the session's ID. If the session expires it will remove the key so that results can auto expire from the K/V store z.Trying to create Consul session %s with TTL %dÚdelete)ÚnameZbehaviorZttlzCreated Consul session %szWriting key %s to Consul)r#ÚvalueÚacquire) rr$rrÚexpiresrÚsessionÚcreater&Úput)rr#r.Z session_namerZ session_idrrrÚsetWs  ÿþ  zConsulBackend.setcCs&| |¡}t d|¡| ¡j |¡S)NzRemoving key %s from Consul)r$rrrr&r,r"rrrr,qs  zConsulBackend.delete)Ú__name__Ú __module__Ú __qualname__Ú__doc__r Zsupports_autoexpirer rr rrr$rr+r4r,Ú __classcell__rrrrrs  r)r8Zkombu.utils.encodingrZkombu.utils.urlrZcelery.backends.baserZcelery.exceptionsrZcelery.utils.logrr Ú ImportErrorr5rÚ__all__r rrrrrÚs