File: //kunden/lib/python3/dist-packages/dulwich/__pycache__/fastexport.cpython-39.pyc
a
�~�_C" � @ sv d Z ddlmZ ddlmZmZmZmZ ddlm Z m
Z m Z m
Z
ddlZdd� ZG dd � d e�ZG d
d � d e
j�ZdS ) z!Fast export/import functionality.� )� commit_tree)�Blob�Commit�Tag�ZERO_SHA)�commands�errors�parser� processorNc C s | � dd�\}}||�d�fS )N� <� � >)�rsplit�rstrip)�text�nameZemail� r �4/usr/lib/python3/dist-packages/dulwich/fastexport.py� split_email+ s r c @ sT e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z ddd�Z
ddd�Z dS )�GitFastExporterz5Generate a fast-export output stream for Git objects.c C s || _ || _i | _d| _d S )Nr )�outf�store�markers� _marker_idx)�selfr r r r r �__init__3 s zGitFastExporter.__init__c C s | j �t|d|j�� d � d S )N� __bytes__�
)r �write�getattr�__repr__�r �cmdr r r � print_cmd9 s zGitFastExporter.print_cmdc C s | j d7 _ d| j f �d�S )Nr z%d�ascii)r �encode)r r r r �_allocate_marker< s z GitFastExporter._allocate_markerc C s& | � � }|j| j|< t�||j�|fS �N)r& �idr r Z BlobCommand�data)r �blob�markerr r r � _export_blob@ s z GitFastExporter._export_blobc C s | � |�\}}| �|� |S r' )r, r# )r r* r" r+ r r r � emit_blobE s
zGitFastExporter.emit_blobc c s� | j �||�D ]�\\}}\}}\}}|d u r:t�|�V qt�|�sX| j | } | �| �}
||krv|d urvt�||�V ||ks�||krd|
} t�||| d �V qd S )N� :) r Z tree_changesr ZFileDeleteCommand�stat�S_ISDIRr- ZFileRenameCommandZFileModifyCommand) r � base_treeZnew_tree�old_path�new_pathZold_modeZnew_modeZ
old_hexshaZ
new_hexshar* r+ Zprefixed_markerr r r � _iter_filesJ s �
�zGitFastExporter._iter_filesNc
C s� t | �||j��}| �� }|jr:|jd }|jdd � }nd }g }t|j�\}} t|j�\}
} t� |||| |j
|j f|
| |j |j
f|j|||�} | |fS )Nr r )�listr4 �treer& �parentsr �author� committerr Z
CommitCommand� author_time�author_timezone� commit_time�commit_timezone�message)
r �commit�refr1 Z file_cmdsr+ �from_�mergesr8 � author_emailr9 �committer_emailr" r r r �_export_commit\ s$
�
�z GitFastExporter._export_commitc C s | � |||�\}}| �|� |S r' )rE r# )r r? r@ r1 r" r+ r r r � emit_commito s
zGitFastExporter.emit_commit)N)N) �__name__�
__module__� __qualname__�__doc__r r# r& r, r- r4 rE rF r r r r r 0 s
r c @ sj e Zd ZdZddd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�GitImportProcessorzKAn import processor that imports into a Git repository using Dulwich.
NFc C s, t j�| ||� || _t| _i | _i | _d S r' )r
�ImportProcessorr �repor � last_commitr � _contents)r rM �params�verboser r r r r { s
zGitImportProcessor.__init__c C s |� d�r | j|dd � S |S )Nr. r )�
startswithr )r Z objectishr r r �
lookup_object� s
z GitImportProcessor.lookup_objectc C s t �|�}| �|j� | jS r' )r Z ImportParserZprocessZ
iter_commandsr )r �stream�pr r r �
import_stream� s
z GitImportProcessor.import_streamc C s2 t �|j�}| jj�|� |jr.|j| j|j< dS )zProcess a BlobCommand.N) r � from_stringr) rM � object_store�
add_object�markr( r )r r" r* r r r � blob_handler� s z GitImportProcessor.blob_handlerc C s dS )z Process a CheckpointCommand.Nr r! r r r �checkpoint_handler� s z%GitImportProcessor.checkpoint_handlerc C s0 t � }|jdur|j}n|j}|\}}}}|j\}} }
} |d | d |_||_t|�|_|d | d |_| |_t|
�|_|j|_g |_ |j
r�| � |j
�|_
| � |j
� |�
� D ]�} | jdk�r| jdur�t�| j�}
| jj�|
� |
j}n | � | j�}| j|f| j| j< q�| jdk�r&| j| j= q�| jdk�rH| j| j | j| j< q�| jdk�rt| j| j | j| j < | j| j = q�| jdk�r�i | _q�t d | j ��q�t | jjd
d � | j� � D ��|_!| j"t#k�r�|j �$| j"� |j%D ]}|j �$| � |�� �q�| jj�&|� |j| j|j'< |j| _"|j(�r,|j| j)|j(< dS ) zProcess a CommitCommand.Nr r
s
filemodifys
filedeletes filecopys
filerenames
filedeleteallzCommand %s not supportedc s s | ]\}\}}|||fV qd S r' r )�.0�path�mode�hexshar r r � <genexpr>� � z4GitImportProcessor.commit_handler.<locals>.<genexpr>)*r r8 r9 r; �intr: r= r< r> r7 rA rS � _reset_baseZ
iter_filesr r) r rW rM rX �addr( Zdatarefr_ rO r^ Zsrc_pathZ dest_pathr2 r3 � Exceptionr �itemsr6 rN r �appendrB rY r@ rZ r )r r" r? r8 Z author_namerC Zauthor_timestampr; Zcommitter_namerD Zcommit_timestampr= Zfilecmdr* Zblob_id�merger r r �commit_handler� sp
�
� � ��
z!GitImportProcessor.commit_handlerc C s dS )zProcess a ProgressCommand.Nr r! r r r �progress_handler� s z#GitImportProcessor.progress_handlerc C sZ | j |krd S i | _|| _ |tkrV| j| j}| jj�|�D ]\}}}||f| j|< q<d S r' )rN rO r rM r6 rX Ziter_tree_contents)r Z commit_idZtree_idr^ r_ r` r r r rd � s
� z GitImportProcessor._reset_basec C s8 |j du rt}n | �|j �}| �|� || jj|j< dS )zProcess a ResetCommand.N)rA r rS rd rM �refsr@ )r r" rA r r r �
reset_handler� s
z GitImportProcessor.reset_handlerc C sB t � }|j|_|j|_|j|_| j�|� |j| jjd|j < dS )zProcess a TagCommand.z
refs/tags/N) r Ztaggerr> �tagr rM rY r( rl )r r" rn r r r � tag_handler� s z GitImportProcessor.tag_handlerc C s t �|j��dS )zProcess a FeatureCommand.N)�fastimport_errorsZUnknownFeatureZ feature_namer! r r r �feature_handler� s z"GitImportProcessor.feature_handler)NFN)rG rH rI rJ r rS rV r[ r\ rj rk rd rm ro rq r r r r rK u s
9 rK )rJ Z
dulwich.indexr Zdulwich.objectsr r r r Z
fastimportr r rp r r
r/ r �objectr rL rK r r r r �<module> s E