Mini Shell
3
�
Ld�� � @ sr d Z ddlmZ ddlmZ ddlZddlZddlZddlZddl Z yddl
m Z W n e k
rt ddl
m Z Y nX ddlmZmZ ddlmZmZmZ ddlmZ dd lmZ dd
lmZ dd lm Z m Z m Z m Z m Z m!Z!m"Z" dd l#m$Z$ d
Z%dZ&dd� Z'dd� Z(dd� Z)dd� Z*dd� Z+G dd� de�Z,G dd � d e�Z-G d d � d e�Z.G d d � d e�Z/G d!d"� d"e0�Z1dS )#a+
Main kickstart file processing module.
This module exports several important classes:
Script - Representation of a single %pre, %pre-install, %post, or %traceback script.
Packages - Representation of the %packages section.
KickstartParser - The kickstart file parser state machine.
� )�print_function)�IteratorN)�
OrderedSet)� constants�version)�KickstartError�KickstartParseError�KickstartParseWarning)�KickstartObject)� load_to_str)�KSOptionParser)�PackageSection�PreScriptSection�PreInstallScriptSection�PostScriptSection�TracebackScriptSection�OnErrorScriptSection� NullSection)�_�endZcommandsc C s. d }d}d}t jr |jtj� �}�xy t| �}W n tk
rF P Y nX |dkrRP |d7 }d }|j� }|jd�s�t jr�|jtj� �}||7 }q$y|j d�d }W n t
t d�| |d��Y nX y t |�}W n< t
k
�r
} z t
t d�t|� |d��W Y d d }~X nX |d k r$||jtj� �7 }q$W |S ) Nr � � z %ksappend� z Illegal url for %%ksappend: %s)�linenoz"Unable to open %%ksappend file: %s)�sixZPY3�encode�sys�getdefaultencoding�next�
StopIteration�strip�
startswith�splitr r r r �str)�lineIter�lr �retvalZksurlZll�contents�e� r) � /usr/lib/python3.6/parser.py�_preprocessStateMachine< s>
*r+ c C s t | jd�dg �}t|�S )a
Preprocess the kickstart file, provided as the string s. This
method is currently only useful for handling %ksappend lines, which
need to be fetched before the real kickstart parser can be run.
Returns the complete kickstart file as a string.
Tr )�iter�
splitlinesr+ )�s�ir) r) r* � preprocessFromStringToStringk s r0 c C sZ y t | �}W n: tk
rF } z ttd�t|� dd��W Y dd}~X nX tt|jd���S )a
Preprocess the kickstart file, given by the filename f. This
method is currently only useful for handling %ksappend lines,
which need to be fetched before the real kickstart parser can be
run. Returns the complete kickstart file as a string.
z'Unable to open input kickstart file: %sr )r NT)r r r r# r+ r, r- )�fr' r( r) r) r* �preprocessKickstartToStringt s
*r2 c C sD t | �} | r@ddl}|jddd�\}}tj|| � tj|� |S dS )a Preprocess the kickstart file, provided as the string s. This
method is currently only useful for handling %ksappend lines,
which need to be fetched before the real kickstart parser can be
run. Returns the location of the complete kickstart file.
r Nz-ks.cfgr z/tmp)r0 �tempfile�mkstemp�os�write�close)r. r3 �outF�outNamer) r) r* �preprocessFromString� s
r: c C sD t | �}|r@ddl}|jddd�\}}tj||� tj|� |S dS )a Preprocess the kickstart file, given by the filename f. This
method is currently only useful for handling %ksappend lines,
which need to be fetched before the real kickstart parser can be
run. Returns the location of the complete kickstart file.
r Nz-ks.cfgr z/tmp)r2 r3 r4 r5 r6 r7 )r1 r. r3 r8 r9 r) r) r* �preprocessKickstart� s
r; c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �PutBackIteratorc C s t |�| _d | _d S )N)r, � _iterable�_buf)�self�iterabler) r) r* �__init__� s
zPutBackIterator.__init__c C s | S )Nr) )r? r) r) r* �__iter__� s zPutBackIterator.__iter__c C s
|| _ d S )N)r> )r? r. r) r) r* �put� s zPutBackIterator.putc C s$ | j r| j }d | _ |S t| j�S d S )N)r> r r= )r? r& r) r) r* r � s
zPutBackIterator.nextc C s | j � S )N)r )r? r) r) r* �__next__� s zPutBackIterator.__next__N)�__name__�
__module__� __qualname__rA rB rC r rD r) r) r) r* r< � s
r< c @ s" e Zd ZejZdd� Zdd� ZdS )�Scriptc O sx t j| f|�|� dj|�| _|jdd�| _|jdd�| _|jdd�| _|jdd�| _|jd d�| _ |jd
t
j �| _ dS ) a| Create a new Script instance. Instance attributes:
:keyword errorOnFail: If execution of the script fails, should anaconda
stop, display an error, and then reboot without
running any other scripts?
:keyword inChroot: Does the script execute in anaconda's chroot
environment or not?
:keyword interp: The program that should be used to interpret this
script.
:keyword lineno: The line number this script starts on.
:keyword logfile: Where all messages from the script should be logged.
:keyword script: A string containing all the lines of the script.
:keyword type: The type of the script, which can be KS_SCRIPT_* from
:mod:`pykickstart.constants`.
r �interpz/bin/sh�inChrootFr N�logfile� errorOnFail�type)
r
rA �join�script�getrI rJ r rK rL r �
KS_SCRIPT_PRErM )r? rO �args�kwargsr) r) r* rA � s zScript.__init__c C s6 d}| j tjkr|d7 }nV| j tjkr0|d7 }n@| j tjkrF|d7 }n*| j tjkr\|d7 }n| j tjkrp|d7 }| jdkr�| jr�|d| j 7 }| j tjkr�| j r�|d 7 }| j d
k r�|d | j 7 }| j
r�|d 7 }| j j d
��r| j
tjkr�|d| j S |d| j S n*| j
tjk�r$|d| j S |d| j S d
S )z9Return a string formatted for output to a kickstart file.r z
%prez
%postz
%tracebackz
%pre-installz
%onerrorz/bin/shz --interpreter=%sz --nochrootNz
--logfile=%sz --erroronfail�
z
%s%%end
z
%sz
%s
%%end
z
%s
)rM r rQ ZKS_SCRIPT_POSTZKS_SCRIPT_TRACEBACKZKS_SCRIPT_PREINSTALLZKS_SCRIPT_ONERRORrI rJ rK rL rO �endswith�_verr �F8)r? r& r) r) r* �__str__� s4
zScript.__str__N)rE rF rG r �DEVELrV rA rX r) r) r) r* rH � s rH c @ s^ e Zd ZdZdejfdd�Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
ejZdS )�Groupz=A class representing a single group in the %packages section.r c C s t j| � || _|| _dS )z�Create a new Group instance. Instance attributes:
name -- The group's identifier
include -- The level of how much of the group should be included.
Values can be GROUP_* from pykickstart.constants.
N)r
rA �name�include)r? r[ r\ r) r) r* rA
s
zGroup.__init__c C s: | j tjkrd| j S | j tjkr,d| j S d| j S dS )z9Return a string formatted for output to a kickstart file.z@%s --nodefaultsz@%s --optionalz@%sN)r\ r �GROUP_REQUIREDr[ � GROUP_ALL)r? r) r) r* rX s
z
Group.__str__c C s | j |j k S )N)r[ )r? �otherr) r) r* �__lt__! s z Group.__lt__c C s | j |j kS )N)r[ )r? r_ r) r) r* �__le__$ s z Group.__le__c C s | j |j kS )N)r[ )r? r_ r) r) r* �__eq__' s z Group.__eq__c C s | j |j kS )N)r[ )r? r_ r) r) r* �__ne__* s z Group.__ne__c C s | j |j kS )N)r[ )r? r_ r) r) r* �__gt__- s z Group.__gt__c C s | j |j kS )N)r[ )r? r_ r) r) r* �__ge__0 s z Group.__ge__N)rE rF rG �__doc__r �
GROUP_DEFAULTrA rX r` ra rb rc rd re r
�__hash__r) r) r) r* rZ s rZ c @ s2 e Zd ZejZdd� Zdd� Zdd� Zdd� Z d S )
�Packagesc O sx t j| f|�|� d| _d| _d| _d| _g | _g | _d| _g | _ t
j | _ g | _
d| _d| _d| _d| _d| _d| _dS )aO Create a new Packages instance. Instance attributes:
addBase -- Should the Base group be installed even if it is
not specified?
nocore -- Should the Core group be skipped? This results in
a %packages section that basically only installs the
packages you list, and may not be a usable system.
default -- Should the default package set be selected?
environment -- What base environment should be selected? Only one
may be chosen at a time.
excludedList -- A list of all the packages marked for exclusion in
the %packages section, without the leading minus
symbol.
excludeDocs -- Should documentation in each package be excluded?
groupList -- A list of Group objects representing all the groups
specified in the %packages section. Names will be
stripped of the leading @ symbol.
excludedGroupList -- A list of Group objects representing all the
groups specified for removal in the %packages
section. Names will be stripped of the leading
-@ symbols.
handleMissing -- If unknown packages are specified in the %packages
section, should it be ignored or not? Values can
be KS_MISSING_* from pykickstart.constants.
packageList -- A list of all the packages specified in the
%packages section.
instLangs -- A list of languages to install.
multiLib -- Whether to use yum's "all" multilib policy.
excludeWeakdeps -- Whether to exclude weak dependencies.
timeout -- Number of seconds to wait for a connection before
yum's or dnf's timing out or None.
retries -- Number of times yum's or dnf's attempt to retrieve
a file should retry before returning an error.
seen -- If %packages was ever used in the kickstart file,
this attribute will be set to True.
TFN)r
rA �addBase�nocore�default� environment� excludedList�excludedGroupList� excludeDocs� groupListr ZKS_MISSING_PROMPT�
handleMissing� packageList� instLangs�multiLib�excludeWeakdeps�timeout�retries�seen)r? rR rS r) r) r* rA 9 s" &zPackages.__init__c C s� d}| j s�| jr |d| j 7 }| j}|j� x |D ]}|d|j� 7 }q2W | j}|j� x|D ]}|d| 7 }q^W | j}|j� x |D ]}|d|j� 7 }q�W | j}|j� x|D ]}|d| 7 }q�W |dkr�| j r�dS d}| j r�|d7 }| j r�|d7 }| j
�s
|d7 }| j �r|d 7 }| j t
jk�r0|d
7 }| jd k �rJ|d | j 7 }| j�rZ|d
7 }| j�rj|d7 }| jd k �r�|d| j 7 }| jd k �r�|d| j 7 }| jtjk�r�|d | d S |d | d S d S )z9Return a string formatted for output to a kickstart file.r z@^%s
z%s
z-%s
z
%packagesz
--defaultz --excludedocsz --nobasez --nocorez --ignoremissingNz --instLangs=%sz --multilibz --excludeWeakdepsz
--timeout=%dz
--retries=%drT z
%end
)rl rm rq �sortrX rs ro rn ry rp rj rk rr r ZKS_MISSING_IGNORErt ru rv rw rx rV r rW )r? ZpkgsZgrps�grp�p�pkgr& r) r) r* rX r s\
zPackages.__str__c C s� t ddtjd�}|jddddtjd� |jddddtjd� |j|j� d�\}}|jrh|jrhtt d ���d
j
|�}|d d � | j D �kr�d S |jr�| j j t
|tjd
�� n4|jr�| j j t
|tjd
�� n| j j t
|tjd
�� d S )Nr )�prog� descriptionr z --nodefaults�
store_trueF)�actionrl �helpr z
--optional)rR z5Group cannot specify both --nodefaults and --optionalr c S s g | ]
}|j �qS r) )r[ )�.0�gr) r) r* �
<listcomp>� s z*Packages._processGroup.<locals>.<listcomp>)r[ r\ )r r rY � add_argumentZparse_known_argsr" Z
nodefaultsZoptionalr r rN rq �appendrZ r r] r^ rg )r? �line�op�nsZextrar{ r) r) r* �
_processGroup� s
zPackages._processGroupc sT t | j�}t | j�}t � }t � }g }x�|D ]�}|j� }|dd� dkrV|dd� | _q*|d dkrv| j|dd� � q*|d dkr�|dd� dkr�| j|dd� kr�d| _q�|d dkr�|jt|dd� d �� q�|j|dd� � q*|j|� q*W d
d � |D �� � fd d �| j D �| _ | j
j |� || |B }|| |B }t |�| _t |�| _dS )
z�Given a list of lines from the input file, strip off any leading
symbols and add the result to the appropriate list.
r � z@^N�@r �-� )r[ c S s g | ]
}|j �qS r) )r[ )r� r� r) r) r* r� � s z Packages.add.<locals>.<listcomp>c s g | ]}|j � kr|�qS r) )r[ )r� r� )�excludedGroupNamesr) r* r� � s )
r rn rs r rm r� r� rZ �addrq ro �extend�list) r? ZpkgListZexistingExcludedSetZexistingPackageSetZnewExcludedSetZ
newPackageSetro r} �strippedr) )r� r* r� � s2
"
z Packages.addN)
rE rF rG r rY rV rA rX r� r� r) r) r) r* ri 5 s
9;ri c @ s� e Zd ZdZd$dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd%d d �Zd&d d �Zd!d"� Zd#S )'�KickstartParsera5 The kickstart file parser class as represented by a basic state
machine. To create a specialized parser, make a subclass and override
any of the methods you care about. Methods that don't need to do
anything may just pass. However, _stateMachine should never be
overridden.
Tc C sh || _ d| _|| _|| _i | _|| _|| _t| _d| _ d| _
| jj | _ | j t _
| j t_
i | _| j� dS )a. Create a new KickstartParser instance. Instance attributes:
errorsAreFatal -- Should errors cause processing to halt, or
just print a message to the screen? This
is most useful for writing syntax checkers
that may want to continue after an error is
encountered.
followIncludes -- If %include is seen, should the included
file be checked as well or skipped?
handler -- An instance of a BaseHandler subclass. If
None, the input file will still be parsed
but no data will be saved and no commands
will be executed.
missingIncludeIsFatal -- Should missing include files be fatal, even
if errorsAreFatal is False?
unknownSectionIsFatal -- Should an unknown %section be fatal? Not all
sections are handled by pykickstart. Some are
user-defined, so there should be a way to have
pykickstart ignore them.
r r N)�errorsAreFatal� errorsCount�followIncludes�handler�
currentdir�missingIncludeIsFatal�unknownSectionIsFatal�STATE_COMMANDS�_state�
_includeDepth�_liner rH rV ri � _sections�
setupSections)r? r� r� r� r� r� r) r) r* rA s
zKickstartParser.__init__c C s t | _d| _dS )zKReset the internal variables of the state machine for a new kickstart file.r N)r� r� r� )r? r) r) r* �_reset, s zKickstartParser._resetc C s
| j | S )zyReturn a reference to the requested section (s must start with '%'s),
or raise KeyError if not found.
)r� )r? r. r) r) r* �
getSection1 s zKickstartParser.getSectionc C s& | j r"| j| j _| j j||�}|S dS )a Given the list of command and arguments, call the Version's
dispatcher method to handle the command. Returns the command or
data object returned by the dispatcher. This method may be
overridden in a subclass if necessary.
N)r� r� Z currentLineZ
dispatcher)r? r rR r& r) r) r* �
handleCommand7 s
z KickstartParser.handleCommandc C s< |j std| ��|j jd�s,td|j ��|| j|j < dS )z�Given an instance of a Section subclass, register the new section
with the parser. Calling this method means the parser will
recognize your new section and dispatch into the given object to
handle it.
z#no sectionOpen given for section %s�%z'section %s tag does not start with a %%N)� sectionOpen� TypeErrorr! r� )r? �objr) r) r* �registerSectionB s
z KickstartParser.registerSectionc C s |j � t| _dS )z�Called at the close of a kickstart section to take any required
actions. Internally, this is used to add scripts once we have the
whole body read.
N)�finalizer� r� )r? r� r) r) r* � _finalizeP s zKickstartParser._finalizec C s8 | j tkrdS |dd� dkr4| jdd� j� | j_dS )z/Kickstart recognizes a couple special comments.N�
z
#platform=)r� r� r� r r� �platform)r? r� r) r) r* �_handleSpecialCommentsX s
z&KickstartParser._handleSpecialCommentsc C s� | j | j }�x�yJt|�}|dkrX| jdkrX| jtjkrNttd�|j |d��| j |� W n t
k
rp P Y nX |d7 }| j |�r�|j r�q|j
� jd��r�|j� d }| j|� r�|d
kr�|j|� qtj|�}|r�|d dkr�| j |� P n�|�r<|d dk�r<t|�dk�s |d �r*t|d��| j|d � qnp|�rT|d d k�rTqnX|�r�| j|d ��r�| jtjk�r�ttd�|j |d��|j|� |d8 }| j |� P q|j|� qW |S ) Nr r z#Section %s does not end with %%end.)r r r� �%end�%includez %ksappend)r� r� )r� r� r r� r rW r r r� r� r �_isBlankOrCommentZallLines�lstripr! r" � _validStateZ
handleLine�shlex�len�_handleIncluderC )r? r$ r r� r� ZpossibleSectionStartrR r) r) r* � _readSectiona sN
z KickstartParser._readSectionc C s |t | jj� �kS )zFIs the given section tag one that has been registered with the parser?)r� r� �keys)r? �str) r) r* r� � s zKickstartParser._validStatec C sX y
|� W nH t k
rR } z,| jd7 _| jr4� nt|tjd� W Y dd}~X nX dS )z�Call the provided function (which doesn't take any arguments) and
do the appropriate error handling. If errorsAreFatal is False, this
function will just print the exception and keep going.
r )�fileN)� Exceptionr� r� �printr �stderr)r? �fn�msgr) r) r* �_tryFunc� s
zKickstartParser._tryFuncc C s |j � p |dkp |j� d dkS )Nr r �#)�isspacer� )r? r� r) r) r* r� � s z!KickstartParser._isBlankOrCommentc C sZ | j s
d S | jd7 _y| j|dd� W n tk
rF | jrB� Y nX | jd8 _d S )Nr F)�reset)r� r� �
readKickstartr r� )r? r1 r) r) r* r� � s z KickstartParser._handleIncludec
s� d��x�yt |��_�jdkr P W n tk
r8 P Y nX �d7 ��j�j�r\�j�j� qtj�jdd�� � d dkr�t� �dks�� d r�t�d���j � d � q�j
t k�r�� d dkr�qn�� d d d k�rl� d }�j |��s6�j
�rttd
�| �d��n.tjtd ��|d � t� �jt�j|d
�� |�_
�j�j
��j� ��fdd�� �j|���n�j� ��fdd�� q�j
tk�r�P q�jdkr|j�j� �d8 ��j|���qW d S )Nr r r T)Zcommentsz%include)r z %ksappendr� z Unknown kickstart section: %szDPotentially unknown section seen at line %(lineno)s: %(sectionName)s)r Z sectionName)r� c s �j �� �S )N)Z handleHeaderr) )rR r r� r) r* �<lambda>� s z/KickstartParser._stateMachine.<locals>.<lambda>c s �j �� �S )N)r� r) )rR r r? r) r* r� s )r r� r r� r� r� r" r� r r� r� r� r� r� r �warnings�warnr r� r r� r� r� r� � STATE_ENDr� rC )r? r$ Z
newSectionr) )rR r r� r? r* �
_stateMachine� sN
z KickstartParser._stateMachinec C s. |r | j � t|jd�dg �}| j|� dS )z5Process a kickstart file, provided as the string str.Tr N)r� r< r- r� )r? r. r� r/ r) r) r* �readKickstartFromString
s z'KickstartParser.readKickstartFromStringc C s� |r | j � tjj|�sd| jd | jkrdtjjtjj| j| jd |��rdtjj| j| jd |�}tjj|�}|jd�s�tjj |�}|| j| j< y t
|�}W n: t k
r� } z t t d�t
|� dd��W Y dd}~X nX | j|dd� dS ) z2Process a kickstart file, given by the filename f.r �/z'Unable to open input kickstart file: %sr )r NF)r� )r� r5 �path�existsr� r� rN �dirnamer! �abspathr r r r# r� )r? r1 r� Zcdr. r( r) r) r* r� s "
*z KickstartParser.readKickstartc C s� i | _ | jt| jtd�� | jt| jtd�� | jt| jtd�� | jt| jtd�� | jt| jtd�� | jt | j�� | jt
| jdd�� | jt
| jdd�� dS )z�Install the sections all kickstart files support. You may override
this method in a subclass, but should avoid doing so unless you know
what you're doing.
)ZdataObjz%addon)r� z %anacondaN) r� r� r r� rH r r r r r
r )r? r) r) r* r� 3 s z KickstartParser.setupSectionsN)TTTT)T)T)rE rF rG rf rA r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r) r) r) r* r� � s$
( AD
r� )2rf Z
__future__r �collections.abcr r5 r r� r r� Z ordered_setr � ImportErrorZ
orderedsetZ pykickstartr r Zpykickstart.errorsr r r Zpykickstart.kor
Zpykickstart.loadr Zpykickstart.optionsr Zpykickstart.sectionsr
r r r r r r Zpykickstart.i18nr r� r� r+ r0 r2 r: r; r< rH rZ ri �objectr� r) r) r) r* �<module> s> $ /
O* H