3
\�me" � @ s� d dl Z d dlZd dlmZmZmZmZ d dlmZm Z d dl
m Z d dl m
Z
d dlmZ d dlmZ e je�ZG dd � d ej�Zdd
d�Zdd� Zdd� ZdS )� N)�distros�helpers�subp�util)� PackageList�net_util)� HostnameConf)�Renderer)�RendererNotFoundError)� PER_INSTANCEc s� e Zd ZdZdZdgZddddd�iZd d
� Zd*d d
�Ze d�dd�Z
e d�� fdd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd+d d!�Zd"d#� Zd$d%� Zd,d&d'�Zd(d)� Z� ZS )-�Distroz/etc/locale.genz /etc/netctlZ systemctlZnetplanz /etc/netplan/50-cloud-init.yamlz# generated by cloud-init
T)Z netplan_pathZnetplan_headerZpostcmdsc C s0 t jj| |||� tj|�| _d| _d|d< d S )NZarchZsshdZ ssh_svcname)r r �__init__r ZRunners�_runnerZosfamily)�self�nameZcfg�paths� r �/usr/lib/python3.6/arch.pyr
s zDistro.__init__Nc C sj |d k r |dkr t jd|� tj� d| dg}tj| jdj|�� tjdgdd� tjd d
|gdd� d S ) Nz/etc/locale.confzFInvalid locale_configfile %s, only supported value is /etc/locale.confz%s UTF-8� �
z
locale-genF)�captureZ localectlz
set-locale)�LOG�warningr Z make_header�
write_file�
locale_gen_fn�joinr )r ZlocaleZout_fn�linesr r r � apply_locale) s zDistro.apply_locale)�pkglistc C s | j � | jd|d� d S )Nr )�pkgs)�update_package_sources�package_command)r r r r r �install_packages< s zDistro.install_packages)�returnc s8 y
t � j� S tk
r2 } z t|�W Y d d }~X nX d S )N)�super�
_get_rendererr
�NotImplementedError)r �e)� __class__r r r% @ s
zDistro._get_rendererc C s. t j|�}tjd||� t|| j| j| jd�S )Nz3Translated ubuntu style network settings %s into %s)� resolv_conf�conf_dir� enable_func)r Ztranslate_networkr �debug�_render_networkZresolve_conf_fn�network_conf_dir�_enable_interface)r �settings�entriesr r r �_write_networkG s
zDistro._write_networkc C s\ dd|g}y(t j |�\}}t|�r0tjd||� W n$ t jk
rV tjtd|� Y nX d S )N�netctlZreenablez(Running %s resulted in stderr output: %sz#Running interface command %s failed)r �lenr r �ProcessExecutionErrorr �logexc)r � device_name�cmd�_out�errr r r r/ U s
zDistro._enable_interfacec C sj dd|g}t jd||� y(tj|�\}}t|�r>t jd||� dS tjk
rd tjt d|� dS X d S )Nr3 Zrestartz8Attempting to run bring up interface %s using command %sz(Running %s resulted in stderr output: %sTz#Running interface command %s failedF)r r, r r4 r r5 r r6 )r r7 r8 r9 r: r r r �_bring_up_interface` s
zDistro._bring_up_interfacec C sr d }y| j |�}W n. tk
r@ tj| jdd�}|r8nd S Y nX |sNtd�}|j|� tj|t|�ddd� d S )N�create_hostname_fileTr �wi� )Zomode�mode) �_read_hostname_conf�IOErrorr Zget_cfg_option_boolZ_cfgr Z set_hostnamer �str)r �hostname�filename�confr< r r r �_write_hostnamer s
zDistro._write_hostnamec C s | j | j�}| j|fS )N)�_read_hostnameZhostname_conf_fn)r Z sys_hostnamer r r �_read_system_hostname� s z Distro._read_system_hostnamec C s t tj|��}|j� |S )N)r r Z load_file�parse)r rC rD r r r r? � s zDistro._read_hostname_confc C s: d }y| j |�}|j}W n tk
r, Y nX |s6|S |S )N)r? rB r@ )r rC �defaultrB rD r r r rF � s
zDistro._read_hostnamec C sL t jd|� ytjddd|g� W n$ tjk
rF tjt d|� Y nX d S )Nz2Non-persistently setting the system hostname to %sZ hostnamectlz --transientz set-hostnamez;Failed to non-persistently adjust the system hostname to %s)r r, r r5 r r6 )r rB r r r �_apply_hostname� s zDistro._apply_hostnamec C s t j|| j|�d� d S )N)�tzZtz_file)r Zset_etc_timezoneZ
_find_tz_file)r rK r r r � set_timezone� s zDistro.set_timezonec C s� |d kr g }ddddg}|r2t |t�r2|j|� n|rJt |t�rJ|j|� |dkrVd}|rd|j|� tjd|�}|j|� tj|dd � d S )
NZpacmanz-Syz--quietz --noconfirmZupgradez-uz%s-%sF)r )�
isinstancerA �append�list�extendr Zexpand_package_listr )r Zcommand�argsr r8 r r r r r! � s
zDistro.package_commandc C s | j jd| jdgtd� d S )Nzupdate-sourcesz-y)Zfreq)r Zrunr! r )r r r r r � s z Distro.update_package_sources)N)N)NN)�__name__�
__module__� __qualname__r r. Zinit_cmdZrenderer_configsr
r r r" r r% r2 r/ r; rE rG r? rF rJ rL r! r �
__classcell__r r )r( r r s,
r �/�
etc/netctl�etc/resolv.confc C s� g }g }t j||�}t j||�}x�| j� D ]�\}}|dkr<q*|j|� tjj||�} d||jd�d|jd�|jd�f |jd�|jdg �d �}
tj | t
|
�� |r�|jd
�r�||� d|kr*|j |d � q*W |r�tj |t |�� |S ) znRender the translate_network format into netctl files in target.
Paths will be rendered under target.
�loZethernetZ bootprotoz%s/%sZaddressZnetmaskZgatewayzdns-nameservers)Z
ConnectionZ InterfaceZIPZAddressZGatewayZDNS�auto)
r Z target_path�itemsrN �os�pathr �getr r �convert_netctlrP �convert_resolv_conf) r1 �targetr* r) r+ ZdevsZ nameserversZdev�infoZnet_fnZnet_cfgr r r r- � s.
r- c C sr g }xbt | �D ]V}| | }|dkr(d}n*t|ttf�rRddjdd� |D �� d }|jd||f � qW dj|�S ) z�Given a dictionary, returns a string in netctl profile format.
netctl profile is described at:
https://git.archlinux.org/netctl.git/tree/docs/netctl.profile.5.txt
Note that the 'Special Quoting Rules' are not handled here.Nr �(� c s s | ]}d | V qdS )z'%s'Nr )�.0�vr r r � <genexpr>� s z!convert_netctl.<locals>.<genexpr>�)z%s=%s
)�sortedrM �tuplerO r rN )r0 �result�key�valr r r r_ � s r_ c C s, d}t | t�r(x| D ]}|d| }qW |S )z4Returns a settings string formatted for resolv.conf.r znameserver %s
)rM rO )r0 rk �nsr r r r` s
r` )rV rW rX N)Zloggingr\ Z cloudinitr r r r Zcloudinit.distrosr r Z"cloudinit.distros.parsers.hostnamer Zcloudinit.net.rendererr Zcloudinit.net.renderersr
Zcloudinit.settingsr Z getLoggerrR r r r- r_ r` r r r r �<module> s
7
$