관리-도구
편집 파일: config_handlers.cpython-39.pyc
a d4h � ) @ s* d dl Z d dlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z d Z g d �Ze� Zd d � Zd d� deddd� eedeeeeedd� eeeeedd� ejdd� ejdejdd� eeeeejddejedd� edded�(Zdd� Zdd� ZG dd� de�Zd d � Zd d � Z d d!� Z dS )"� N� )� constants)� log_utils)�utils)�config)� http_utils)� ConfigParserz /etc/sysconfig/kcare/kcare.conf)�USE_CONTENT_FILE_V3�FORCE_JSON_SIG_V3�ENABLE_CRASHREPORTER� KCORE_OUTPUT� KMSG_OUTPUTc C s | � � dv S )N)�1�TRUE�YES�Y)�upper)�value� r �5/usr/libexec/kcare/python/kcarectl/config_handlers.py�bool_converter s r c C s | � � S �N��strip��vr r r �<lambda>& � r c C s | � � S r r r r r r r * r c C s | � d�S �N�/��rstripr r r r r 3 r c C s | pd S r r r r r r r 9 r c C s | � d�S r r r r r r r ; r c C s | � d�S r r r r r r r ? r c C s dd� | � d�D �S )Nc S s g | ]}|� � �� �qS r )r �lower)�.0�ptchr r r � <listcomp>I r z <lambda>.<locals>.<listcomp>�,)�splitr r r r r I r )(�AFTER_UPDATE_COMMAND�AUTO_STICKY_PATCHSET� AUTO_UPDATE�AUTO_UPDATE_DELAY�BEFORE_UPDATE_COMMAND�CHECK_SSL_CERTSr � FORCE_GIDr � HTTP_TIMEOUT�IGNORE_UNKNOWN_KERNELr �KCORE_OUTPUT_SIZE� KDUMPS_DIRr �LIBCARE_DISABLED�!LIBCARE_PIDLOGS_MAX_TOTAL_SIZE_MB�LIBCARE_SOCKET_TIMEOUT�LIB_AUTO_UPDATE� PATCH_LEVEL� PATCH_METHOD� PATCH_SERVER� PATCH_TYPE�PREFIX�PREV_PATCH_TYPE�REGISTRATION_URL� PRINT_LEVEL� REPORT_FQDN�SILENCE_ERRORS�STATUS_CHANGE_GAP� STICKY_PATCH�STICKY_PATCHSET� UPDATE_DELAY� UPDATE_POLICY�UPDATE_SYSCTL_CONFIG�USERSPACE_PATCHESr �KERNEL_VERSION_FILE�KCARE_UNAME_FILE�SUCCESS_TIMEOUTc K s� t t�}|�� }|�� | �� D ]�\}}d}|d }|d }tt|��D ]P}|| �|�sh|| �|�rH|d u rx||= n|d t|� d ||< d} q�qH|s |� |d t|� d � q t � td� |�� d S )NF�=� z = � T� ) �open�CONFIG� readlines�close�items�range�len� startswith�str�appendr � atomic_write�join) �kwargsZcf�linesZpropr �updatedZprop_eqZprop_sp�ir r r � update_configQ s" r_ c C s� i }t �d�}| D ]<}|�|�}|r:|�� \}}|sFd }n td| ��|||< qt|�tt� }|rztdd�t|�� ��|� � D ]R\}}t| } |d u s�| d u r�q�z | |� W q� t y� td||f ��Y q�0 q�t f i |�� d S )Nz^([^=]+)=([^=]*)$z8Invalid parameter format: %s. Format should be KEY=VALUEzUnknown parameter: %sz, zBad value for %s: %s) �re�compile�match�groups� SystemExit�set�POSSIBLE_CONFIG_OPTIONSrZ �sortedrS � Exceptionr_ ) �paramsZparams_for_update�pattern�paramrb �keyr Zunknown_params�var_name�convertr r r �update_config_from_argsf s, ro c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� FakeSecHeadc C s || _ d| _d S )Nz [asection] )�fp�sechead)�selfrq r r r �__init__� s zFakeSecHead.__init__c C s. | j r z| j W d | _ S d | _ 0 n | j�� S d S r )rr rq �readline)rs r r r ru � s � zFakeSecHead.readlinec c s* | j r| j V d | _ | jD ] }|V qd S r )rr rq )rs �liner r r �__iter__� s zFakeSecHead.__iter__N)�__name__� __module__� __qualname__rt ru rw r r r r rp � s rp c s� t �� i �tddd�d�� z,ttt��} tjr<� �| � n � � | � W n t y^ i Y S 0 d � �fdd� }dD ],\}}t � |�st� � d|�}|rt|tj|< qtt�� D ]\}}|||d� q��S ) NrN )� HTTP_PROXY� HTTPS_PROXY)�defaultsc sR z� � d| �}W n ty& |}Y n0 |d urN|r<||�}|�| < t�| � d S )N�asection)�getrh �CONFIG_OPTIONS�add)�name�defaultrn r �Zcp�resultr r �read_var� s z%get_config_settings.<locals>.read_var))�httpr{ )�httpsr| r~ )rn )NN)r� �clearr rp rO rP r �PY2�readfp� read_filerh r �get_proxy_from_envr �os�environrf rS )r r� �scheme�variable�proxyrm rn r r� r �get_config_settings� s& r� c C sL i }| � � D ]:\}}|�� }|�d�r |�dd��dd��� }|||< q |S )ao Checking headers for feature flags which start with 'KC-Flag-' and reformat it to dictionary with keys in upper case and without 'KC-Flag-' prefix and dashes replaced with underscores. For unification all header keys are checking in upper case. For example: 'KC-Flag-Some-Value' -> 'SOME_VALUE' :return: dict {'SOME_VALUE': 'value', ...} zKC-FLAG-rN �-�_)rS r rV �replace)�headers�flagsrl r Z key_upperZ formatted_keyr r r �%get_config_options_from_feature_flags� s r� c C s� t | �}|�� D ]l\}}|tvr"q|dur|tvrz&tt|��tj|< tj � d||� W q t yz tj � d||� Y q0 qdS )z� Set global variables using feature flag from patchserver headers. Checks that option is allowed by whitelist and update global variable using globals() :param headers: Response headers from patchserver :return: None Nz'patchserver config override: %s with %szGInvalid value during attempt to override config from patchserver %s: %s) r� rS �FEATURE_FLAGS_WHITELISTr� �bool�intr �__dict__r �kcarelog�info� ValueError�error)r� Zoptions_from_psrl r r r r �set_config_from_patchserver� s r� ) r� r` rN r r r r r �py23r �Optional�Set�ListrP r� re r� r r� rW r r" rf r_ ro �objectrp r� r� r� r r r r �<module> st �- (