U
K��^ � @ sD d dl Z d dlZd dlZd dlmZmZ e �d�ZG dd� d�ZdS )� N)�StoredDataObserver� UdevObserverzsubiquitycore.proberc @ s& e Zd Zdd� Zdd� Zddd�ZdS ) �Proberc C sF d | _ |r*t|��}t�|�| _ W 5 Q R X || _t�d�| j �� d S )Nz Prober() init finished, data:{})� saved_config�open�yamlZ safe_load� debug_flags�log�debug�format)�selfZmachine_configr Zmc� r
�,/usr/share/subiquity/subiquitycore/prober.py�__init__ s
zProber.__init__c C s0 | j d k r t| j d |�}nt|�}||�� fS )NZnetwork)r r r �start)r ZreceiverZobserverr
r
r �
probe_network$ s
�zProber.probe_networkNc C s� | j d k rpd}|d k rd}|| jkr6t�d� dd | j d �� }|d k rl| j d D ]}||krVi ||< qV|S ddlm} |� j|d�S ) Nz bpfail-fullzbpfail-restricted� � r Zstorage)�Storage)� probe_types)r r �time�sleep�copyZprobert.storager Zprobe)r r �flag�r�kr r
r
r � get_storage, s
zProber.get_storage)N)�__name__�
__module__� __qualname__r r r r
r
r
r r s r ) Zloggingr r Zprobert.networkr r Z getLoggerr r r
r
r
r �<module> s
|