U
    ƒIÀd¶F  ã                   @   sÈ   d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlm	Z	 ddl
mZmZmZ ddlmZ ddlmZmZmZmZmZmZ e e¡Zd ¡ Ze d	¡Zd
Zdd„ ZeZ G dd„ de!ƒZ"dS )é    )ÚBytesION)ÚZipInfoé   )Ú	sysconfigÚdetect_encodingÚZipFile)Úfinder)ÚFileOperatorÚget_export_entryÚconvert_pathÚget_executableÚget_platformÚin_venvañ  
<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
<assembly xmlns="urn:schemas-microsoft-com:asm.v1" manifestVersion="1.0">
 <assemblyIdentity version="1.0.0.0"
 processorArchitecture="X86"
 name="%s"
 type="win32"/>

 <!-- Identify the application security requirements. -->
 <trustInfo xmlns="urn:schemas-microsoft-com:asm.v3">
 <security>
 <requestedPrivileges>
 <requestedExecutionLevel level="asInvoker" uiAccess="false"/>
 </requestedPrivileges>
 </security>
 </trustInfo>
</assembly>s   ^#!.*pythonw?[0-9.]*([ 	].*)?$zÍ# -*- coding: utf-8 -*-
import re
import sys
from %(module)s import %(import_name)s
if __name__ == '__main__':
    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
    sys.exit(%(func)s())
c                 C   sX   d| krT|   d¡rB|  dd¡\}}d|krT|  d¡sTd||f } n|   d¡sTd|  } | S )Nú z/usr/bin/env r   ú"z%s "%s"z"%s")Ú
startswithÚsplit)Ú
executableÚenvÚ_executable© r   úT/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pip/_vendor/distlib/scripts.pyÚenquote_executable5   s    

r   c                   @   sæ   e Zd ZdZeZdZd*dd„Zdd„ Ze	j
 d	¡rBd
d„ Zdd„ Zdd„ Zd+dd„Zdd„ ZeZdd„ Zdd„ ZdZdd„ Zd,dd„Zdd„ Zed d!„ ƒZejd"d!„ ƒZejd#ksÆejd	krÎejd#krÎd$d%„ Zd-d&d'„Z d.d(d)„Z!dS )/ÚScriptMakerz_
    A class to copy or create scripts from source scripts or callable
    specifications.
    NTFc                 C   s‚   || _ || _|| _d| _d| _tjdkp:tjdko:tjdk| _t	dƒ| _
|pRt|ƒ| _tjdkprtjdkortjdk| _tj| _d S )NFÚposixÚjava)Ú úX.YÚnt)Ú
source_dirÚ
target_dirÚadd_launchersÚforceÚclobberÚosÚnameÚ_nameÚset_modeÚsetÚvariantsr	   Ú_fileopÚ_is_ntÚsysÚversion_info)Úselfr   r    r!   Údry_runÚfileopr   r   r   Ú__init__P   s    ÿ

ÿzScriptMaker.__init__c                 C   s@   |  dd¡r<| jr<tj |¡\}}| dd¡}tj ||¡}|S )NÚguiFÚpythonÚpythonw)Úgetr+   r$   Úpathr   ÚreplaceÚjoin)r.   r   ÚoptionsÚdnÚfnr   r   r   Ú_get_alternate_executablea   s
    z%ScriptMaker._get_alternate_executabler   c              
   C   s\   z0t |ƒ�}| d¡dkW  5 Q R £ W S Q R X W n& ttfk
rV   t d|¡ Y dS X dS )zl
            Determine if the specified executable is a script
            (contains a #! line)
            é   z#!zFailed to open %sFN)ÚopenÚreadÚOSErrorÚIOErrorÚloggerÚwarning)r.   r   Úfpr   r   r   Ú	_is_shelli   s    
&zScriptMaker._is_shellc                 C   sD   |   |¡r*dd l}|jj d¡dkr<|S n| ¡  d¡r<|S d| S )Nr   zos.nameÚLinuxz
jython.exez/usr/bin/env %s)rE   r   ÚlangÚSystemÚgetPropertyÚlowerÚendswith)r.   r   r   r   r   r   Ú_fix_jython_executableu   s    
z"ScriptMaker._fix_jython_executablec                 C   s‚   t jdkrd}n8t|ƒt|ƒ d }tjdkr4d}nd}d|koF||k}|r^d| | d	 }n d
}|d| | d 7 }|d7 }|S )a  
        Build a shebang line. In the simple case (on Windows, or a shebang line
        which is not too long or contains spaces) use a simple formulation for
        the shebang. Otherwise, use /bin/sh as the executable, with a contrived
        shebang which allows the script to run either under Python or sh, using
        suitable quoting. Thanks to Harald Nordgren for his input.

        See also: http://www.in-ulm.de/~mascheck/various/shebang/#length
                  https://hg.mozilla.org/mozilla-central/file/tip/mach
        r   Té   Údarwini   é   ó    s   #!ó   
