Current File : //proc/self/root/proc/3522530/root/usr/lib/python2.7/site-packages/salt/tokens/rediscluster.pyo
�
���^c @@ s� d Z d d l m Z m Z m Z y d d l Z e Z Wn e k
rQ e Z n Xd d l
Z
d d l Z d d l Z d d l
Z d d l m Z e j e � Z d Z d � Z d � Z d � Z d � Z d
� Z d � Z d S( u�
Provide token storage in Redis cluster.
To get started simply start a redis cluster and assign all hashslots to the connected nodes.
Add the redis hostname and port to master configs as eauth_redis_host and eauth_redis_port.
Default values for these configs are as follow:
.. code-block:: yaml
eauth_redis_host: localhost
eauth_redis_port: 6379
:depends: - redis-py-cluster Python package
i ( t absolute_importt print_functiont unicode_literalsN( t sixu redisclusterc C@ s t s t d f St S( NuL Could not use redis for tokens; rediscluster python client is not installed.( t HAS_REDISt Falset __virtualname__( ( ( s< /usr/lib/python2.7/site-packages/salt/tokens/rediscluster.pyt __virtual__) s
c C@ s{ | j d d � } | j d d � } y t j d | d | d t � SWn0 t j j k
rv } t j d | | | � d SXd S(
u|
Connect to the redis host and return a StrictRedisCluster client object.
If connection fails then return None.
u eauth_redis_hostu localhostu eauth_redis_porti� t hostt portt decode_responsesu( Failed to connect to redis at %s:%s - %sN( t gett redisclustert StrictRedisClustert Truet
exceptionst RedisClusterExceptiont logt warningt None( t optst
redis_hostt
redis_portt err( ( s< /usr/lib/python2.7/site-packages/salt/tokens/rediscluster.pyt
_redis_client0 s
c C@ s* t | � } | s i St t | j d d � � } t j | t j d � � j � � } yD x= | j | � d k r� t j | t j d � � j � � } q[ WWn' t
k
r� } t j d | | � i SX| | d <t
j j | � } y | j | | j | � � Wn' t
k
r%} t j d | | � i SX| S( u�
Mint a new token using the config option hash_type and store tdata with 'token' attribute set
to the token.
This module uses the hash of random 512 bytes as a token.
:param opts: Salt master config options
:param tdata: Token data to be stored with 'token' attirbute of this dict set to the token.
:returns: tdata with token if successful. Empty dict if failed.
u hash_typeu md5i u: Authentication failure: cannot get token %s from redis: %su tokenu9 Authentication failure: cannot save token %s to redis: %sN( R t getattrt hashlibR R t text_typet ost urandomt hexdigestR t ExceptionR R t saltt payloadt Serialt sett dumps( R t tdatat redis_clientt hash_typet tokR t serial( ( s< /usr/lib/python2.7/site-packages/salt/tokens/rediscluster.pyt mk_tokenA s.
$,
c C@ sv t | � } | s i St j j | � } y | j | j | � � } | SWn' t k
rq } t j d | | � i SXd S( u�
Fetch the token data from the store.
:param opts: Salt master config options
:param tok: Token value to get
:returns: Token data if successful. Empty dict if failed.
u: Authentication failure: cannot get token %s from redis: %sN( R R R! R" t loadsR R R R ( R R( R& R) R% R ( ( s< /usr/lib/python2.7/site-packages/salt/tokens/rediscluster.pyt get_tokenf s
c C@ sX t | � } | s d Sy | j | � i SWn& t k
rS } t j d | | � n Xd S( u�
Remove token from the store.
:param opts: Salt master config options
:param tok: Token to remove
:returns: Empty dict if successful. None if failed.
Nu Could not remove token %s: %s( R t deleteR R R ( R R( R& R ( ( s< /usr/lib/python2.7/site-packages/salt/tokens/rediscluster.pyt rm_token} s
c C@ s� g } t | � } | s g St j j | � } y* g | j � D] } | j d � ^ q> SWn$ t k
r~ } t j d | � g SXd S( u}
List all tokens in the store.
:param opts: Salt master config options
:returns: List of dicts (token_data)
u utf8u Failed to list keys: %sN( R R R! R" t keyst decodeR R R ( R t retR& R) t kR ( ( s< /usr/lib/python2.7/site-packages/salt/tokens/rediscluster.pyt list_tokens� s *( t __doc__t
__future__R R R R R R t ImportErrorR R t loggingR t salt.payloadR t salt.extR t getLoggert __name__R R R R R* R, R. R3 ( ( ( s< /usr/lib/python2.7/site-packages/salt/tokens/rediscluster.pyt <module> s&
%
Mini Shell