CasperSecurity
o
>I�c�
� @ sB d dl Z d dlZd dlZd dlZd dlZdd� ZG dd� d�ZdS )� Nc C s t jdu r t � t _t jS )z'Return a singleton _Telemetry instance.N)�
_Telemetry�
_telemetry� r r �7/usr/lib/python3/dist-packages/DistUpgrade/telemetry.py�get s
r c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )r Nc C s� i | _ i | _| �� | _tjg d�tjdd��� d �� | j d< | � d� d| _
z t d��}|� � | j d < W d � W d S 1 s@w Y W d S t
yQ Y d S w )
N)� lsb_releasez-rz-sT)�stdout�universal_newlinesr �From�startz/var/log/upgrade/telemetryz /var/log/installer/media-info� InstallMedia)�_metrics� _stages_hist�_get_current_uptime� _start_time�
subprocess�Popen�PIPE� communicate�strip� add_stage�
_dest_path�open�readline�FileNotFoundError)�self�fr r r �__init__) s(
���
&� �z_Telemetry.__init__c
C s� d}z#t d��}t|�� �� d �}W d � W |S 1 s w Y W |S tttfyD } zt�dt |� � W Y d}~|S d}~ww )z6Get current uptime info. None if we couldn't fetch it.Nz /proc/uptimer z)Exception while fetching current uptime: )
r �float�read�splitr �OSError�
ValueError�logging�warning�str)r �uptimer �er r r r 8 s
������z _Telemetry._get_current_uptimec C s6 | � � }| jdu s
|du rdS || jt|| j �< dS )z(Record installer stage with current timeN)r r r �int)r �
stage_name�nowr r r r C s z_Telemetry.add_stagec C � || j d< dS )zRecord updater type�TypeN�r
)r � updater_typer r r �set_updater_typeJ � z_Telemetry.set_updater_typec C r+ )z*Record if the user had third party sources�ThirdPartySourcesNr- )r �usingr r r � set_using_third_party_sourcesN r0 z(_Telemetry.set_using_third_party_sourcesc
C s� | j | jd< tj�| j�}z<tj�|�st�|� t| jd��}t �
| j|� W d � n1 s1w Y t� | jt j
t jB t jB t jB � W dS tye } zt�dt|� � W Y d}~dS d}~ww )z<Close telemetry collection
Save to destination file�Stages�wNz(Exception while storing telemetry data: )r r
�os�path�dirnamer �exists�makedirsr �json�dump�chmod�stat�S_IRUSR�S_IWUSR�S_IRGRP�S_IROTHr! r# r$ r% )r �
target_dirr r' r r r �doneR s*
�
��
����z_Telemetry.done)
�__name__�
__module__� __qualname__r r r r r/ r3 rD r r r r r % s r )r# r; r6 r> r r r r r r r �<module> s