File: //lib/python3/dist-packages/DistUpgrade/__pycache__/telemetry.cpython-38.pyc
U
B��]�
� @ sB d dl Z d dlZd dlZd dlZd dlZdd� ZG dd� d�ZdS )� Nc C s t jdkrt � 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 | _| �� | _tjdddgtjdd��� d �� | j d< | � d� d | _
z&t d
��}|� � | j d < W 5 Q R X W n t
k
r� Y nX d S ) NZ lsb_releasez-rz-sT)�stdoutZuniversal_newlinesr ZFrom�startz/var/log/upgrade/telemetryz /var/log/installer/media-infoZ InstallMedia)�_metrics� _stages_hist�_get_current_uptime� _start_time�
subprocess�Popen�PIPEZ communicate�strip� add_stage�
_dest_path�open�readline�FileNotFoundError)�self�fr r r �__init__) s"
��
z_Telemetry.__init__c
C sp d}z,t d��}t|�� �� d �}W 5 Q R X W n: tttfk
rj } zt�dt |� � W 5 d}~X Y nX |S )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 Zuptimer �er r r r 8 s
"�z _Telemetry._get_current_uptimec C s6 | � � }| jdks|dkr dS || jt|| j �< dS )z(Record installer stage with current timeN)r r r
�int)r Z
stage_nameZnowr r r r C s z_Telemetry.add_stagec C s || j d< dS )zRecord updater typeZTypeN�r )r Z updater_typer r r �set_updater_typeJ s z_Telemetry.set_updater_typec C s || j d< dS )z*Record if the user had third party sourcesZThirdPartySourcesNr# )r Zusingr r r � set_using_third_party_sourcesN s z(_Telemetry.set_using_third_party_sourcesc
C s� | j | jd< tj�| j�}zbtj�|�s2t�|� t| jd��}t �
| j|� W 5 Q R X t� | jt j
t jB t jB t jB � W n4 tk
r� } zt�dt|� � W 5 d}~X Y nX dS )z<Close telemetry collection
Save to destination fileZStages�wz(Exception while storing telemetry data: N)r
r �os�path�dirnamer �exists�makedirsr �json�dump�chmod�stat�S_IRUSR�S_IWUSR�S_IRGRP�S_IROTHr r r r )r Z
target_dirr r! r r r �doneR s$
����z_Telemetry.done)
�__name__�
__module__� __qualname__r r r r r$ r% r4 r r r r r % s r )r r, r' r/ r
r r r r r r �<module> s