| Current File : //snap/core22/current/usr/share/subiquity/subiquitycore/__pycache__/utils.cpython-310.pyc |
o
K��^� � @ s> d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ e�d�Z dd� Z
dd� Z dej ej dd dd
�d ee
d ejfd
d�Zej ej ddd ddd�d ee
d ejfdd�Zej ej ejdd�d ee
d e jjfdd�Zd ee
de
d ee
fdd�Zejej ej dd dd�d ee
d ejfdd�Zd!d d �Zd d � ZdS )"� N)�Listzsubiquitycore.utilsc C s( | d u r
t j�� } n| �� } d| d< | S )N�C�LC_ALL)�os�environ�copy)�env� r �+/usr/share/subiquity/subiquitycore/utils.py�
_clean_env s
r c C sb | d u rt j} | �� }| �� D ] \}}|�d�r.|d td� � }|r(|||< n||= ||= q|S )N�_ORIG)r r r �items�endswith�len)r �ret�key�val�key_to_restorer r r
� orig_environ% s
�r zutf-8�replace)�input�stdout�stderr�encoding�errorsr �cmd�returnc
K s� |du r
t j|d< n|�|�}t�d| � z.t j| f|||t|�d�|��}|rBt|jt �r5|j�
|�|_t|j t �rB|j �
|�|_ W n t j yY } z t�dt
| �� � d} ~ ww t�d|j|j� |S )z�A wrapper around subprocess.run with logging and different defaults.
We never ever want a subprocess to inherit our file descriptors!
N�stdinzrun_command called: %s)r r r r zrun_command %sz"run_command %s exited with code %s)�
subprocess�DEVNULL�encode�log�debug�runr �
isinstancer �bytes�decoder �CalledProcessError�str�args�
returncode)
r r r r r r r �kw�cp�er r r
� run_command4 s.
�� ���r. F)r r r r r r �checkc
� s� �|d u rd|vrt j|d< n
t j|d< |�|�}t�d| � tj| ||t|�d�|��I d H } | j |d�I d H \}}|rO|d urF|�
|�}|d urO|�
|�}t�d| | j � |re| j dkret � | j | ��t �
| | j ||�S )Nr zarun_command called: %s�r r r )r z#arun_command %s exited with code %sr )r r �PIPEr r! r" �asyncio�create_subprocess_execr � communicater& r* r' �CompletedProcess)
r r r r r r r r/ r+ �procr r r
� arun_commandP s2 �
�
� �
�r7 )r r r r c � s0 �t �d| � tj| ||t|�d�|��I d H S )Nzastart_command called: %sr0 )r! r" r2 r3 r )r r r r r r+ r r r
�astart_commandl s � � �r8 �split_onc � s �t | dd�I d H }|j�|�S )NT)r/ )r7 r �split)r r9 r, r r r
�split_cmd_outputu s � r; )r r r r r r c K s, t �d| � tj| f|||t|�d�|��S )z�A wrapper around subprocess.Popen with logging and different defaults.
We never ever want a subprocess to inherit our file descriptors!
zstart_command called: %s)r r r r )r! r" r �Popenr )r r r r r r r r+ r r r
�
start_commandz s ��r= �SHA-512c sd ddddd�}||vrt d�|d�|�� ����d� d }d�� fd
d �|D ��}t�| || | �S ) Nz$6$z$5$z$1$� )r> zSHA-256�MD5�DESz&Invalid algo({}), must be one of: {}. �,z@abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789./z c s g | ]}t �� ��qS r )�random�choice)�.0�c��salt_setr r
�
<listcomp>� s z"crypt_password.<locals>.<listcomp>)� Exception�format�join�keys�crypt)�passwd�algo�algos�saltr rG r
�crypt_password� s �rS c C s> t �d� tg d�� tddg� tg d�� tg d�� dS )z; Stop subiquity service; which also restores getty service zdisabling subiquity service)�mkdirz-pz/run/subiquity�touchz/run/subiquity/complete)� systemctl�start�
--no-blockzgetty@tty1.service)rV �stoprX z(snap.subiquity.subiquity-service.servicezserial-subiquity@*.serviceN)r! �infor. r r r r
�disable_subiquity� s
r[ )r> )r2 rN �loggingr rC r �typingr � getLoggerr! r r r1 r( r5 r. r7 r �Processr8 r; r< r= rS r[ r r r r
�<module> sJ
�
� �
� �
� �
