XylotrechusZ
�
� h
� � � d Z ddlmZ ddlmZmZ ddlmZmZ ddl m
Z
m Z ddl m
Z
mZ ddlmZ dZ G d � d
ee� � Zd S ) z*Contains the MemoryDatabase implementation� )�
LooseObjectDB)� ObjectDBR� ObjectDBW)�OStream�IStream)� BadObject�UnsupportedOperation)�ZippedStoreShaWriter�DecompressMemMapReader)�BytesIO)�MemoryDBc �R � � e Zd ZdZ� fd�Zd� Zd� Zd� Zd� Zd� Z d� Z
d � Z d
� Z � xZ
S ) r
a/ A memory database stores everything to memory, providing fast IO and object
retrieval. It should be used to buffer results and obtain SHAs before writing
it to the actual physical storage, as it allows to query whether object already
exists in the target storage before introducing actual IOc � �� t � � � � � t d� � | _ t � � | _ d S )Nzpath/doesnt/matter)�super�__init__r �_db�dict�_cache)�self� __class__s ��]/builddir/build/BUILD/cloudlinux-venv-1.0.8/venv/lib/python3.11/site-packages/gitdb/db/mem.pyr zMemoryDB.__init__' s: �� �
�������� �!5�6�6��� �f�f�� � � � c � � t d� � �)Nz$MemoryDB's always stream into memory)r )r �streams r � set_ostreamzMemoryDB.set_ostream. s � �"�#I�J�J�Jr c �~ � t � � }| j � |� � | j � |� � }|� � � |� d� � t
|� � � d�� � }t |j |j
|j |� � | j |j < |S )Nr F)�close_on_deletion)
r
r r �store�close�seekr �getvaluer �binsha�type�sizer )r �istream�zstream�
decomp_streams r r zMemoryDB.store1 s� � �&�(�(�� �� � �W�%�%�%��(�.�.��)�)���
�
����� � �Q���� /�w�/?�/?�/A�/A�UZ�[�[�[�
�&-�g�n�g�l�G�L�Zg�&h�&h�� �G�N�#��r c � � || j v S �N)r �r �shas r �
has_objectzMemoryDB.has_object@ s � ��d�k�!�!r c �, � | � |� � S r) )r r* s r �infoz
MemoryDB.infoC s � ��{�{�3� � � r c � � | j | }|j � d� � |S # t $ r}t |� � |�d }~ww xY w)Nr )r r r �KeyErrorr )r r+ �ostream�es r r zMemoryDB.streamG s[ � � (��k�#�&�G� �N� � �� "� "� "��N��� (� (� (��C�.�.�a� '����� (���s �(+ �
A
�A�A
c �* � t | j � � S r) )�lenr �r s r r$ z
MemoryDB.sizeQ s � ��4�;� � � r c �4 � | j � � � S r) )r �keysr5 s r �sha_iterzMemoryDB.sha_iterT s � ��{� � �!�!�!r c �" � d}|D ]�}|� |� � r�| � |� � }t |j � � � � � }t |j |j ||� � }|� |� � |dz
}��|S )ar Copy the streams as identified by sha's yielded by sha_iter into the given odb
The streams will be copied directly
**Note:** the object will only be written if it did not exist in the target db
:return: amount of streams actually copied into odb. If smaller than the amount
of input shas, one or more objects did already exist in odbr � )r, r r �datar r# r$ r )r r8 �odb�countr+ r1 �sior% s r � stream_copyzMemoryDB.stream_copyX s� � � ��� � �C��~�~�c�"�"�
�� �k�k�#�&�&�G��'�.�-�-�/�/�0�0�C� �g�l�G�L�#�s�C�C�G� �I�I�g� � � � �Q�J�E�E�� r )�__name__�
__module__� __qualname__�__doc__r r r r, r. r r$ r8 r? �
__classcell__)r s @r r
r
s� �� � � � � �A� A�
� � � � �K� K� K�
�
�
� "� "� "� � � �(� (� (� � � �"� "� "�� � � � � � r r
N)rC �gitdb.db.looser �
gitdb.db.baser r �
gitdb.baser r � gitdb.excr r � gitdb.streamr
r �ior �__all__r
� r r �<module>rM s ��
1� 0� (� (� (� (� (� (�� � � � � � � �
� � � � � � � �
� � � � � � � �
� � � � � � � �
� � � � � �
��M� M� M� M� M�y�)� M� M� M� M� Mr