CasperSecurity
o
>I�c� � @ s� d dl mZmZ d dl mZ d dlZd dlZd dlZd dl Z dZ
G dd� de�Z e dkree d�Z
ee
�d d
�� ee
�d d �� ee
�d d
�� ee
�d dd�� ee
�d dd�� ee
�d dd�� dS dS )� )�
NoOptionError�NoSectionError)� ConfigParserNz&/etc/update-manager/release-upgrades.dc @ s2 e Zd Z d dd�Zdd� Zdd� Zd d
� ZdS ) �DistUpgradeConfig�DistUpgrade.cfgNc C s� t �| � tjg d�tjdd��� d �� }|| _tj �
||�}tj � |d | �r0|d| 7 }g | _ |rEt
�
|d �D ]}| j �|� q<| j |g7 _ |d u rSt}|d urgt
�
|d �D ]}| j �|� q^| �| j � d S )N)� lsb_releasez-cz-sT)�stdout�universal_newlinesr �.z/*.cfg)�SafeConfigParser�__init__�
subprocess�Popen�PIPE� communicate�strip�datadir�os�path�join�exists� config_files�glob�append�CONFIG_OVERRIDE_DIR�read)�selfr �name� override_dir� defaults_dir� from_release�maincfg�cfg� r# �E/usr/lib/python3/dist-packages/DistUpgrade/DistUpgradeConfigParser.pyr " s0
��� zDistUpgradeConfig.__init__c C sx z.t |�tkr| �||�W S t |�tkr| �||�W S t |�tkr(| �||�W S | �||�W S tt fy; | Y S w )N)
�type�bool�
getboolean�float�getfloat�int�getint�getr r )r �section�option�defaultr# r# r$ �getWithDefault= s �z DistUpgradeConfig.getWithDefaultc C sD z| � ||�}W n
ttfy g Y S w dd� |�d�D �}|S )Nc S � g | ]}|� � �qS r# �r ��.0�xr# r# r$ �
<listcomp>N � z-DistUpgradeConfig.getlist.<locals>.<listcomp>�,)r, r r �split)r r- r. �tmp�itemsr# r# r$ �getlistI s �zDistUpgradeConfig.getlistc C s� z| � ||�}W n ty g Y S w tj�| j|�}tj�|�s)t�d| � t |��}dd� |D �}W d � n1 s?w Y dd� |D �S )Nz getListFromFile: no '%s' foundc S r1 r# r2 r3 r# r# r$ r6 Z r7 z5DistUpgradeConfig.getListFromFile.<locals>.<listcomp>c S s" g | ]
}|� d �s|dks|�qS )�#� )�
startswith)r4 �sr# r# r$ r6 [ s " )
r, r r r r r r �logging�error�open)r r- r. �filename�p�fr; r# r# r$ �getListFromFileQ s �
�z!DistUpgradeConfig.getListFromFile)r NN)�__name__�
__module__� __qualname__r r0 r< rG r# r# r# r$ r ! s
� r �__main__r
�Distro�MetaPkgs� ForcedPurges�Sources� ValidMirrors� EnableApportT�Foo�False)� configparserr r r r r
�os.pathr rA r r r rH �c�printr< rG r0 �setr# r# r# r$ �<module> s" =