Mini Shell
3
C��e� � @ s d dl Z d dlZ d dlZd dlZd dlZd dlZd dlmZ d dl m
Z
d dl m Z m
Z
mZmZ d dlmZ d dlmZ d dlmZ d dlmZmZ d d lmZ d d
l m Z d d l m Z d d l m!Z! e!e"�Z#ej$j%� Z&d dd�Z'dd� Z(dd� Z)dd� Z*dd� Z+dd� Z,dd� Z-d d � Z.dS ) � N)�Size)�isys)� errorHandler� ERROR_RAISE�InvalidImageSizeError�MissingImageError)� DEVICE_TREE)�STORAGE)�MountFilesystemError)�
DeviceData�DeviceFormatData)�utils)�InstallTreeMetadata)�DiscInfo)�get_module_logger�/mnt/install/cdimagec # C s yt j| � W n tk
r" dS X t}t jj|d�}t jj| �rh| jd�rht jj| �g}t jj | �} n
t j
| �}�x�|D �]�}t jj| |�}t j d|� t
j|�s�qzt j d||� ytjj||ddd� W n tk
r� wzY nX t j|t j��stjj|� qzt j d � t� }y|j|� |j}W n2 tk
�rX } zt jd
| � wzW Y dd} ~ X nX t j d |� ||k�r�t jd ||� tjj|� qzt|��s�t jd
|� tjj|� qzt j|�tj d �r�t jd|� t d|�}
t j |
�t k�r�|
�t j d|� tjj|� |S W dS )a� Find the first iso image in path.
:param str path: path to the directory with iso image(s); this also supports pointing to
a specific .iso image
:param str mount_path: path for mounting the ISO when checking it is valid
FIXME once payloads are modularized:
- this should move somewhere else
- mount_path should lose the legacy default
:return: basename of the image - file name without path
:rtype: str or None
Nz .discinfoz.isoz Checking %szMounting %s on %s�iso9660�ro)�fstype�optionszReading .discinfoz".discinfo file can't be loaded: %sz
discArch = %sz8Architectures mismatch in find_first_iso_image: %s != %sz*%s doesn't have a valid repodata, skippingi z%s appears to be corruptedz$size is not a multiple of 2048 byteszFound disc at %s)!�os�stat�OSError�_arch�path�join�isfile�endswith�basename�dirname�listdir�log�debugr Z
isIsoImage�blivet�util�mount�access�R_OKZumountr �load�arch� Exception�warning�_check_repodata�ST_SIZEr r �cbr �info) r �
mount_pathr) Z
discinfo_path�files�fnZwhatZ disc_infoZ disc_archZex�exn� r4 � /usr/lib64/python3.6/image.py�find_first_iso_image/ sd
r6 c C sb t � }|j| �stjd� |j� }|s6|j� }t|�}|sHtjd� dS |j� rTdS tjd� dS )Nz!Can't read install tree metadata!z'There is no usable repository availableFTz'There is no valid repository available.) r Z load_filer! r+ Zget_base_repo_metadataZget_metadata_repos�#_search_for_install_root_repositoryr" Zis_valid)r0 Zinstall_tree_metaZrepo_mdZrepo_mdsr4 r4 r5 r, � s
r, c C s x| D ]}|j dkr|S qW d S )N�.)Z
relative_path)ZreposZrepor4 r4 r5 r7 � s
r7 c C s� x�t jj| �r| }n@t| �}|d kr@t� }tj|�tkr|�nqt jjd| |f �}yt j
j ||ddd� W n0 t k
r� t� }tj|�tkr�|�nwY qX P qW d S )Nz%s/%sr r )r r )
r r r r6 r r r. r �normpathr# r$ r% r )ZisodirZtreeZimager3 r4 r4 r5 �
mountImage� s$ r: c C s� t jt�} x~| j� D ]r}tj� }zVytj||� W n tk
rH wY nX zddl m
} ||�sbwW dtj ||� X W dt j
|� X |S W dS )z�Find a device with a valid optical install media.
Return the first device containing a valid optical install
media for this product.
FIXME: This is duplicated in SetUpCdromSourceTask.run
:return: a device name or None
r )�is_valid_install_diskN)r � get_proxyr ZFindOpticalMedia�tempfileZmkdtemp�
payload_utilsZ mount_devicer
Z(pyanaconda.modules.payloads.source.utilsr; Zunmount_devicer �rmdir)� device_treeZdevZ
mountpointr; r4 r4 r5 �find_optical_install_media� s
rA c C s t jt�} | j� S )z�Find potential HDISO sources.
Return a generator yielding Device instances that may have HDISO install
media somewhere. Candidate devices are simply any that we can mount.
:return: a list of device names
)r r< r ZFindMountablePartitions)r@ r4 r4 r5 � find_potential_hdiso_sources� s
rB c C sz t j| j|��}tj| j|��}t j| j|jd ��}|jjdd�jdd�|j t
|j �|j |jjd�pt|jjd�ptdd�S ) z�Get info about a potential HDISO source.
:param device_tree: a proxy of a device tree
:param device_name: a device name
:return: a dictionary with a device info
r �model� �_� �labelZuuid)rC r �size�formatrG )
r Zfrom_structureZ
GetDeviceDatar Z
GetFormatData�parentsZattrs�get�replacer r rH � description)r@ Z device_nameZ device_dataZ format_dataZ disk_datar4 r4 r5 �get_hdiso_source_info� s rN c C s dj f | �S )z�Get a description of a potential HDISO source.
:param device_info: a dictionary with a device info
:return: a string with a device description
z({model} {path} ({size}) {format} {label})rI )Z device_infor4 r4 r5 � get_hdiso_source_description s rO )r )/r Zos.pathr r= Z blivet.utilr# Z blivet.archZ blivet.sizer Z
pyanacondar Zpyanaconda.errorsr r r r Z+pyanaconda.modules.common.constants.objectsr Z,pyanaconda.modules.common.constants.servicesr Z(pyanaconda.modules.common.errors.storager
Z,pyanaconda.modules.common.structures.storager r Zpyanaconda.payloadr
r> Z(pyanaconda.payload.install_tree_metadatar Zproductmd.discinfor Zpyanaconda.anaconda_loggersr �__name__r! r) Zget_archr r6 r, r7 r: rA rB rN rO r4 r4 r4 r5 �<module> s4
T "