File: //usr/lib/python3/dist-packages/mercurial/__pycache__/windows.cpython-310.pyc
o
�]LbW � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z ddl
m Z ddl m
Z
ddlmZmZmZm Z mZ z d dlZej W n
eyV d dlZY nw e�d�ZejZejZej Z ej Z ej Z ej Z ej Z ej!Z!ej"Z"ej#Z#ej$Z$ej%j&Z&ej'Z'ej(Z(dZ)G d d
� d
e*�Z+G d d � d e*�Z,dcdd�Z-ej.Z.dd� Z/dd� Z0G dd� de*�Z1dd� Z2dd� Z3dd � Z4d d � Z5ddd d!�Z6d"d#� Z7d$d%� Z8d&d'� Z9d(d)� Z:d*d+� Z;d,d-� Z<d.d/� Z=e�>d0�Z?e�>d1�Z@d2d3� ZAejBjCZDejEZFd4d5� ZGd6d7� ZHdaIdaJd8d9� ZKd:d;� ZLd<d=� ZMd>d?� ZNd@dA� ZOejPejQhZRdBdC� ZSdedDdE�ZTdedFdG�ZUdHdI� ZVdJdK� ZWdLdM� ZXdNdO� ZYdPdQ� ZZdRdS� Z[G dTdU� dUe*�Z\dfdVdW�Z]dXZ^dYdZ� Z_d[d\� Z`d]d^� Zad_d`� Zbdadb� ZcdS )g� )�absolute_importN� )�_)�getattr)�encoding�error�policy�pycompat�win32�osutil� c @ s| e Zd ZdZdZdZdZdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd � Zd S ) �mixedfilemodewrappera) Wraps a file handle when it is opened in read/write mode.
fopen() and fdopen() on Windows have a specific-to-Windows requirement
that files opened with mode r+, w+, or a+ make a call to a file positioning
function when switching between reads and writes. Without this extra call,
Python will raise a not very intuitive "IOError: [Errno 0] Error."
This class wraps posixfile instances when the file is opened in read/write
mode and automatically adds checks or inserts appropriate file positioning
calls when necessary.
r r � c C s t �| d|� t �| dd� d S )N�_fp�_lastopr )�object� __setattr__��self�fp� r �3/usr/lib/python3/dist-packages/mercurial/windows.py�__init__J s z mixedfilemodewrapper.__init__c C � | j �� | S �N�r � __enter__�r r r r r N s
z mixedfilemodewrapper.__enter__c C � | j �|||� d S r �r �__exit__)r �exc_type�exc_val�exc_tbr r r r R � z mixedfilemodewrapper.__exit__c C � t | j|�S r �r r �r �namer r r � __getattr__U � z mixedfilemodewrapper.__getattr__c C s | j �||�S r )r r )r r( �valuer r r r X � z mixedfilemodewrapper.__setattr__c C s | j �dtj� d S )Nr )r �seek�os�SEEK_CURr r r r � _noopseek[ r$ z mixedfilemodewrapper._noopseekc O s" t �| d| j� | jj|i |��S �Nr )r r �OPNONEr r- �r �args�kwargsr r r r- ^ s zmixedfilemodewrapper.seekc C s0 | j | jkr
| �� t�| d| j� | j�|�S r1 )r �OPREADr0 r r �OPWRITEr �write)r �dr r r r8 b s zmixedfilemodewrapper.writec O �6 | j | jkr
| �� t�| d| j� | jj|i |��S r1 )r r6 �
_noopeseekr r r7 r �
writelinesr3 r r r r< i � z mixedfilemodewrapper.writelinesc O r: r1 )r r7 r0 r r r6 r �readr3 r r r r> p r= zmixedfilemodewrapper.readc O r: r1 )r r7 r0 r r r6 r �readliner3 r r r r? w r= z mixedfilemodewrapper.readlinec O r: r1 )r r7 r0 r r r6 r � readlinesr3 r r r r@ ~ r= z mixedfilemodewrapper.readlinesN)�__name__�
__module__� __qualname__�__doc__r2 r6 r7 r r r r) r r0 r- r8 r< r> r? r@ r r r r r
9 s" r
c @ �8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�fdproxyzeWraps osutil.posixfile() to override the name attribute to reflect the
underlying file name.
c C s || _ || _d S r )r( r )r r( r r r r r � s
zfdproxy.__init__c C r r r r r r r r � s
zfdproxy.__enter__c C r r r )r r! � exc_value� tracebackr r r r � r$ zfdproxy.__exit__c C s
t | j�S r )�iterr r r r r �__iter__� �
zfdproxy.__iter__c C r% r r& r' r r r r) � r* zfdproxy.__getattr__N) rA rB rC rD r r r rJ r) r r r r rF � s rF � r���c
C s� z&t �| ||�}tjrt| |�}d|v r|�dtj� d|v r$t|�W S |W S t y@ } zt
|j dt �
| �|jf ��d}~ww )z/Open a file with even more POSIX-like semantics� ar � +z%s: %sN)r � posixfiler �ispy3rF r- r. �SEEK_ENDr
� WindowsError�IOError�errnor � strfromlocal�strerror)r( �mode� bufferingr �errr r r rP � s
���rP c C s z| � � W S ty Y dS w �NF)�isatty�AttributeError)r r r r �_isatty� s
�r^ c C sj d} t �� }|dks|dkrn|dkrt�|dkr!| dd� } n| | } qt �d� t �d� t�| �S ) z�Prompt for password with echo off, using Windows getch().
This shouldn't be called directly- use ``ui.getpass()`` instead, which
checks if the session is interactive first.
� T�
�
��NrM )�msvcrt�getwch�KeyboardInterrupt�putwchr �
unitolocal)�pw�cr r r � get_password� s �
rk c @ rE )
� winstdoutz�Some files on Windows misbehave.
When writing to a broken pipe, EINVAL instead of EPIPE may be raised.
When writing too many bytes to a console at the same, a "Not enough space"
error may happen. Python 3 already works around that.
c C s || _ tj o
t|�| _d S r )r r rQ r^ �throttler r r r r � s zwinstdout.__init__c C r% r )r r )r �keyr r r r) � r* zwinstdout.__getattr__c C s& z| j �� W d S ty Y d S w r )r �closerT r r r r ro � s
�zwinstdout.closec
C s� z0| j s | j�|�W S d}t|�}d}||k r.|| }| j�|||� � |}||k sW d S W d S tyQ } z|jdkrCt�|�sC� | �� ttj d��d }~ww )Ni�> r � Broken pipe)
rm r r8 �lenrT rU r
�lasterrorwaspipeerrorro �EPIPE)r �s�limit�l�start�end�instr r r r8 � s$ � ��zwinstdout.writec
C s> z| j �� W S ty } z t�|�s� ttjd��d }~ww )Nrp )r �flushrT r
rr rU rs )r ry r r r rz s
��zwinstdout.flushN) rA rB rC rD r r) ro r8 rz r r r r rl � s rl c C � dS �NTr r r r r �
openhardlinks
� r} c C s( | dd� }|d dkr|dd� }|S )z<parses the output produced by patch and returns the filename� Nr � `r rM r )� output_line�pfr r r �parsepatchoutput s r� c C sp d| � � v rdp d}|rd||f p|}|�d�s |�d�r't�td�| ��t|�}|r6d|t|�|f }|S ) z$Build argument list for ssh or Plinks plinks -Ps -ps %s@%s� -� /s9 illegal ssh hostname or username starting with - or /: %ss %s %s %s)�lower�
startswithr �Abortr �
shellquote)�sshcmd�host�user�port�pflagr4 r r r �sshargs s ��r� c C � d S r r )�frv �xr r r �setflags( r~ r� Fc C r� r r )�src�dstrX �enforcewritabler r r �copymode, r~ r� c C r{ r[ r ��pathr r r � checkexec0 r~ r� c C r{ r[ r r� r r r � checklink4 r~ r� c C s: t | dd �}|d ur|� dkrt�|� tj� d S d S d S )N�filenor )r rd �setmoder. �O_BINARY)�fd�fnor r r � setbinary8 s �r� c C s | � tjd�S )Nr� )�replacer �ossepr� r r r �pconvert@ r, r� c C s | � dd�S )Nr� � \)r� r� r r r � localpathD r* r� c C s t tj�| ��S r )r� r. r� �normpathr� r r r r� H s r� c C s
t �| �S r )r �upperr� r r r �normcaseL rK r� s ^[a-z]:z^[a-z]:c C sj t j�| �}t|t�r t�|�r |dd� �� |dd � }|S t�|�r3|dd� �� |dd � }|S )Nr r ) r. r� �abspath�
isinstance�bytes�
DRIVE_RE_B�matchr� �
DRIVE_RE_S)r� �abs_pathr r r r� T s
� r� c C r{ r[ r )�s1�s2r r r �samestath r~ r� c sD t � fdd�dD ��s
� S t�tjtj �d }d}d}t� �}||k �r�� ||d � }|dkrf� |d d � � t� �}z� �d�}|d
� d |� d
7 }W �nF tye ||� 7 }|d }Y �n3w |d kr�� |d d � � t� �}z� �d �}W n ty� |d � 7 }|d }Y �nw � d |� }|d | d 7 }n�|d k�rL� |d |d
� dkr�� |d
d � � t� �}z&� �d�}� d |� }|� |d�dkr�|d | d 7 }n|d| d 7 }W n� ty� |d� 7 }|d }Y n�w d}|d7 }� ||d � }|dk�r*||v �r*||7 }|d7 }� ||d � }|dk�r*||v �s|� |d�dk�r<|d | d 7 }n|d | 7 }|dk�rK|d8 }nL|dk�rj|d |k �rj� |d |d
� dv �rj|d7 }n.|dk�r�|d |k �r�� |d |d
� dv �r�|� |d |d
� 7 }|d7 }n||7 }|d7 }||k s%|S )a� Convert shell variables in the form $var and ${var} inside ``path``
to %var% form. Existing Windows style variables are left unchanged.
The variables are limited to the given environment. Unknown variables are
left unchanged.
>>> e = {b'var1': b'v1', b'var2': b'v2', b'var3': b'v3'}
>>> # Only valid values are expanded
>>> shelltocmdexe(b'cmd $var1 ${var2} %var3% $missing ${missing} %missing%',
... e)
'cmd %var1% %var2% %var3% $missing ${missing} %missing%'
>>> # Single quote prevents expansion, as does \$ escaping
>>> shelltocmdexe(b"cmd '$var1 ${var2} %var3%' \$var1 \${var2} \\", e)
'cmd "$var1 ${var2} %var3%" $var1 ${var2} \\'
>>> # $$ is not special. %% is not special either, but can be the end and
>>> # start of consecutive variables
>>> shelltocmdexe(b"cmd $$ %% %var1%%var2%", e)
'cmd $$ %% %var1%%var2%'
>>> # No double substitution
>>> shelltocmdexe(b"$var1 %var1%", {b'var1': b'%var2%', b'var2': b'boom'})
'%var1% %var1%'
>>> # Tilde expansion
>>> shelltocmdexe(b"~/dir ~\dir2 ~tmpfile \~/", {})
'%USERPROFILE%/dir %USERPROFILE%\\dir2 ~tmpfile ~/'
c 3 s � | ]}|� v V qd S r r )�.0rj r� r r � <genexpr>� s � z shelltocmdexe.<locals>.<genexpr>s $'~s _-� r r � 'N� "� %� $r � {� }s ${� ~)r� r� s
%USERPROFILE%r� )r� r� )
�anyr �sysbytes�string�
ascii_letters�digitsrq �index�
ValueError�get)r� �env�varchars�resr� �pathlenrj �varr r� r �
shelltocmdexel s�
� �
� ��
�
�Mr� c C sP t du r t�d�a tdu rt�d�ja| r t| �s t �| �s | S dt �d| � S )a�
>>> shellquote(br'C:\Users\xyz')
'"C:\\Users\\xyz"'
>>> shellquote(br'C:\Users\xyz/mixed')
'"C:\\Users\\xyz/mixed"'
>>> # Would be safe not to quote too, since it is all double backslashes
>>> shellquote(br'C:\\Users\\xyz')
'"C:\\\\Users\\\\xyz"'
>>> # But this must be quoted
>>> shellquote(br'C:\\Users\\xyz/abc')
'"C:\\\\Users\\\\xyz/abc"'
Ns (\\*)("|\\$)s [^a-zA-Z0-9._:/-]s "%s"s \1\1\\\2)�_quotere�re�compile�_needsshellquote�search�sub�rt r r r r� � s
r� c C s$ | � d�r| �d�r| dd� S | S )Nr� r rM )r� �endswithr� r r r �_unquote s r� c C s t �tt j| dd��S )z3Parse a command string in cmd.exe way (best-effort)F)�posix)r �maplistr� �
shlexsplitr� r r r �
shellsplit s r� c C r{ r| r ��str r r �isowner r~ r� c s� t j�dd�}dd� |�� �tj�D �� tj� | �d �� � v r#dg� � fdd�}tj
| v r2|| �S t j�d d��tj�D ]}|tj� || ��}|d
urP| S q=|tj� tj�
| ���S ) a* Find executable for command searching like cmd.exe does.
If command is a basename then PATH is searched for command.
PATH isn't searched if command is an absolute or relative path.
An extension from PATHEXT is found and added if not present.
If command isn't found None is returned.s PATHEXTs .COM;.EXE;.BAT;.CMDc S s g | ]}|�qS r r )r� �extr r r �
<listcomp># s zfindexe.<locals>.<listcomp>r r� c s* � D ]}| | }t j�|�r| S qdS )z:Will append extension (if needed) and return existing fileN)r. r� �exists)� pathcommandr� �
executable��pathextsr r � findexisting' s �z findexe.<locals>.findexistings PATHN)r �environr� r� �splitr � ospathsepr. r� �splitextr� �join�
expanduser�
expandvars)�command�pathextr� r� r� r r� r �findexe s
�r� c # s� � i }t j� | D ]X}t|�}tj�|�\}}|sd}|�|d�}|du rYz� fdd�t|d�D �}W n tyR } z|j t j
t j t j fvrF� i }W Y d}~nd}~ww |�
||�}|�|d�V qdS )z�Stat each file in files. Yield each stat, or None if a file
does not exist or has a type we don't care about.
Cluster and cache stat per directory to minimize number of OS stat calls.� .Nc s* i | ]\}}}� |j �tv rt|�|�qS r )�st_mode� _wantedkindsr� )r� �n�krt ��getkindr r �
<dictcomp>K s
�z statfiles.<locals>.<dictcomp>T)�stat�S_IFMTr� r. r� r� r� �listdir�OSErrorrU �ENOENT�EINVAL�ENOTDIR�
setdefault)�files�dircache�nf�dir�base�cache�dmaprZ r r� r � statfiles< s. �
��� �r c C s | s t �t�� �S dS )ziReturn the name of the user with the given uid.
If uid is None, return the name of the current user.N)r �fsencode�getpass�getuser)�uidr r r �usernameZ s r
c C r{ )zkReturn the name of the group with the given gid.
If gid is None, return the name of the current group.Nr )�gidr r r � groupnamec s r c
C sD t �| �}zt�|�}W n ty } zt|��d }~ww t �|�S r )r �fsdecoder. �readlinkr� r] r )�pathnamer� �link�er r r r j s
��
r c C s� t | �rdS t�| � tj�| �\}}|s tj�|�\}}|rJ|rLzt |�r)W dS t�|� W n ttfy; Y dS w tj�|�\}}|rN|s!dS dS dS dS )z�special version of os.removedirs that does not remove symlinked
directories or junction points if they actually contain filesN)r� r. �rmdirr� r� r� r� )r( �head�tailr r r �
removedirsw s
��r c
C s\ z t �| |� W dS ty- } z|jtjkr� t|� t �| |� W Y d}~dS d}~ww )z=atomically rename file src to dst, replacing dst if it existsN)r. �renamer� rU �EEXIST�unlink)r� r� r r r r r � s ��r c C s dd� t jgt jd d� D �S )Nc S s g | ]}t �|��qS r )r �
strtolocal)r� �argr r r r� � s z gethgcmd.<locals>.<listcomp>r )�sysr� �argvr r r r �gethgcmd� s r c C s t �r )�KeyError)r( r r r � groupmembers� � r c C r{ r[ r )r� r r r �isexec� r~ r! c @ s e Zd Zdd� Zdd� ZdS )� cachestatc C r� r r )r r� r r r r � r~ zcachestat.__init__c C r{ r[ r r r r r � cacheable� r~ zcachestat.cacheableN)rA rB rC r r# r r r r r" � s r" c C s� |du r t jt jf}n
t|ttf�s|f}|D ]?}z4t �|t�| ���!}|r+t�|�p,|}t � ||�d }t�
|�W d � W S 1 sGw Y W q t yV Y qw dS )a) Look up a key/value name in the Windows registry.
valname: value name. If unspecified, the default value for the key
is used.
scope: optionally specify scope for registry lookup, this can be
a sequence of scopes to look up in order. Default (CURRENT_USER,
LOCAL_MACHINE).
Nr ) �winreg�HKEY_CURRENT_USER�HKEY_LOCAL_MACHINEr� �list�tuple�OpenKeyr rV � QueryValueExrh �EnvironmentError)rn �valname�scopert �hkeyr( �valr r r � lookupreg� s *� ��r0 Tc C r{ )z(check whether a stat result is a symlinkFr r� r r r �
statislink� r r1 c C r{ )z1check whether a stat result is an executable fileFr r� r r r �
statisexec� r r2 c C s t � �r ��NotImplementedError)�fdsr r r �poll� s r6 c C s< g } t �| �}|s n| �|�}|sn|�|� qd�|�S )z$Read all available data from a pipe.Tr� )r
�peekpiper> �appendr� )�pipe�chunks�sizert r r r �readpipe� s
�
r< c C s t d��)Nzunsupported platformr3 )�sockr� r r r �bindunixsocket� s r> )rL rM r[ r )NN)d�
__future__r rU r rd r. r� r� r� r �i18nr r r r_ r r r r
�_winregr$ �CloseKey� ImportError� importmodr �
getvolumename�getfsmountpoint� getfstyper �
hidewindow�makedir�nlinks�oslink�
samedevice�samefile�setsignalhandler�
spawndetachedr� r� �testpidr �umaskr r
rF rP r� r^ rk rl r} r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �
normcasespecsr� � normcasespec�
upperfallback�normcasefallbackr� r� r� r� r� r� r� r� r� �S_IFREG�S_IFLNKr� r r
r r r r r r r! r" r0 � expandglobsr1 r2 r6 r< r> r r r r �<module> s�
�
M
2
}