← Back
Editing: shared_memory.cpython-310.pyc
o 0�iH � @ s� d Z ddgZddlmZ ddlZddlZddlZddlZddlZddl Z ej dkr0ddl Z dZ nddl Z dZ d d lmZ ejejB Zd Ze rId Znd Zdd� ZG dd� d�ZdZG dd� d�ZdS )z�Provides shared memory for direct access across processes. The API of this package is currently provisional. Refer to the documentation for details. � SharedMemory� ShareableList� )�partialN�ntFT� )�resource_tracker� z/psm_Zwnsm_c C sB t tt� d } | dksJ d��tt�| � }t|�t ks J �|S )z6Create a random filename for the shared memory object.� z_SHM_NAME_PREFIX too long)�_SHM_SAFE_NAME_LENGTH�len�_SHM_NAME_PREFIX�secretsZ token_hex)�nbytes�name� r �D/opt/alt/python310/lib64/python3.10/multiprocessing/shared_memory.py�_make_filename( s r c @ s� e Zd ZdZdZdZdZdZej Z dZ e rdndZ ddd �Zd d � Zd d � Zdd� Zedd� �Zedd� �Zedd� �Zdd� Zdd� ZdS )r a� Creates a new shared memory block or attaches to an existing shared memory block. Every shared memory block is assigned a unique name. This enables one process to create a shared memory block with a particular name so that a different process can attach to that same shared memory block using that same name. As a resource for sharing data across processes, shared memory blocks may outlive the original process that created them. When one process no longer needs access to a shared memory block that might still be needed by other processes, the close() method should be called. When a shared memory block is no longer needed by any process, the unlink() method should be called to ensure proper cleanup.N���i� TFr c C sx |dkst d��|rttjB | _|dkrt d��|d u r&| jtj@ s&t d��tr�|d u rL t� }z tj || j| j d�| _ W n t yF Y q,w || _ n| jrSd| n|}tj || j| j d�| _ || _ z |ro|rot�| j |� t�| j �}|j}t�| j |�| _W n ty� | �� � w t�| j d� n�|r� |d u r�t� n|}t�tjtjtj |d ? d @ |d @ |�}z1t� � }|tj kr�|d ur�t t j t�!t j �|tj ��W t�"|� q�tjd ||d �| _W t�"|� nt�"|� w || _ n?|| _ t�#tj$d |�}zt�%|tj$ddd�}W t�"|� nt�"|� w z t�&|�}W t�'|� nt�'|� w tjd ||d �| _|| _(t)| j�| _*d S )Nr z!'size' must be a positive integerz4'size' must be a positive number different from zeroz&'name' can only be None if create=TrueT)�mode�/� shared_memory� l �� r )ZtagnameF)+� ValueError�_O_CREX�os�O_RDWR�_flags�O_EXCL� _USE_POSIXr � _posixshmemZshm_open�_mode�_fd�FileExistsError�_name�_prepend_leading_slash� ftruncate�fstat�st_size�mmap�_mmap�OSError�unlinkr �register�_winapiZCreateFileMappingZINVALID_HANDLE_VALUEZNULLZPAGE_READWRITEZ GetLastErrorZERROR_ALREADY_EXISTS�errnoZEEXIST�strerrorZ CloseHandleZOpenFileMappingZ FILE_MAP_READZ MapViewOfFileZVirtualQuerySizeZUnmapViewOfFile�_size� memoryview�_buf) �selfr �create�sizeZstatsZ temp_nameZh_mapZlast_error_codeZp_bufr r r �__init__K s� � �� � � � ��� zSharedMemory.__init__c C s$ z| � � W d S ty Y d S w �N)�closer* �r3 r r r �__del__� s �zSharedMemory.__del__c C s | j | jd| jffS )NF)� __class__r r5 r9 r r r � __reduce__� s ��zSharedMemory.__reduce__c C s | j j� d| j�d| j� d�S )N�(z, size=�))r; �__name__r r5 r9 r r r �__repr__� s zSharedMemory.__repr__c C � | j S )z4A memoryview of contents of the shared memory block.)r2 r9 r r r �buf� � zSharedMemory.bufc C s. | j }tr| jr| j �d�r| j dd� }|S )z4Unique name that identifies the shared memory block.r r N)r# r r$ � startswith)r3 Z reported_namer r r r � s zSharedMemory.namec C rA )zSize in bytes.)r0 r9 r r r r5 � rC zSharedMemory.sizec C s` | j dur | j �� d| _ | jdur| j�� d| _tr,| jdkr.t�| j� d| _dS dS dS )zkCloses access to the shared memory from this instance but does not destroy the shared memory block.Nr r )r2 �releaser) r8 r r! r r9 r r r r8 � s �zSharedMemory.closec C s0 t r| jrt�| j� t�| jd� dS dS dS )z�Requests that the underlying shared memory block be destroyed. In order to ensure proper cleanup of resources, unlink should be called once (and only once) across all processes which have access to the shared memory block.r N)r r# r Z shm_unlinkr � unregisterr9 r r r r+ � s �zSharedMemory.unlink)NFr )r? � __module__� __qualname__�__doc__r# r! r) r2 r r r r r r$ r6 r: r< r@ �propertyrB r r5 r8 r+ r r r r r 2 s* n �utf8c @ s e Zd ZdZedededededdj diZ dZ d d � d d � d d � d d � d�Z e dd� �Zd6dd�dd�Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zed$d%� �Zed&d'� �Zed(d)� �Zed*d+� �Z ed,d-� �Z ed.d/� �Z ed0d1� �Z d2d3� Z d4d5� Z!e"e#j$�Z%dS )7r a� Pattern for a mutable list-like object shareable via a shared memory block. It differs from the built-in list type in that these lists can not change their overall length (i.e. no append, insert, etc.) Because values are packed into a memoryview as bytes, the struct packing format for any storable value must require no more than 8 characters to describe its format.�q�dzxxxxxxx?z%dsNzxxxxxx?x� c C s | S r7 r ��valuer r r �<lambda> � zShareableList.<lambda>c C s | � d��t�S �N� )�rstrip�decode� _encodingrO r r r rQ s c C s | � d�S rS )rU rO r r r rQ s c C s d S r7 r )Z_valuer r r rQ rR )r r r � c C s6 t | ttdjf�s dS t | t�rdS t | t�rdS dS )z�Used in concert with _back_transforms_mapping to convert values into the appropriate Python objects when retrieving them from the list as well as when storing them.Nr r r rX )� isinstance�str�bytesr; rO r r r �_extract_recreation_code s z&ShareableList._extract_recreation_code�r c s� |d u s|d urq|p d}�fdd�|D �}t |��_tdd� |D ���jks(J �d}dg�_|D ]}||d dkr<�jnt|d d� �7 }�j�|� q0�fd d�|D �}t�d �j d � |� �j �j �}t |d |d ��_nt |��_|d ur�t� tjd �j �jjd�jg�j�R � tjd � |��jj�jg� fdd�|D ��R � tj�j �jj�jg� fdd�|D ��R � tj�j �jj�jg|�R � d S t ���_tt��j �jjd���_d S )Nr c sP g | ]$}t |ttf�s� jt|� n� jt|� � jt|�� j d f �qS )r )rY rZ r[ �_types_mapping�type� _alignmentr ��.0�itemr9 r r � <listcomp>- s �� ��z*ShareableList.__init__.<locals>.<listcomp>c s s � | ] }t |�d kV qdS )rN N)r )rb �fmtr r r � <genexpr>6 s � z)ShareableList.__init__.<locals>.<genexpr>r r �sc s g | ]}� � |��qS r )r\ ra r9 r r rd ? s �rL � T)r4 r5 c 3 s( � | ]}t |t�r|�� �n|V qd S r7 )rY rZ �encode�rb �v��_encr r rf Z s �& c 3 s � | ]}|� � �V qd S r7 )ri rj rl r r rf ` s � rN )r � _list_len�sum�_allocated_offsetsr` �int�append�structZcalcsize�_format_size_metainfo�join�_format_packing_metainfo� _format_back_transform_codesr �shmrW � pack_intorB �_offset_data_start�_offset_packing_formats� _offset_back_transform_codes�list� unpack_from)r3 Zsequencer Z_formats�offsetre Z_recreation_codesZrequested_sizer )rm r3 r r6 * s� � & ����� ������� � � �zShareableList.__init__c C sj |dkr|n|| j }|| j ks| j dk rtd��t�d| jj| j|d �d }|�d�}|�t �}|S )z>Gets the packing format for a single value stored in the list.r � Requested position out of range.�8srN rT ) rn � IndexErrorrs r~ rx rB r{ rU rV rW )r3 �positionrk re � fmt_as_strr r r �_get_packing_formats s �� z!ShareableList._get_packing_formatc C sF || j ks | j dk rtd��t�d| jj| j| �d }| j| }|S )z9Gets the back transformation function for a single value.r r� �b)rn r� rs r~ rx rB r| �_back_transforms_mapping)r3 r� �transform_codeZtransform_functionr r r �_get_back_transform� s �� z!ShareableList._get_back_transformc C sh || j ks | j dk rtd��t�d| jj| j|d |�t�� | � |�}t�d| jj| j | |� dS )zvSets the packing format and back transformation code for a single value in the list at the specified position.r r� r� rN r� N) rn r� rs ry rx rB r{ ri rW r\ r| )r3 r� r� rP r� r r r �!_set_packing_format_and_transform� s � �z/ShareableList._set_packing_format_and_transformc C sr |dkr|n|| j }z| j| j| }t�| �|�| jj|�\}W n ty- td��w | � |�}||�}|S )Nr zindex out of range) rn rz rp rs r~ r� rx rB r� r� )r3 r� r rk Zback_transformr r r � __getitem__� s � � zShareableList.__getitem__c C s� |dkr|n|| j }z| j| }| j| }| �|�}W n ty' td��w t|ttf�s9| jt |� }|}n0| j|d | }t|t�rL|� t �n|}t |�|krXt d��|d dkra|}n| jt |f }| �|||� t�|| jj||� d S )Nr z assignment index out of ranger z(bytes/str item exceeds available storager rg )rn rp rz r� r� rY rZ r[ r^ r_ ri rW r r r� rs ry rx rB ) r3 r� rP Z item_offsetr Zcurrent_formatZ new_formatZ encoded_valueZallocated_lengthr r r � __setitem__� s: � �� ��zShareableList.__setitem__c C s t | j| jjd�dfS )Nr] r )r r; rx r r9 r r r r< � � zShareableList.__reduce__c C s t �d| jjd�d S )NrL r )rs r~ rx rB r9 r r r �__len__� r� zShareableList.__len__c C s" | j j� dt| �� d| jj�d�S )Nr= z, name=r> )r; r? r} rx r r9 r r r r@ � s "zShareableList.__repr__c s d� � fdd�t� j�D ��S )z=The struct packing format used by all currently stored items.rh c 3 s � | ]}� � |�V qd S r7 )r� )rb �ir9 r r rf � s � �z'ShareableList.format.<locals>.<genexpr>)ru �rangern r9 r r9 r �format� s �zShareableList.formatc C s d| j d S )z>The struct packing format used for the items' storage offsets.rL r �rn r9 r r r rt � s z#ShareableList._format_size_metainfoc C � d| j S )z>The struct packing format used for the items' packing formats.r� r� r9 r r r rv � � z&ShareableList._format_packing_metainfoc C r� )z>The struct packing format used for the items' back transforms.r� r� r9 r r r rw � r� z*ShareableList._format_back_transform_codesc C s | j d d S )Nr rN r� r9 r r r rz � s z ShareableList._offset_data_startc C s | j | jd S )Nr )rz rp r9 r r r r{ � � z%ShareableList._offset_packing_formatsc C s | j | jd S )NrN )r{ rn r9 r r r r| r� z*ShareableList._offset_back_transform_codesc s t � fdd�| D ��S )zCL.count(value) -> integer -- return number of occurrences of value.c 3 s � | ]}� |kV qd S r7 r )rb �entryrO r r rf s � z&ShareableList.count.<locals>.<genexpr>)ro )r3 rP r rO r �count s zShareableList.countc C s0 t | �D ] \}}||kr| S qt|�d���)zpL.index(value) -> integer -- return first index of value. Raises ValueError if the value is not present.z not in this container)� enumerater )r3 rP r� r� r r r �index s �zShareableList.indexr7 )&r? rG rH rI rq �float�boolrZ r[ r; r^ r` r� � staticmethodr\ r6 r� r� r� r� r� r< r� r@ rJ r� rt rv rw rz r{ r| r� r� � classmethod�types� GenericAlias�__class_getitem__r r r r r � sV �� I! )rI �__all__� functoolsr r( r r. rs r r� r r- r r rh r �O_CREATr r r r r r rW r r r r r �<module> s2 F
Save File
Cancel