File: //usr/local/lib/python3.9/site-packages/celery/backends/__pycache__/azureblockblob.cpython-39.pyc
a
X>h� � @ s� d Z ddlmZ ddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
z2dd lm mZ dd
lmZmZ dd lmZ W n ey� d ZY n0 d Ze e�Zd
ZG dd� de
�Zd S )z0The Azure Storage Block Blob backend for Celery.� )� Transport)�cached_property)� bytes_to_str)�ImproperlyConfigured)�
get_logger� )�KeyValueStoreBackendN)�ResourceExistsError�ResourceNotFoundError)�BlobServiceClient)�AzureBlockBlobBackendzazureblockblob://c sh e Zd ZdZd� fdd� Zeefdd��Zedd� �Z d d
� Z
d d � Z d
d� Z dd� Z
ddd�Z� ZS )r z,Azure Storage Block Blob backend for Celery.Nc s| t � j|i |�� tdu s$tjdk r,td��| jj}| �|�| _|pJ|d | _ |�
dd�| _ |�
dd�| _ |�
d d
�| _
dS ) z�
Supported URL formats:
azureblockblob://CONNECTION_STRING
azureblockblob://DefaultAzureCredential@STORAGE_ACCOUNT_URL
azureblockblob://ManagedIdentityCredential@STORAGE_ACCOUNT_URL
NZ12zWYou need to install the azure-storage-blob v12 library touse the AzureBlockBlob backendZ azureblockblob_container_nameZazureblockblob_base_path� Z!azureblockblob_connection_timeout� Zazureblockblob_read_timeout�x )�super�__init__� azurestorage� __version__r Zapp�conf�
_parse_url�_connection_string�_container_name�get� base_path�_connection_timeout�
_read_timeout)�self�urlZcontainer_name�args�kwargsr �� __class__� �H/usr/local/lib/python3.9/site-packages/celery/backends/azureblockblob.pyr s � ��z AzureBlockBlobBackend.__init__c C s |t |�d � }|s td��|S )Nz Invalid URL)�lenr )�clsr �prefixZconnection_stringr" r" r# r < s z AzureBlockBlobBackend._parse_urlc C s� d| j v sd| j v r:t�| j �\}}t||| j| jd�}ntj| j | j| jd�}z |j| jd� d| j� d�}W n t y� d| j� d �}Y n0 t
� |� |S )
z�Return the Azure Storage Blob service client.
If this is the first call to the property, the client is created and
the container is created if it doesn't yet exist.
ZDefaultAzureCredentialZManagedIdentityCredential)Z account_urlZ
credential�connection_timeout� read_timeout)r' r( )�namez Container created with name �.zContainer with name z* already.exists. This will not be created.) r �AzureStorageQueuesTransport� parse_urir r r Zfrom_connection_stringZcreate_containerr r �LOGGER�info)r Z credential_r �client�msgr" r" r# �_blob_service_clientD s2 �����
z*AzureBlockBlobBackend._blob_service_clientc C s` t |�}t�d| j|� | jj| j| j� |� �d�}z|�� �� � � W S t
yZ Y dS 0 dS )zwRead the value stored at the given key.
Args:
key: The key for which to read the value.
z Getting Azure Block Blob %s/%s�� container�blobN) r r- �debugr r1 �get_blob_clientr Z
download_blob�readall�decoder
�r �key� blob_clientr" r" r# r k s � zAzureBlockBlobBackend.getc C sN t |�}t�d| j� d|� �� | jj| j| j� |� �d�}|j|dd� dS )z�Store a value for a given key.
Args:
key: The key at which to store the value.
value: The value to store.
zCreating azure blob at �/r2 T)� overwriteN)r r- r5 r r1 r6 r Z upload_blob)r r: �valuer; r" r" r# �set~ s �zAzureBlockBlobBackend.setc s � fdd�|D �S )zqRead all the values for the provided keys.
Args:
keys: The list of keys to read.
c s g | ]}� � |��qS r" )r )�.0r: �r r" r# �
<listcomp>� � z.AzureBlockBlobBackend.mget.<locals>.<listcomp>r" )r �keysr" rA r# �mget� s zAzureBlockBlobBackend.mgetc C sH t |�}t�d| j� d|� �� | jj| j| j� |� �d�}|�� dS )zlDelete the value at a given key.
Args:
key: The key of the value to delete.
zDeleting azure blob at r< r2 N)r r- r5 r r1 r6 r Z delete_blobr9 r" r" r# �delete� s �z AzureBlockBlobBackend.deleteFc sF |rt � | j� �S | j�d�}d� � fdd�|D �}t � d�|�� �S )N�;z AccountKey=c s$ g | ] }|� � �r � � d �n|�qS )z**)�
startswith)r@ �part�Zaccount_key_prefixr" r# rB � s �z0AzureBlockBlobBackend.as_uri.<locals>.<listcomp>)�"AZURE_BLOCK_BLOB_CONNECTION_PREFIXr �split�join)r Zinclude_passwordZconnection_string_partsZ redacted_connection_string_partsr" rJ r# �as_uri� s ��
���z AzureBlockBlobBackend.as_uri)NN)F)�__name__�
__module__� __qualname__�__doc__r � classmethodrK r r r1 r r? rE rF rN �
__classcell__r" r" r r# r s �!
& r ) rR Z"kombu.transport.azurestoragequeuesr r+ Z kombu.utilsr Zkombu.utils.encodingr Zcelery.exceptionsr Zcelery.utils.logr �baser Zazure.storage.blobZstorager4 r Zazure.core.exceptionsr r
r � ImportError�__all__rO r- rK r r" r" r" r# �<module> s