File: //opt/saltstack/salt/lib/python3.10/site-packages/gitdb/db/__pycache__/loose.cpython-310.pyc
o
;j " � @ s� d dl mZ d dlmZmZmZ d dlmZmZ d dl m
Z
m Z m Z m
Z
d dlmZmZ d dlmZmZmZmZmZmZmZmZmZmZm Z m Z d dl m Z m Z m!Z!m"Z" d dl#m$Z$ d d l%Z%d d l&Z&d d l'Z'd d l(Z(d
Z)G d d � d eee�Z*d S )
� )�suppress)�
FileDBBase� ObjectDBR� ObjectDBW)� BadObject�AmbiguousObjectName)�DecompressMemMapReader�FDCompressedSha1Writer�FDStream�
Sha1Writer)�OStream�OInfo) �file_contents_ro_filepath�ENOENT�
hex_to_bin�
bin_to_hex�exists�chmod�isfile�remove�rename�dirname�basename�join)�
chunk_size�loose_object_header_info� write_object� stream_copy)� force_bytesN)�
LooseObjectDBc s� e Zd ZdZeZedd�Zej dkredd�Z� fdd�Z
dd � Z d
d � Z d d
� Z
dd� Z� fdd�Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Z� ZS ) r z/A database which operates on loose object filesZ444� �ntZ644c s& t � �|� t� | _ttdd�| _d S )N� O_NOATIMEr )�super�__init__�dict�_hexsha_to_file�getattr�os�_fd_open_flags)�self� root_path�� __class__� �B/opt/saltstack/salt/lib/python3.10/site-packages/gitdb/db/loose.pyr$ M s zLooseObjectDB.__init__c C s t |dd� |dd� �S )z{
:return: path at which the object with the given hexsha would be stored,
relative to the database rootN� )r )r* �hexshar. r. r/ � object_pathV s zLooseObjectDB.object_pathc C s` t t��
| j| W d � S 1 sw Y | �| �|��}t|�r,|| j|< |S t|��)z�
:return: readable object path to the object identified by hexsha
:raise BadObject: If the object file does not existN)r �KeyErrorr& �db_pathr2 r r )r* r1 �pathr. r. r/ �readable_db_object_path\ s
�
z%LooseObjectDB.readable_db_object_pathc C sL d}| � � D ]}t|��t|��r|durt|��|}q|du r$t|��|S )z�:return: 20 byte binary sha1 string which matches the given name uniquely
:param name: hexadecimal partial name (bytes or ascii string)
:raise AmbiguousObjectName:
:raise BadObject: N)�sha_iterr �
startswithr r r )r* Zpartial_hexsha� candidate�binshar. r. r/ �partial_to_complete_sha_hexl s �z)LooseObjectDB.partial_to_complete_sha_hexc
C s� | � | �t|���}zt|| jd�W S tyE } z'|jtkr<z t|�W W Y d}~S ty; } zt|�|�d}~ww t|�|�d}~ww )z~
:return: memory map of that file to allow random read access
:raise BadObject: if object could not be located)�flagsN) r4 r2 r r r) �OSError�errnor r )r* �shar4 �eZnew_er. r. r/ �_map_loose_object s
��
��z LooseObjectDB._map_loose_objectc s, |durt |t�stdtj ��t� �|�S )zI:raise TypeError: if the stream does not support the Sha1Writer interfaceNz,Output stream musst support the %s interface)�
isinstancer � TypeError�__name__r# � set_ostream)r* �streamr, r. r/ rE � s zLooseObjectDB.set_ostreamc C sP | � |�}zt|�\}}t|||�W t|d�r |�� S S t|d�r'|�� w w )N�close)rA r r
�hasattrrG )r* r? �m�typ�sizer. r. r/ �info� s
�
�zLooseObjectDB.infoc C s, | � |�}tj|dd�\}}}t||||�S )NT)Zclose_on_deletion)rA r �newr )r* r? rI �typerK rF r. r. r/ rF � s
zLooseObjectDB.streamc C s* z
| � t|�� W dS ty Y dS w )NTF)r6 r r )r* r? r. r. r/ �
has_object� s �zLooseObjectDB.has_objectc C s� d}| � � }|du r"tjd| jd�\}}|jdu r t|�}nt|�}z2z'|jdur5t|j|j t
j | j � nt
|j|j|j|j | j d� W |rJ|�� n|rR|�� w w W n
|r^t|� � d}|jrh|j}n|jdd�}|r�| �| �|��}t|�}tj|dd� t|�r�t|� nt||� dD ]%}tt �� t || j � W d � n1 s�w Y t � |d � q�t d
�!|���t"|�|_|S ) z.note: The sha we produce will be hex by natureN�obj)�prefix�dir)r T)Zas_hex)�exist_ok)� � � � � �$ �1 �@ �Q �d �y � � � g @�@z&Impossible to apply `chmod` to file {})#Zostream�tempfileZmkstempZ
_root_pathr: r r
r �read�write�sys�maxsize�stream_chunk_sizer rN rK rG r r1 r? r4 r2 r r( �makedirsr r r �PermissionErrorr �new_objects_mode�time�sleep�formatr ) r* ZistreamZtmp_path�writer�fdr1 Zobj_pathZobj_dirZexp_backoff_msr. r. r/ �store� s`
����
��
��
zLooseObjectDB.storec c s\ � t �| �� �D ]#\}}}t|�}t|�dkrq|D ]}t|�dkr#qt|| �V qqd S )Nr0 �& )r( �walkr+ r �lenr )r* �root�dirs�filesZ root_base�fr. r. r/ r7 � s � ��zLooseObjectDB.sha_iterc C s t t| �� ��S )N)rs �tupler7 )r* r. r. r/ rK s zLooseObjectDB.size)rD �
__module__� __qualname__�__doc__r rg �intrj r( �namer$ r2 r6 r; rA rE rL rF rO rp r7 rK �
__classcell__r. r. r, r/ r ? s$
Lr )+�
contextlibr Z
gitdb.db.baser r r Z gitdb.excr r Z gitdb.streamr r r
r Z
gitdb.baser r
Z
gitdb.utilr r r r r r r r r r r r Z gitdb.funr r r r Zgitdb.utils.encodingr rb r( re rk �__all__r r. r. r. r/ �<module> s 8