Mini Shell
3
Z��e�S � 0 @ s2 d dl Z d dlZd dlZd dlmZ d dlZd dlZd dlZd dlZd dl Z d dl
m Z d dl m
Z
d dlmZ d dlmZmZ d dlmZmZmZ d dlmZ d d lmZm Z d d
l m Z d d l m Z d d l!m"Z" d d
l#m$Z$ d dl%m&Z&m'Z' d dl(m)Z)m*Z* d dl+m,Z,m-Z-m.Z.m/Z/ d dl0m1Z1m2Z2 d dl3m4Z4 d dl3m5Z6 d dl7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z? d dl@mAZA d dlBmCZC d dlDmEZEmFZFmGZG eEeH�ZIeF� ZJeIjKd�ZLeIjKd�ZMe dd� �ZNG d d � d e6�ZOG d d � d eO�ZPG d d!� d!e*ed"�ZQG d#d$� d$eQ�ZRejSZSG d%d&� d&ejT�ZTG d'd(� d(ejU�ZUG d)d*� d*e)�ZVG d+d,� d,e?�ZWeReReReReReReReReSeReReReReReReReReReReReTeReReReReReReReReReReReReReReReReReReReReReReReReReRd-�/ZXd(eUiZYeAe�ZZG d.d/� d/eZ�Z[G d0d1� d1e4�Z\G d2d3� d3e4�Z]d4d5� Z^dCd7d8�Z_d9d:� Z`d;d<� Zad=d>� Zbd?d@� ZcdAdB� ZddS )D� N)�ABCMeta)�contextmanager)�util)�conf)�VERSION�commands)� AddonSection� AddonData�
AddonRegistry)� IPMI_ABORTED)� ScriptError� errorHandler)�flags)�_)�BOSS)�KickstartReport)� F22_PwPolicy�F22_PwPolicyData)� BaseHandler�KickstartCommand)�KS_SCRIPT_POST�
KS_SCRIPT_PRE�KS_SCRIPT_TRACEBACK�KS_SCRIPT_PREINSTALL)�KickstartError�KickstartParseWarning)�KickstartParser)�Script)� NullSection�PackageSection�PostScriptSection�PreScriptSection�PreInstallScriptSection�OnErrorScriptSection�TracebackScriptSection�Section)�returnClassForVersion)�anaconda_logging)�get_module_logger�get_stdout_logger�get_anaconda_root_logger�scriptZparsingc c sN y
d V W n> t k
rH } z"t| � tjt� tjd� W Y d d } ~ X nX d S )N� )r �printr � ipmi_reportr �sys�exit)�e� r2 �!/usr/lib64/python3.6/kickstart.py�check_kickstart_errorD s
r4 c @ s e Zd ZdZdd� ZdS )�AnacondaKSScriptz� Execute a kickstart script
This will write the script to a file named /tmp/ks-script- before
execution.
Output is logged by the program logger, the path specified by --log
or to /tmp/ks-script-\*.log
c
C s^ | j r |}nd}tjdd|d �\}}tj|| jjd�� tj|� tj|d� | j r�| j rld|| j f }n| j }tj
j |�}tj
j |�s�tj
|� ndtj
j|� }tj|d d
d ��(}tj| jd tj
j|� g||d
�}W dQ R X |dk�rZtjd|| j� | j�rZd} t|d��}dj|j� �} W dQ R X dt_t j t | j| �� tj t � t!j"d� dS )zk Run the kickstart script
@param chroot directory path to chroot into before execution
�/� z
ks-script-z/tmpzutf-8i� z%s/%sz /tmp/%s.log�wi� )Zpermz/tmp/%s)�stdout�rootNr z5Error code %s running the kickstart script at line %s�rT)#ZinChroot�tempfileZmkstemp�os�writer+ �encode�close�chmodZlogfile�path�dirname�exists�makedirs�basenamer Zopen_with_permZexecWithRedirectZinterp�
script_log�error�linenoZ errorOnFail�open�join� readlinesr Zkspromptr
�cbr r. r r/ r0 )
�self�chrootZ
scriptRoot�fdrB Zmessages�d�fpZrc�errr2 r2 r3 �runX s:
zAnacondaKSScript.runN)�__name__�
__module__� __qualname__�__doc__rT r2 r2 r2 r3 r5 P s r5 c s$ e Zd Z� fdd�Zdd� Z� ZS )�AnacondaInternalScriptc s t � j||� d| _d S )NT)�super�__init__Z_hidden)rN �args�kwargs)� __class__r2 r3 r[ � s z AnacondaInternalScript.__init__c C s dS )Nr7 r2 )rN r2 r2 r3 �__str__� s z AnacondaInternalScript.__str__)rU rV rW r[ r_ �
__classcell__r2 r2 )r^ r3 rY � s rY c @ s e Zd ZdZdd� Zdd� ZdS )�RemovedCommanda Kickstart command that was moved on DBus.
This class should simplify the transition to DBus.
Kickstart command that was moved on DBus should inherit this
class. Methods parse, setup and execute should be modified to
access the DBus modules or moved on DBus.
c C s dS )z�Generate this part of a kickstart file from the module.
This method shouldn't be overridden, because we use DBus modules
to generate these parts of a kickstart file.
Make sure that each DBus module is used only once.
r7 r2 )rN r2 r2 r3 r_ � s zRemovedCommand.__str__c C s t jd| j� dS )z�Do not parse anything.
We can keep this method for the checks if it is possible, but
it shouldn't parse anything.
z)Command %s will be parsed in DBus module.N)�logZwarning�
currentCmd)rN r\ r2 r2 r3 �parse� s zRemovedCommand.parseN)rU rV rW rX r_ rd r2 r2 r2 r3 ra � s
ra )� metaclassc @ s e Zd ZdZdd� ZdS )�UselessCommandz�Kickstart command that was moved on DBus and doesn't do anything.
Use this class to override the pykickstart command in our command map,
when we don't want the command to do anything. It is not allowed to
subclass this class.
c K s t d��d S )Nz7It is not allowed to subclass the UselessCommand class.)� TypeError)�clsr] r2 r2 r3 �__init_subclass__� s z UselessCommand.__init_subclass__N)rU rV rW rX ri r2 r2 r2 r3 rf � s rf c @ s e Zd Zdd� ZdS )�Loggingc C sv t jjt jkr0t j| j }|t j_t jt� |� t jjd krrt | j
�dkrr| j
}| j rfd| j
| j f }t jj |� d S )Nr z%s:%s)
r' ZloggerZloglevelZ
DEFAULT_LEVELZ logLevelMap�levelZsetHandlersLevelr* Z
remote_syslog�len�hostZportZ updateRemote)rN rk Z
remote_serverr2 r2 r3 �execute� s zLogging.executeN)rU rV rW rn r2 r2 r2 r3 rj � s rj c sT e Zd ZdZ� fdd�Zedd� �Zdd� Zedd � �Z� fd
d �Z d d
� Z
� Z S )�RepoDatar c sf |j dd�| _|j dd�| _|j dd�| _|j dd�| _|j dd�| _|j d d�| _t� j||� dS )
zr Add enabled kwarg
:param enabled: The repo has been enabled
:type enabled: bool
�enabledT�repo_idN�treeinfo_originF� partition�iso_path�mount_dir_suffix) �poprp rq rr rs rt ru rZ r[ )rN r\ r] )r^ r2 r3 r[ � s zRepoData.__init__c C s0 | |j |j|j|j|j|j|j|j|j|j d�
S )N)
�name�baseurl�
mirrorlist�metalink�proxyrp rr rs rt ru )
rw rx ry rz r{ rp rr rs rt ru )rh �otherr2 r2 r3 � create_copy� s zRepoData.create_copyc C s$ | j � r | jdkr d| j� | _dS )zcGenerate persistent mount directory suffix
This is valid only for HD repositories
NZ addition_)�is_harddrive_basedru �_generate_mount_dir_suffix)rN r2 r2 r3 �generate_mount_dir� s zRepoData.generate_mount_dirc C s t | j�}| jd7 _|S )Nr, )�str�_RepoData__mount_counter)rh �suffixr2 r2 r3 r s
z#RepoData._generate_mount_dir_suffixc s | j rt� j� S dS dS )zDon't output disabled reposr7 N)rp rZ r_ )rN )r^ r2 r3 r_
s
zRepoData.__str__c C s
| j d k S )N)rs )rN r2 r2 r3 r~ s zRepoData.is_harddrive_based) rU rV rW r� r[ � classmethodr} r� r r_ r~ r` r2 r2 )r^ r3 ro � s
ro c s8 e Zd ZdZdeiZdeiZ� fdd�Zdd� Z � Z
S )�AnacondaSectionHandlerz2A handler for only the anaconda ection's commands.ZpwpolicyZ PwPolicyDatac s t � j| j| jd� d S )N)�mappingZ dataMapping)rZ r[ �
commandMap�dataMap)rN )r^ r2 r3 r[ " s z AnacondaSectionHandler.__init__c C sT d}t | jj� �}x,|D ]$}x | j| D ]}|t|�7 }q(W qW |rPd| d }|S )z Return the %anaconda sectionr7 z
%anaconda
z%end
)�sortedZ _writeOrder�keysr� )rN ZretvalZlstZprio�objr2 r2 r3 r_ % s
z AnacondaSectionHandler.__str__) rU rV rW rX r r� r r� r[ r_ r` r2 r2 )r^ r3 r� s
r� c s< e Zd ZdZdZ� fdd�Zdd� Zdd� Zd d
� Z� Z S ) �AnacondaSectionz)A section for anaconda specific commands.z %anacondac s t � j||� d| _d S )Nr )rZ r[ �cmdno)rN r\ r] )r^ r2 r3 r[ 8 s zAnacondaSection.__init__c C sL | j s
d S | jd7 _tj|dd�}|d | j _| j| j _| j j|| j�S )Nr, T)Zcommentsr )�handlerr� �shlex�splitrc Z currentLineZ
dispatcher)rN �liner\ r2 r2 r3 �
handleLine< s
zAnacondaSection.handleLinec C s t j| ||� dS )z.Process the arguments to the %anaconda header.N)r% � handleHeader)rN rI r\ r2 r2 r3 r� F s z AnacondaSection.handleHeaderc C s t j| � dS )z0Let %anaconda know no additional data will come.N)r% �finalize)rN r2 r2 r3 r� J s zAnacondaSection.finalize)
rU rV rW rX � sectionOpenr[ r� r� r� r` r2 r2 )r^ r3 r� 4 s
r� )/ZauthZ
authconfigZ
authselectZautopartZbtrfsZ
bootloaderZcdromZ clearpartZeulaZfcoeZfirewallZ firstboot�groupZ harddriveZhmcZ
ignorediskZiscsiZ iscsinameZkeyboardZlangZloggingZlogvol�methodZmountZnetworkZnfsZnvdimm�partrs ZraidZrealmZreqpartZrhsmZrootpwZselinuxZservicesZsshkeyZskipxZsnapshotZ
syspurposeZtimezoneZurl�userZvolgroupZxconfigZzerombrZzfcpc s. e Zd ZeZd� fdd� Z� fdd�Z� ZS )�AnacondaKSHandlerNc s� |d kr g }|d krt }|d kr$t}t� j||d� i � _i }xf|D ]^\}}tjjtjjtjj |���}tjj
|�svqDt j ||� fdd��}|rD|d |d�||< qDW t
|�� _t� � _d S )N)�commandUpdates� dataUpdatesc s t | � j�S )N)�
issubclass�AddonClassType)rh )rN r2 r3 �<lambda>� s z,AnacondaKSHandler.__init__.<locals>.<lambda>r )rw )r� r� rZ r[ ZonPartr= rB rF rC �abspath�isdirr Zcollectr
�addonsr� �anaconda) rN Z addon_pathsr� r� r� Z module_namerB Zaddon_id�classes)r^ )rN r3 r[ � s&
zAnacondaKSHandler.__init__c s> t j� }|j� j� }t� j� d | d t| j� t| j� S )N�
z
) r � get_proxyZGenerateKickstart�striprZ r_ r� r� r� )rN r{ �modules)r^ r2 r3 r_ � s zAnacondaKSHandler.__str__)NNN)rU rV rW r r� r[ r_ r` r2 r2 )r^ r3 r� � s %r� c s. e Zd Zd� fdd� Zdd� Zdd� Z� ZS ) �AnacondaPreParserTc s t � j|dd� d S )NF)�missingIncludeIsFatal)rZ r[ )rN r� �followIncludes�errorsAreFatalr� )r^ r2 r3 r[ � s zAnacondaPreParser.__init__c C s d S )Nr2 )rN rI r\ r2 r2 r3 �
handleCommand� s z AnacondaPreParser.handleCommandc C s� | j t| jtd�� | j t| jdd�� | j t| jdd�� | j t| jdd�� | j t| jdd�� | j t| jdd�� | j t| jdd�� | j t| jjd d�� d S )
N)�dataObjz %pre-install)r� z%postz%onerrorz
%tracebackz %packagesz%addonz %anaconda)�registerSectionr! r� r5 r r� )rN r2 r2 r3 �
setupSections� s z AnacondaPreParser.setupSections)TTT)rU rV rW r[ r� r� r` r2 r2 )r^ r3 r� � s r� c s6 e Zd Zdddef� fdd� Zdd� Zdd� Z� ZS )�AnacondaKSParserTc s || _ t� j|� d S )N)� scriptClassrZ r[ )rN r� r� r� r� r� )r^ r2 r3 r[ � s zAnacondaKSParser.__init__c C s | j s
d S tj| ||�S )N)r� r r� )rN rI r\ r2 r2 r3 r� � s z AnacondaKSParser.handleCommandc C s� | j t| j| jd�� | j t| j| jd�� | j t| j| jd�� | j t| j| jd�� | j t| j| jd�� | j t| j�� | j t | j�� | j t
| jj �� d S )N)r� ) r� r! r� r� r" r r$ r# r r r� r� )rN r2 r2 r3 r� � s z AnacondaKSParser.setupSections)rU rV rW r5 r[ r� r� r` r2 r2 )r^ r3 r� � s r� c C s6 t t� �}t� � |j| � W d Q R X t|jj� d S )N)r� r� r4 �
readKickstart�
runPreScriptsr� �scripts)�f�ksparserr2 r2 r3 �
preScriptPass� s
r� Fc sX t | �}g � tj�d � �fdd� }y�tj� �� |t_tjdtd� |r�tj� }tj |j
|��}x |j D ]}tj |j
t� qfW |j� s�djtt|j��} t| ��|j|� � r�ttd�� x � D ]}
tt|
�j� � q�W |r�td��W d Q R X W nd tk
�rR } zFtj| � ttd�t| �j� � tjt � t j d � t j d
� W Y d d } ~ X nX d S ) Nc s* �| |||||� t |t�r&� j| � d S )N)r� r �append)�message�category�filenamerI �filer� )�
kswarnings� showwarningr2 r3 �
ksshowwarning� s
z%parseKickstart.<locals>.ksshowwarning�always)r� z
z:
Some warnings occurred during reading the kickstart file:zVPlease modify your kickstart file to fix the warnings or remove the `ksstrict` option.z[
An error occurred during reading the kickstart file:
%s
The installer will now terminate.�
r, )NN)!r� �warningsr� �catch_warnings� simplefilterr r r� r Zfrom_structureZReadKickstartFileZwarning_messages�warnr� Zis_validrK �mapr� Zerror_messagesr r� r- r r� � parsing_logrH r r. r �timeZsleepr/ r0 ) r� r� Z strict_modeZ pass_to_bossr� r� ZbossZreportr� r� r8 r1 r2 )r� r� r3 �parseKickstart� s>
r� c C sX d}x4t tjd��D ]"}t|d�}||j� 7 }|j� qW t| td�}|j|dd� d S )Nr7 z$/usr/share/anaconda/post-scripts/*ksr; )r� F)�reset)r� �globrJ �readr@ r� rY ZreadKickstartFromString)Zksdatar� �fnr� r� r2 r2 r3 �appendPostScripts1 s
r� c C sR dd� | D �}t |�dkr d S tjd� x|D ]}|jtjj� q.W tjd� d S )Nc S s g | ]}|j tkr|�qS r2 )�typer )�.0�sr2 r2 r3 �
<listcomp>C s z"runPostScripts.<locals>.<listcomp>r z"Running kickstart %%post script(s)z,All kickstart %%post script(s) have been run)rl rG �inforT r �targetZ system_root)r� Z postScriptsr+ r2 r2 r3 �runPostScriptsB s
r� c C s\ dd� | D �}t |�dkr d S tjd� tjtd�� x|D ]}|jd� q<W tjd� d S )Nc S s g | ]}|j tkr|�qS r2 )r� r )r� r� r2 r2 r3 r� O s z!runPreScripts.<locals>.<listcomp>r z!Running kickstart %%pre script(s)z Running pre-installation scriptsr6 z+All kickstart %%pre script(s) have been run)rl rG r� � stdoutLogr rT )r� Z
preScriptsr+ r2 r2 r3 r� N s
r� c C sN dd� | D �}t |�dkr d S tjd� x|D ]}|jd� q.W tjd� d S )Nc S s g | ]}|j tkr|�qS r2 )r� r )r� r� r2 r2 r3 r� ^ s z(runPreInstallScripts.<locals>.<listcomp>r z)Running kickstart %%pre-install script(s)r6 z3All kickstart %%pre-install script(s) have been run)rl rG r� rT )r� ZpreInstallScriptsr+ r2 r2 r3 �runPreInstallScripts] s
r� c C s: t jd� x tdd� | �D ]}|jd� qW t jd� d S )Nz'Running kickstart %%traceback script(s)c S s
| j tkS )N)r� r )r� r2 r2 r3 r� m s z%runTracebackScripts.<locals>.<lambda>r6 z1All kickstart %%traceback script(s) have been run)rG r� �filterrT )r� r+ r2 r2 r3 �runTracebackScriptsk s
r� )FF)er� r= Zos.path�abcr r� r/ r< r� r� �
contextlibr Zpyanaconda.corer Z&pyanaconda.core.configuration.anacondar Zpyanaconda.core.kickstartr r ZCOMMANDSZpyanaconda.addonsr r r
Zpyanaconda.core.constantsr Zpyanaconda.errorsr r
Zpyanaconda.flagsr Zpyanaconda.core.i18nr Z,pyanaconda.modules.common.constants.servicesr Z.pyanaconda.modules.common.structures.kickstartr Zpyanaconda.pwpolicyr r Zpykickstart.baser r Zpykickstart.constantsr r r r Zpykickstart.errorsr r Zpykickstart.parserr r ZKSScriptZpykickstart.sectionsr r r r! r" r# r$ r% Zpykickstart.versionr& Z
pyanacondar' Zpyanaconda.anaconda_loggersr( r) r* rU rb r� ZgetChildrG r� r4 r5 rY ra rf ZEularj ro r� r� r� r� Z
superclassr� r� r� r� r� r� r� r� r� r� r2 r2 r2 r3 �<module> s� (
<
= ".
<