s
   #!/bin/sh
s	   '''exec' s    "$0" "$@"
s   ' ''')r$   r%   Úlenr,   Úplatform)r.   r   Úpost_interpÚsimple_shebangÚshebang_lengthÚmax_shebang_lengthÚresultr   r   r   Ú_build_shebang�   s    

ÿzScriptMaker._build_shebangó    c                 C   s~  d}| j r| j }d}nˆt ¡ s&tƒ }nxtƒ rLtj t d¡dt 	d¡ ¡}nRtj t 	d¡dt 	d¡t 	d¡f ¡}tj 
|¡sžtj t 	d¡dt 	d¡ ¡}|r®|  ||¡}tj d	¡rÄ|  |¡}|rÐt|ƒ}| d
¡}tjdkrüd|krüd|krü|d7 }|  ||¡}z| d
¡ W n" tk
�r8   td| ƒ‚Y nX |d
k�rzz| |¡ W n& tk
�rx   td||f ƒ‚Y nX |S )NTFÚscriptszpython%sÚEXEÚBINDIRz
python%s%sÚVERSIONr   úutf-8Úcliz	-X:Framesz-X:FullFramess
    -X:Framesz,The shebang (%r) is not decodable from utf-8z?The shebang (%r) is not decodable from the script encoding (%r))r   r   Úis_python_buildr   r   r$   r6   r8   Úget_pathÚget_config_varÚisfiler<   r,   rS   r   rL   r   ÚencoderY   ÚdecodeÚUnicodeDecodeErrorÚ
