U
|
X ±d¿# ã @ s d Z ddlZddlZddlmZ ddlmZ ddl m
|
Z
|
m Z ddl m Z e e¡ZdZdZdZG d d
|
d
|
ZG d d d eZG d d deZG dd deZG dd deZG dd dZdddZdd Zedkr ddl Z e j d Z e j dd
Z ee e dS ) ap
|
The code in this module supports the --icon parameter on Windows.
|
(For --icon support under Mac OS, see building/osx.py.)
|
|
The only entry point, called from api.py, is CopyIcons(), below. All the elaborate structure of classes that follows
|
is used to support the operation of CopyIcons_FromIco(). None of these classes and globals are referenced outside
|
this module.
|
é N)Úconfig)Ú
|
pywintypesÚwin32api)Únormalize_icon_typeé é é c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )Ú Structurec C sT t | j¡ }| _tt | jd| ¡| _i }| _t| j D ]\}}|||< q>d S )Nó )
|
ÚstructÚcalcsizeÚ_format_Ú _sizeInBytesÚlistÚunpackÚ_fields_Ú _indexes_Ú enumerateÚ_names_)ÚselfÚsizeZindexesÚiÚnm© r úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\PyInstaller/utils/win32/icon.pyÚ__init__$ s
|
|
zStructure.__init__c C sD t d| ¡ | jD ]"}| d¡st d|t| |¡ qt d¡ d S )Nz
|
DUMP of %sÚ_z %20s = %sÚ )ÚloggerÚinfor Ú
|
startswithÚgetattr)r Únamer r r Údump+ s
|
|
|
zStructure.dumpc
|
C s\ || j kr | j| }| j| S z | j| W S tk
|
rV } zt||W 5 d }~X Y nX d S ©N)r r r Ú__dict__ÚKeyErrorÚAttributeError)r r" ÚindexÚer r r Ú __getattr__2 s
|
|
|
zStructure.__getattr__c C s. || j kr | j| }|| j|< n
|
|| j|< d S r$ )r r r r% )r r" Úvaluer( r r r Ú __setattr__; s
|
|
zStructure.__setattr__c C s t j| jf| j S r$ )r Úpackr r ©r r r r ÚtostringB s zStructure.tostringc C s$ | | j¡}tt | j|¡| _d S r$ )Úreadr r r r r r )r ÚfileÚdatar r r ÚfromfileE s zStructure.fromfileN) Ú__name__Ú
|
__module__Ú __qualname__r r# r* r, r/ r3 r r r r r # s r c @ s e Zd ZdZdZdS )Ú ICONDIRHEADER©Z
|
idReservedZidTypeÚidCountÚhhhN©r4 r5 r6 r r r r r r r7 J s r7 c @ s e Zd ZdZdZdS )Ú ICONDIRENTRY)ÚbWidthÚbHeightÚ bColorCountÚ bReservedÚwPlanesÚ wBitCountÚ dwBytesInResÚ dwImageOffsetZbbbbhhiiNr; r r r r r< O s r< c @ s e Zd ZdZdZdS )Ú
|
GRPICONDIRr8 r: Nr; r r r r rE T s rE c @ s e Zd ZdZdZdS )ÚGRPICONDIRENTRY)r= r> r? r@ rA rB rC ÚnIDZbbbbhhihNr; r r r r rF Y s rF c @ s& e Zd Zdd Zdd Zd ddZdS )
|
ÚIconFilec C sÎ || _ zt| j d}W n& tk
|
r< td| j dY nX | g | _g | _t }| _| |¡ t |j
|
D ] }t }| |¡ | j |¡ qp| jD ]&}| |jd¡ | j | |j¡¡ qW 5 Q R X d S )NÚrbzUnable to open icon file ú!r )ÚpathÚopenÚOSErrorÚ
|
SystemExitÚentriesÚimagesr7 Úheaderr3 Úranger9 r< ÚappendÚseekrD r0 rC )r rK r1 rQ r Úentryr) r r r r ` s"
|
|
|
zIconFile.__init__c C s
|
| j ¡ S r$ )rQ r/ r. r r r Ú grp_icon_dirv s zIconFile.grp_icon_diré c C sZ d}| j D ]J}t }|jd d
D ]}t||t|| q"||_|d }|| ¡ }q
|
|S )Nó éÿÿÿÿrW )rO rF r Úsetattrr! rG r/ )r Úidr2 rU r) Únr r r Úgrp_icondir_entriesy s
|
z IconFile.grp_icondir_entriesN)rW )r4 r5 r6 r rV r] r r r r rH _ s rH rW c C s¶ t t|}t d|¡ t | d¡}d}t|D ]v\}}| ¡ }|| |¡ }t |t
|
||¡ t d|t |¡ |j D ].}t |t ||¡ t d|t |¡ |d }qtq.t |d¡ dS )zà
|
Use the Win API UpdateResource facility to apply the icon resource(s) to the .exe file.
|
|
:param str dstpath: absolute path of the .exe file being built.
|
:param str srcpath: list of 1 or more .ico file paths
|
úCopying icons from %sr rW z/Writing RT_GROUP_ICON %d resource with %d bytesz)Writing RT_ICON %d resource with %d bytesN)ÚmaprH r r r ÚBeginUpdateResourcer rV r] ÚUpdateResourceÚ RT_GROUP_ICONÚlenrP ÚRT_ICONÚEndUpdateResource) ÚdstpathÚsrcpathr[ ZiconsÚhdstZiconidr Úfr2 r r r ÚCopyIcons_FromIco
s
|
|
rj c
|
C sØ t |tr|g}dd }tt||}t|dkrhg }|D ]"}| t|d ddtjd ¡ q:t | |S |d \}}t|ddtjd }t
|
j |¡d }| ¡ d kr°t | |gS |d
|
k rÈt d ||¡ n t d |¡ zt |dt¡}W n< tjk
|
r" } ztd ||j|j¡W 5 d
|
}~X Y nX t | d¡} |d
|
krLt |t¡d }
|
n"|dkrht |t¡| }
|
n| }
|
t |t|
|
¡} t | t|
|
| ¡ t |t ¡D ]$} t |t | ¡} t | t | | ¡ qt |¡ t | d¡ d
|
S )aÇ
|
Called from building/api.py to handle icons. If the input was by --icon on the command line, srcpath is a single
|
string. However, it is possible to modify the spec file adding icon=['foo.ico','bar.ico'] to the EXE() statement.
|
In that case, srcpath is a list of strings.
|
|
The string format is either path-to-.ico or path-to-.exe,n for n an integer resource index in the .exe. In either
|
case, the path can be relative or absolute.
|
c S sB z | d¡\}}| ¡ t|fW S tk
|
r< | df Y S X dS )zv
|
Convert "pathname" to tuple ("pathname", None)
|
Convert "pathname,n" to tuple ("pathname", n)
|
ú,N)ÚsplitÚstripÚintÚ
|
ValueError)Úsrg r( r r r Úsplitter¯ s
|
zCopyIcons.<locals>.splitterrW r )Úicorr Zworkpath)Zexerr z.icoNzCopying icon from %s, %dr^ z2Unable to load icon file {}
|
{} (Error code {}))!Ú
|
isinstanceÚstrr r_ rc rS r r ZCONFrj ÚosrK ÚsplitextÚlowerr r r Z LoadLibraryExÚLOAD_LIBRARY_AS_DATAFILEr ÚerrorrN ÚformatÚstrerrorÚwinerrorr` ZEnumResourceNamesrb Z LoadResourcera rd Ú FreeLibraryre ) rf rg rq Zsrcsrp r( ZsrcextZhsrcZW32Erh Zgrpnamer2 Ziconnamer r r Ú CopyIcons¡ sH
|
|
|
ÿ
|
|
|
r~ Ú__main__)rW ) Ú__doc__Úos.pathru r ZPyInstaller.logÚlogÚloggingZ PyInstallerr ZPyInstaller.compatr r ZPyInstaller.building.iconr Ú getLoggerr4 r rd rb rx r r7 r< rE rF rH rj r~ ÚsysÚargvrf rg r r r r Ú<module> s.
|
'&
|
R
|
|
|
