|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/hgext/lfs/__pycache__/ |
Upload File : |
o
�]Lb3C � @ s� d dl mZ d dlZd dlmZ d dlmZmZmZ d dl m
Z
m Z d dl m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZ d dl m Z m Z d dl!m"Z" d d l#m$Z$m%Z% d
d l&m'Z' d d
l(m)Z)m*Z* e�� Z+e+�,ed�dd� �Z-e+�,ed�dd� �Z.e+�,ed�dd� �Z/dd� Z0dd� Z1dd � Z2dYd d �Z3ddej4fd d �Z5d!d"� Z6d#d$� Z7e+�,ed%�d&d'� �Z8e+�,ej9d(�d)d*� �Z:e+�,ej9d+�d,d-� �Z;d.d/� Z<e+�,ed0�d1d2� �Z=e+�,ed3�d4d5� �Z>e+�,ej?d6�d7d8� �Z@d9d:� ZAd;d<� ZBd=d>� ZCd?d@� ZDdAdB� ZEe+�,edC�dDdE� �ZFe+�,e
dF�dGdH� �ZGdIdJ� ZHdZdLdM�ZIdZdNdO�ZJdPdQ� ZKe+�,e dR�dSdT� �ZLe+�,e dU�e+�,e dV�dWdX� ��ZMdS )[� )�absolute_importN)�_)�bin�hex�short)�getattr�setattr)�bundle2� changegroup�cmdutil�context�error�exchange� exthelper� localrepo�pycompat�revlog�scmutil�util�vfs�wireprotov1server)�actions�engine)�
repository)� storageutil�
stringutil� )�lfutil� )� blobstore�pointers makefilestoragec K s( d|v r
|� tj� | d||d�|��S )N� lfs)� requirements�features� )�addr �REPO_FEATURE_LFS)�origr" r# �kwargsr$ r$ �3/usr/lib/python3/dist-packages/hgext/lfs/wrapper.py�localrepomakefilestorage: s r* s allsupportedversionsc C s | |�}|� d� |S )Ns 03)r% )r' �ui�versionsr$ r$ r) �allsupportedversionsB s
r- s
_capabilitiesc C s: | ||�}t �|jd�rd|jv r|�d� |�d� |S )z5Wrap server command to announce lfs server capability� lfslocalblobstorer! s lfs-serve)r � safehasattr�svfsr" �append)r' �repo�proto�capsr$ r$ r) �
_capabilitiesI s
r5 c C � dS �NFr$ )�self�textr$ r$ r) �bypasscheckhashY s r: c C s� t �|�}|�� }| jj}|�|�s | j|_| jj�|g|� |j |dd�}i }|�
� D ]}|� d�rA|t d�d� }|| ||< q,|sI|� d�rOt
�||�}|dfS )a% Read filelog content from local blobstore transform for flagprocessor.
Default tranform for flagprocessor, returning contents from blobstore.
Returns a 2-typle (text, validatehash) where validatehash is True as the
contents of the blobstore should be checked using checkhash.
F)�verifys x-hg-Ns
T)r � deserialize�oid�opener�lfslocalblobstore�has�filename�lfsremoteblobstore� readbatch�read�keys�
startswith�lenr �packmeta)r8 r9 �pr= �store�hgmeta�k�namer$ r$ r) �
readfromstore] s
� rN c
C s� t �|�\}}|r|dkr||d � }tt�|��� �}| jj�||� d| }t j
|dt |� d�}t �
|�s<d|d< |d urPt�|�D ]
\}}||d| < qE|�� } | dfS ) Nr s sha256:%ss %d)r= �size� 0� x-is-binarys x-hg-%sF)r � parsemetar �hashlib�sha256�digestr> r? �writer �
gitlfspointerrG r �binaryr � iteritems� serialize)
r8 r9 rK �offsetr= �longoid�metadatarL �v�rawtextr$ r$ r) � writetostorez s
r` c C sP |d u r|d u r
dS | � |�}n| �|�}|| jkr dS | �|�}t|tj@ �S r7 )�rev�node�nullid�flags�boolr �REVIDX_EXTSTORED)�rlogrb ra rd r$ r$ r) �_islfs� s
rh c
K sp |j jj�d�} | r't|�} t�|�\}
}|r| |8 } | |j j| �r'| tj O } | ||||||f||| d�|
��S )Ns lfstrack)�
cachedeltarb rd )
�_revlogr> �options�getrG r rR rA r rf )r' r8 r9 � transaction�link�p1�p2ri rb rd �kwds�lfstrack�textlen�metar[ r$ r$ r) �filelogaddrevision� s,
��
�ru c C s\ t |j|�r)|j�|�}|sdS t�|�}d|v r'd|v r'|d t|d �fS dS | ||�S )NFs x-hg-copys x-hg-copyrev)rh rj �rawdatar r< r )r' r8 rb r_ r] r$ r$ r) �filelogrenamed� s
rw c C s: t |j|d�r|j�|�}t�|�}t|d �S | ||�S )N)ra s size)rh rj rv r r< �int)r' r8 ra r_ r] r$ r$ r) � filelogsize� s
ry s _verify_revisionc C sj t ||d�r,|�|�}t�|�}|jj�|�� �r |tj M }n |tj @ r,|d �
|� | ||||� d S )N)rb s safe_renamed) rh rv r r< r> r? r@ r= r rf r% )r' �rl� skipflags�staterb r_ r] r$ r$ r) �_verify_revision� s
r} s cmpc C sP |� � r#t|ddd� �� r#t�|�� �}t�|�� �}|�� |�� kS | ||�S )z+returns True if text is different than fctx�islfsc S r6 r7 r$ r$ r$ r$ r) �<lambda>� s z filectxcmp.<locals>.<lambda>)r~ r r r< rv r= )r' r8 �fctxro rp r$ r$ r) �
filectxcmp� s
r� s isbinaryc C s2 |� � rt�|�� �}tt|�dd���S | |�S )NrQ r )r~ r r< rv re rx rl )r' r8 r] r$ r$ r) �filectxisbinary s r� c C s t | �� j| �� �S �N)rh �filelogrj �filenode)r8 r$ r$ r) � filectxislfs s r� s _updatecatformatterc C s( | |||||� |j || �� d� d S )N)rv )�datarv )r' �fm�ctx�matcher�path�decoder$ r$ r) �_updatecatformatter s r� s wrapconvertsinkc s2 | |�}|j dkrG � fdd�d|j�� � |_|S )Ns hgc s e Zd Z� �fdd�Z� ZS )z convertsink.<locals>.lfssinkc sl t �| �j} | ||||||||�}
d| jjvr4| j|
� t� fdd�|D ��r4| jj�d� t�| j� |
S )Nr! c 3 s( � | ]\}}|� v o� | � � V qd S r� )r~ )�.0�f�n�r� r$ r) � <genexpr>= s �& z9convertsink.<locals>.lfssink.putcommit.<locals>.<genexpr>)�super� putcommitr2 r" �anyr% r �writereporequirements) r8 �files�copies�parents�commit�source�revmap�full�cleanp2�pcrb )� __class__�lfssinkr� r) r� ! s" �
z&convertsink.<locals>.lfssink.putcommit)�__name__�
__module__� __qualname__r� �
__classcell__r$ �r� )r� r) r� s r� )�repotyper� )r' �sinkr$ r� r) � convertsink s
#r� s __init__c C s^ | ||� |j �� D ]\}}|�d�r||j |< q
dD ]}t�||�r,t||t||�� qd S )Nr! )r. � lfsremoteblobstore)rk �itemsrF r r/ r r )r' r8 �othervfsrL r^ rM r$ r$ r) �vfsinitJ s
� ��r� c
C s� t �| jd�s dS g }t� }| jj}|D ]3\}}| | }|�|�D ]%}t||�} | rF| �� |vrF|�| �� �sF|| _ |�
| � |� | �� � q!q|rUt �
| ��||� dS dS )zSEnsure that required LFS blobs are present, fetching them as a group if
needed.r. N)r r/ r0 �setr? �walk�pointerfromctxr= r@ rA r1 r% r �remoterC )
r2 �
revmatches�pointers�oids�
localstorera �matchr� r� rI r$ r$ r) �_prefetchfilesX s$
���r� c C s" t �| jd�s dS t| jjtj�S )Nr� T�r r/ r0 �
isinstancerB r � _nullremote�r2 r$ r$ r) �_canskipuploadq s r� c C s$ t �| jd�s dS t| jjtj� S )Nr� Fr� r� r$ r$ r) � candownloadz s r� c C s$ t | �rdS t| |�}t| |� dS )zuupload lfs blobs introduced by revs
Note: also used by other extensions e. g. infinitepush. avoid renaming.
N)r� �extractpointers� uploadblobs)r2 �revsr� r$ r$ r) �uploadblobsfromrevs� s
r� c C s t | j| jj�S )z�Prepush hook.
Read through the revisions to push, looking for filelog entries that can be
deserialized into metadata so that we can block the push on their upload to
the remote blobstore.
)r� r2 �outgoing�missing)�pushopr$ r$ r) �prepush� s r� s pushc O s� d|j v rB|�� s |�d�s td�}tj|d td�d��|jj}zt� ||�
� �|j_| ||g|�R i |��W ||j_S ||j_w | ||g|�R i |��S )z�bail on push if the extension isn't enabled on remote when needed, and
update the remote store based on the destination path.r! s: required features are not supported in the destination: %ss& enable the lfs extension on the server)�hint) r" �local�capabler r
�Abortr0 rB r r� �url)r' r2 r� �argsr( �m� remotestorer$ r$ r) �push� s
�r� s writenewbundlec O s, t ||j� | ||||||g|�R i |��S )z;upload LFS blobs added by outgoing revisions on 'hg bundle')r� r� ) r' r+ r2 r� rA �
bundletyper� r� r( r$ r$ r) �writenewbundle� s ���r� c C s� | j �d� i }| j j}|td�td�t|���.}|D ]}| | }t|��� D ]}|||�� < q'|�� qt |�� dd� d�W d � S 1 sIw Y dS )z1return a list of lfs pointers added by given revss& lfs: computing set of blobs to upload
s
lfs searchs
changesetsc S s | � � S r� )r= )rI r$ r$ r) r � s z!extractpointers.<locals>.<lambda>)�keyN)
r+ �debug� makeprogressr rG �pointersfromctx�valuesr= � increment�sorted)r2 r� r� r� �progress�rr� rI r$ r$ r) r� � s �
$�r� Fc
C s� | }|| vr"|s
dS || � � v r| � � }n
|| �� v r | �� }ndS || }t|�� j|�� �s2dS zt�|�� �}| |krA|W S i W S tj y` } zt
� t d�|t
|�� �|f ��d}~ww )a? return a pointer for the named file from the given changectx, or None if
the file isn't LFS.
Optionally, the pointer for a file deleted from the context can be returned.
Since no such pointer is actually stored, and to distinguish from a non LFS
file, this pointer is represented by an empty dict.
Ns# lfs: corrupted pointer (%s@%s): %s
)ro rp rh r� rj r� r r< rv �InvalidPointerr
r� r r rb )r� r� �removed�_ctxr� rI �exr$ r$ r) r� � s2
����r� c C sJ i }| � � �� }| �� D ]}||�sq t| ||d�}|dur"|||< q |S )z�return a dict {path: pointer} for given single changectx.
If ``removed`` == True and the LFS file was removed from ``ctx``, the value
stored for the path is an empty dict.
)r� N)r2 � narrowmatchr� r� )r� r� �resultr� r� rI r$ r$ r) r� � s �r� c C s$ |sdS | j j}|�|| j j� dS )z*upload given pointers from local blobstoreN)r0 rB �
writebatchr? )r2 r� �
remoteblobr$ r$ r) r� s r� s finishdatamigrationc C s� | ||||� t �|jd�rCt �|jd�rE|jjj}|jjj}|�� D ]#\}}} | D ]}
|�td�|
� t� |�
|
�|�
|
�� q*q#d S d S d S )Nr. s copying lfs blob %s
) r r/ r0 r? r r� rV r r rn �join) r' r+ �srcrepo�dstrepor" � srclfsvfs� dstlfsvfs�dirpath�dirsr� r= r$ r$ r) �upgradefinishdatamigration s � �
��r� s preservedrequirementss supporteddestrequirementsc C s | |�}d|j v r|�d� |S )Nr! )r" r% )r' r2 �reqsr$ r$ r) �upgraderequirements s
r� )NN)F)N�
__future__r rS �mercurial.i18nr �mercurial.noder r r �mercurial.pycompatr r � mercurialr r
r r r
r r r r r r r r �vfsmodr �mercurial.upgrade_utilsr �upgrade_actionsr �upgrade_engine�mercurial.interfacesr �mercurial.utilsr r �
largefilesr � r r �eh� wrapfunctionr* r- r5 r: rN r` rh �REVIDX_DEFAULT_FLAGSru rw ry r} � basefilectxr� r� r� r� r� � readonlyvfsr� r� r� r� r� r� r� r� r� r� r� r� r� r� r$ r$ r$ r) �<module> sv @
�)
.
!