OwlCyberSecurity - MANAGER
Edit File: IcnsImagePlugin.cpython-38.pyc
U tCf)- � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZm Z m Z d dl m Z e ejd�Zertd dlmZ dZdd� Zd d � Zd d � Zd d� Zdd� ZG dd� d�ZG dd� de j �Zdd� Ze�ejedd� � e� ejd� ej dk�re� eje� e� ejd� e d k�r�e!ej"�d k �r<e#d � e�$� ee%ej"d d ��Z&e&j'd! D ](Z(e(e&_(e&�)� e&j*Z*e*�+d"e( � �qZe�%ej"d �Z*e*�+d#� ej d$k�r�e�,d#� dS )%� N)�Image� ImageFile�PngImagePlugin)�i8Zjp2klib_version)�Jpeg2KImagePlugin� c C s t �d| �t��S )Nz>4sI)�structZunpack�read� HEADERSIZE)�fobj� r �5/usr/lib/python3/dist-packages/PIL/IcnsImagePlugin.py� nextheader$ s r c C sD |\}}| � |� | �d�}|dkr,td��t| |d |d f|�S )N� s z'Unknown signature, expecting 0x00000000)�seekr � SyntaxError�read_32)r � start_length�size�start�length�sigr r r �read_32t( s r c C sd |\}}| � |� |d |d |d |d f}|d |d }||d krp| �|�}t�d||dddd�}n�t�d|d�}td�D ]�} g } |} | dk�r| �d�} | s��qt| �} | d@ r�| d } | �d�} t| �D ]}| �| � q�n| d } | �| �| �� | | 8 } | dkr��qq�| dk�r.td | ��t�d |d � | �dd dd�}|j � |j | � q�d|iS ) zn Read a 32bit RGB icon resource. Seems to be either uncompressed or an RLE packbits-like scheme. r � � � �RGB�rawN� �} z Error reading channel [%r left]�L� ) r r r � frombuffer�new�ranger �appendr �join�imZputband)r r r r r � pixel_size�sizesqZindatar' Zband_ix�dataZ bytesleftZbyte� blocksize�i�bandr r r r 2 s> r c C sf |d }| � |� |d |d |d |d f}|d |d }t�d|| �|�dddd�}d|iS )Nr r r r r �A)r r r" r )r r r r r( r) r- r r r �read_mk\ s r/ c C s� |\}}| � |� | �d�}|d d� dkrT| � |� t�| �}t�|j� d|iS |d d� dks||d d� dks||dkr�ts�td ��| � |� | �|�}t � |�}t � |�}t�|j� |j dkr�|�d�}d|iS td ��d S ) N� r s �PNG �RGBAr s �O�Qs � s jP � zQUnsupported icon subimage format (rebuild PIL with JPEG 2000 support to fix this)z Unsupported icon subimage format)r r r Z PngImageFiler Z_decompression_bomb_checkr � enable_jpeg2k� ValueError�io�BytesIOr ZJpeg2KImageFile�modeZconvert) r r r r r r r' Z jp2kstream�fr r r �read_png_or_jpeg2000f s6 ���� r8 c @ s� e Zd Zdefgdefgdefgdefgdefgdefdefdefgd efgd efgd efd efgd efdefdefgdefgdefdefdefgd� Zdd� Zdd� Z dd� Z dd � Z d d d �Z d S )!�IcnsFiles ic10s ic09s ic14s ic08s ic13s ic07s it32s t8mks icp6s ic12s ih32s h8mks icp5s il32s l8mks ic11s icp4s is32s s8mk) )� r: r )r: r: r )� r; r )r; r; r )r r r )r r r )�@ r< r )� r= r )�0 r> r )r= r= r )� r? r )r? r? r c C s� i | _ }|| _t|�\}}|dkr,td��t}||k r�t|�\}}|dkrTtd��|t7 }|t8 }||f||< |�|tj� ||7 }q0dS )z@ fobj is a file-like object as an icns resource � icnsznot an icns filer zinvalid block headerN)�dctr r r r r r4 �SEEK_CUR)�selfr rA r Zfilesizer, r+ r r r �__init__� s zIcnsFile.__init__c C sB g }| j �� D ].\}}|D ] \}}|| jkr|�|� qqq|S �N)�SIZES�itemsrA r% )rC �sizesr ZfmtsZfmt�readerr r r � itersizes� s zIcnsFile.itersizesc C s | � � }|std��t|�S )Nz No 32bit icon resources found)rJ r �max)rC rH r r r �bestsize� s zIcnsFile.bestsizec C sD i }| j | D ]0\}}| j�|�}|dk r|�|| j||�� q|S )z� Get an icon resource as {channel: array}. Note that the arrays are bottom-up like windows bitmaps and will likely need to be flipped or transposed in some way. N)rF rA �get�updater )rC r rA �coderI Zdescr r r � dataforsize� s zIcnsFile.dataforsizeNc C s� |d kr| � � }t|�dkr.|d |d df}| �|�}|�dd �}|rL|S |�d��� }z|�|d � W n tk r� Y nX |S )Nr r r r1 r r. )rL �lenrP rM �copyZputalpha�KeyError)rC r Zchannelsr' r r r �getimage� s zIcnsFile.getimage)N) �__name__� __module__� __qualname__r8 r r/ r rF rD rJ rL rP rT r r r r r9 � s6 ���� r9 c @ sB e Zd ZdZdZdZdd� Zedd� �Zej dd� �Zd d � Z d S ) � IcnsImageFilea! PIL image support for Mac OS .icns files. Chooses the best resolution, but will possibly load a different size image if you mutate the size attribute before calling 'load'. The info dictionary has a key 'sizes' that is a list of sizes that the icns file has. ZICNSzMac OS icns resourcec C s\ t | j�| _d| _| j�� | jd< | j�� | _| jd | jd | jd | jd f| _d S )Nr1 rH r r r ) r9 �fp�icnsr6 rJ �inforL � best_sizer �rC r r r �_open� s �zIcnsImageFile._openc C s | j S rE )�_sizer] r r r r s zIcnsImageFile.sizec C s� |}|| j d kr0t|�dkr0|d |d df}|| j d kr�t|�dkr�|d dkr�dd� | j d D �}||kr�| j d |�|� }|| j d kr�td��|| _d S ) NrH r r r r c S s, g | ]$}|d |d |d |d f�qS )r r r r )�.0r r r r � <listcomp> s z&IcnsImageFile.size.<locals>.<listcomp>z2This is not one of the allowed sizes of this image)r[ rQ �indexr3 r_ )rC �valueZ info_sizeZ simple_sizesr r r r s" � � ��c C s� t | j�dkr@| j| _| jd | jd | jd | jd f| _tj�| � | jrd| jj| jkrdd S | �� | j�| j�}|�� |j| _|j | _ |j| _| � � d S )Nr r r r ) rQ r r\ r �loadr' Z load_preparerZ rT r6 Zload_end)rC r' r r r rd s � zIcnsImageFile.loadN) rU rV rW �__doc__�formatZformat_descriptionr^ �propertyr �setterrd r r r r rX � s rX c C sB t |d�r|�� t�d���}dd� | j�dg �D �}d}d}dD ]�}d�||�}tj� ||d �} ||kr|t � || � n$|�|| � ||ft j��} | �| � tj� ||d �}|�|d | � |d |d ft j��} | �|� |d }qDd d dd||g} tj| tjtjd�} | j�� | �� }|�r4t�|| ��W 5 Q R X dS )z� Saves the image as a series of PNG files, that are then converted to a .icns file using the macOS command line utility 'iconutil'. macOS only. �flushz.iconsetc S s i | ] }|j |�qS r )�width)r` r' r r r � <dictcomp>@ s z_save.<locals>.<dictcomp>Z append_imagesN)r? r= r r; r: z icon_{}x{}z.pngz@2x.pngr Ziconutilz-crZ z-o)�stdout�stderr)�hasattrri �tempfileZTemporaryDirectoryZ encoderinforM rf �os�pathr& �shutilZcopyfileZresizer ZLANCZOS�save� subprocess�Popen�PIPEZDEVNULLrl �close�waitZCalledProcessError)r' rY �filenameZiconsetZprovided_imagesZlast_wZ second_path�w�prefixZ first_pathZim_wZim_w2Z convert_cmdZ convert_procZretcoder r r �_save3 s: � & � r| c C s | d d� dkS )Nr r@ r )�xr r r �<lambda>c r! r~ z.icns�darwinz image/icns�__main__r z(Syntax: python IcnsImagePlugin.py [file]r �rbrH zout-%s-%s-%s.pngzout.pngZwindows)-r4 rp rr r rt �sysro ZPILr r r Z PIL._binaryr rn Zcorer2 r r r r r r/ r8 r9 rX r| Z register_openrf Zregister_extension�platformZ register_saveZ register_mimerU rQ �argv�print�exit�openZimfr[ r rd r' rs Z startfiler r r r �<module> sN * eH0