o Df@sdZddlmZddlmZddlmZddlmZddl m Z zddl Z Wn e y1dZ Ynwe e ZdZd ZGd d d eZdS) zConsul 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.csheZdZdZeZdZdZdZfddZddZ d d Z d d Z d dZ ddZ ddZddZZS)rz'Consul.io K/V store backend for Celery.T consistentNcsDtj|i||jdurttd|_|jdit|jdS)N) super__init__consulrCONSUL_MISSING one_client_init_from_paramsrurl)selfargskwargs __class__r O/home/ubuntu/webapp/venv/lib/python3.10/site-packages/celery/backends/consul.pyr %s  zConsulBackend.__init__cKs>td||||_||_||_|ddr||_dSdS)Nz,Setting on Consul client to connect to %s:%dr)loggerdebugpathhostnameportgetclientr)rrr virtual_hostparamsr r rr3s zConsulBackend._init_from_paramscCs|jp tj|j|j|jdS)N)hostr consistency)rr Consulrrr!)rr r rrAszConsulBackend.clientcCs&t|}|jdur |S|jd|S)N/)rrrkeyr r r_key_to_consul_keyFsz ConsulBackend._key_to_consul_keycCsJ||}td|z|j|\}}|dWSty$YdSw)Nz"Trying to fetch key %s from ConsulValue)r&rrrkvr TypeError)rr%_datar r rrJs    zConsulBackend.getccs|D]}||VqdS)N)r)rkeysr%r r rmgetSszConsulBackend.mgetcCsht|}||}td||j|}|jj|d|jd}td|td||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 %ddelete)namebehaviorttlzCreated Consul session %szWriting key %s to Consul)r%valueacquire) rr&rrexpiresrsessioncreater(put)rr%r2 session_namer session_idr r rsetWs   zConsulBackend.setcCs&||}td||j|S)NzRemoving key %s from Consul)r&rrrr(r.r$r r rr.qs  zConsulBackend.delete)__name__ __module__ __qualname____doc__r supports_autoexpirer!rr rrr&rr-r:r. __classcell__r r rrrs  r)r>kombu.utils.encodingrkombu.utils.urlrcelery.backends.basercelery.exceptionsrcelery.utils.logrr ImportErrorr;r__all__r rr r r rs