ValueError)r.   ÚencodingrT   r9   Úenquoter   Úshebangr   r   r   Ú_get_shebang    s`    ÿ
ÿþÿ

ÿÿ

ÿÿ
zScriptMaker._get_shebangc                 C   s$   | j t|j|j d¡d |jd� S )NÚ.r   )ÚmoduleÚimport_nameÚfunc)Úscript_templateÚdictÚprefixÚsuffixr   )r.   Úentryr   r   r   Ú_get_script_textå   s    
þzScriptMaker._get_script_textc                 C   s   t j |¡}| j| S ©N)r$   r6   ÚbasenameÚmanifest)r.   ÚexenameÚbaser   r   r   Úget_manifestì   s    zScriptMaker.get_manifestc                 C   s6  | j o
| j}tj d¡}| |¡s*||7 }|s8|| }nš|dkrL|  d¡}n
|  d¡}tƒ }	t|	dƒ�R}
tj	 
d¡}|r¨t t|ƒ¡d d… }td|d�}|
 ||¡ n|
 d|¡ W 5 Q R X |	 ¡ }|| | }|D �]X}tj | j|¡}|�rºtj |¡\}}| d	¡�r|}d
| }z| j ||¡ W nŠ tk
�r¶   t d¡ d| }tj |¡�rht |¡ t ||¡ | j ||¡ t d¡ zt |¡ W n tk
�r°   Y nX Y nX nl| j�rÞ| d| ¡�sÞd||f }tj |¡�r| j�st d|¡ qÖ| j ||¡ | j �r&| j !|g¡ | "|¡ qÖd S )Nr_   ÚpyÚtÚwÚSOURCE_DATE_EPOCHé   z__main__.py)ÚfilenameÚ	date_timez.pyz%s.exez:Failed to write executable - trying to use .deleteme logicz%s.deletemez0Able to replace executable using .deleteme logicrm   z%s.%szSkipping existing file %s)#r!   r+   r$   Úlinesepre   rK   Ú_get_launcherr   r   Úenvironr5   ÚtimeÚgmtimeÚintr   ÚwritestrÚgetvaluer6   r8   r    Úsplitextr   r*   Úwrite_binary_fileÚ	ExceptionrB   rC   ÚexistsÚremoveÚrenameÚdebugr#   r'   Úset_executable_modeÚappend)r.   Únamesrk   Úscript_bytesÚ	filenamesÚextÚuse_launcherr„   ÚlauncherÚstreamÚzfÚsource_date_epochrƒ   ÚzinfoÚzip_datar%   ÚoutnameÚnÚeÚdfnamer   r   r   Ú_write_scriptð   sb    






zScriptMaker._write_scriptú-c                 C   sn   t ƒ }d| jkr| |¡ d| jkr<| d|| jd f ¡ d| jkrj| d|| j| jd | jd f ¡ |S )Nr   ÚXz%s%sr   r   z	%s%s%s.%sr   )r(   r)   Úaddr-   Úvariant_separator)r.   r%   rX   r   r   r   Úget_script_filenames,  s    



 ÿz ScriptMaker.get_script_filenamesc           
      C   sŒ   d}|r0|  dg ¡}|r0dd |¡ }| d¡}| jd||d�}|  |¡ d¡}|  |j¡}|rr|  dd¡rrd	}	nd
}	|  |||||	¡ d S )NrZ   Úinterpreter_argsz %sr   r_   ©r9   r2   FÚpywr}   )r5   r8   re   rl   rv   r©   r%   r¤   )
r.   ru   r—   r9   rT   Úargsrk   ÚscriptÚscriptnamesr˜   r   r   r   Ú_make_script7  s    
zScriptMaker._make_scriptc                 C   s�  d}t j | jt|ƒ¡}t j | jt j |¡¡}| jsT| j 	||¡sTt
 d|¡ d S zt|dƒ}W n  tk
r‚   | jsz‚ d }Y nFX | ¡ }|s t
 d|¡ d S t | dd¡¡}|rÈd}| d¡pÆd	}|�s|rÚ| ¡  | j ||¡ | jrü| j |g¡ | |¡ n„t
 d
|| j¡ | jj�s~t|jƒ\}	}
| d¡ |  |	|¡}d|k�rVd}nd}t j |¡}|  |g|| ¡ ||¡ |�rŒ| ¡  d S )NFznot copying %s (up-to-date)Úrbz%s is an empty file (skipping)s   
rQ   Tr   rZ   zcopying and adjusting %s -> %sr   s   pythonwr¬   r}   ) r$   r6   r8   r   r   r    rx   r"   r*   ÚnewerrB   r’   r>   rA   r/   ÚreadlinerC   ÚFIRST_LINE_REÚmatchr7   ÚgroupÚcloseÚ	copy_filer'   r“   r”   Úinfor   Úseekrl   r¤   r?   )r.   r®   r—   Úadjustr    ÚfÚ
first_linerµ   rT   ri   Úlinesrk   r˜   r¡   r   r   r   Ú_copy_scriptG  sR    
ÿ


zScriptMaker._copy_scriptc                 C   s   | j jS rw   ©r*   r/   )r.   r   r   r   r/   z  s    zScriptMaker.dry_runc                 C   s   || j _d S rw   rÀ   )r.   Úvaluer   r   r   r/   ~  s    r   c                 C   st   t  d¡dkrd}nd}tƒ dkr&dnd}d|||f }t d	d
¡d }t|ƒ |¡}|snd||f }t|ƒ‚|jS )NÚPé   Z64Z32z	win-arm64z-armr   z
%s%s%s.exerm   r   r   z(Unable to find resource %s in package %s)	ÚstructÚcalcsizer   Ú__name__Úrsplitr   Úfindrh   Úbytes)r.   ÚkindÚbitsZplatform_suffixr%   Zdistlib_packageÚresourceÚmsgr   r   r   r…   †  s    ÿzScriptMaker._get_launcherc                 C   s6   g }t |ƒ}|dkr"|  ||¡ n| j|||d� |S )aÓ  
        Make a script.

        :param specification: The specification, which is either a valid export
                              entry specification (to make a script from a
                              callable) or a filename (to make a script by
                              copying from a source location).
        :param options: A dictionary of options controlling script generation.
        :return: A list of all absolute pathnames written to.
        Nr«   )r
   r¿   r°   )r.   Úspecificationr9   r—   ru   r   r   r   Úmake™  s    zScriptMaker.makec                 C   s$   g }|D ]}|  |  ||¡¡ q|S )zÂ
        Take a list of specifications and make scripts from them,
        :param specifications: A list of specifications.
        :return: A list of all absolute pathnames written to,
        )ÚextendrÏ   )r.   Úspecificationsr9   r—   rÎ   r   r   r   Úmake_multiple¬  s    zScriptMaker.make_multiple)TFN)rZ   N)N)N)N)"rÆ   Ú
__module__Ú__qualname__Ú__doc__ÚSCRIPT_TEMPLATErq   r   r1   r<   r,   rS   r   rE   rL   rY   rl   rv   Ú_DEFAULT_MANIFESTry   r|   r¤   r¨   r©   r°   r¿   Úpropertyr/   Úsetterr$   r%   r&   r…   rÏ   rÒ   r   r   r   r   r   G   s:       ÿ

E:
3


r   )#Úior   Úloggingr$   ÚrerÄ   r,   r‡   Úzipfiler   Úcompatr   r   r   Ú	resourcesr   Úutilr	   r
   r   r   r   r   Ú	getLoggerrÆ   rB   Ústripr×   Úcompiler´   rÖ   r   Ú_enquote_executableÚobjectr   r   r   r   r   Ú<module>   s"    


