| Current File : //snap/core22/2115/usr/share/subiquity/subiquitycore/__pycache__/snapd.cpython-310.pyc |
o
K��^d � @ s� d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlZd dlm Z m
Z
d dl m Z d dl
mZ d dlZe�d�ZG dd� d�ZG d d
� d
�ZG d d � d �ZG d
d� d�ZG dd� d�ZG dd� d�ZdS )� N)�partial)�
quote_plus� urlencode)�
run_in_thread)� run_commandzsubiquitycore.snapdc @ �, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�SnapdConnectionc C s || _ d�t|��| _d S )Nzhttp+unix://{}/)�root�formatr �url_base)�selfr �sock� r �+/usr/share/subiquity/subiquitycore/snapd.py�__init__' s zSnapdConnection.__init__c K sT |r
|dt |� 7 }t�� �}|j| j| dd�W d � S 1 s#w Y d S )N�?�< )�timeout)r �requests_unixsocket�Session�getr )r �path�args�sessionr r r r + s
$�zSnapdConnection.getc K s\ |r
|dt |� 7 }t�� �}|j| j| t�|�dd�W d � S 1 s'w Y d S )Nr r )�datar )r r r �postr �json�dumps)r r �bodyr r r r r r 1 s
�$�zSnapdConnection.postc C s� t �d� tj�| jd�}tj|dd� ttj�|d�d��}|�|� � � W d � n1 s0w Y | jdkrCdd gg d
�g}nd d gg}|D ]}t
|� qJd S )
Nz(restarting snapd to pick up proxy configz"etc/systemd/system/snapd.service.dT)�exist_okzsnap_proxy.conf�w�/� systemctlz
daemon-reload)r" �restartz
snapd.service�sleep�2) �log�debug�osr �joinr �makedirs�open�write�proxy_systemd_dropinr )r �proxy�
dropin_dir�fp�cmds�cmdr r r �configure_proxy9 s
� �
�
�z SnapdConnection.configure_proxyN)�__name__�
__module__� __qualname__r r r r3 r r r r r & s
r c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�_FakeFileResponsec C �
|| _ d S �N)r )r r r r r r M �
z_FakeFileResponse.__init__c C � d S r: r �r r r r �raise_for_statusP � z"_FakeFileResponse.raise_for_statusc C s8 t | j��
}t�|�W d � S 1 sw Y d S r: )r+ r r �load)r r0 r r r r S s $�z_FakeFileResponse.jsonN�r4 r5 r6 r r> r r r r r r8 K � r8 c @ r7 )�_FakeMemoryResponsec C r9 r: �r )r r r r r r Z r; z _FakeMemoryResponse.__init__c C r< r: r r= r r r r> ] r? z$_FakeMemoryResponse.raise_for_statusc C s | j S r: rD r= r r r r ` s z_FakeMemoryResponse.jsonNrA r r r r rC X rB rC c @ s e Zd ZdZdd� Zdd� ZdS )� ResponseSetzqResponses for a endpoint that returns different data each time.
Motivating example is v2/changes/$change_id.c C s || _ d| _d S )Nr )�files�index)r rF r r r r i s
zResponseSet.__init__c C sh | j | j }ttj�dd��}|dkr)| j| t| j �kr)t| j �d | _t|�S | j|7 _t|�S )N�SUBIQUITY_REPLAY_TIMESCALE� )rF rG �intr( �environr �lenr8 )r �f�dr r r �nextm s �zResponseSet.nextN)r4 r5 r6 �__doc__r rO r r r r rE d s rE c @ r )
�FakeSnapdConnectionc C s || _ || _i | _|| _d S r: )�
snap_data_dir� scale_factor�
response_sets� output_base)r rR rS rU r r r r z s
z FakeSnapdConnection.__init__c C s t �d� t�d| j � d S )Nz3pretending to restart snapd to pick up proxy config� )r& r' �timer$ rS )r r. r r r r3 � s
z#FakeSnapdConnection.configure_proxyc K sv |dkr |d dkr | j d }t|d��� tdddd d
��S |dkr2|d d kr2tdd dd
d
��S td�||f���)Nzv2/snaps/subiquity�action�refreshz/run/subiquity/updatingr �async�7�� �OK)�type�changez status-code�status�switch�8�Acceptedz*Don't know how to fake POST response to {})rU r+ �closerC � Exceptionr
)r r r r �update_marker_filer r r r � s&
�� �zFakeSnapdConnection.postc K s� t �d| j � |�dd�}|r |dtt|�� �� 7 }|| jv r(| j| �� S t j
� | j |�}t j
�
|d �r>t|d �S t j
�|�r]tt�t j
� |d���}t|� }| j|< |�� S td�||f���)NrI r! �-z.jsonz*.jsonz)Don't know how to fake GET response to {})rW r$ rS �replacer �sorted�itemsrT rO r( r r) rR �existsr8 �isdir�globrE re r
)r r r �filename�filepathrF �rsr r r r � s
�zFakeSnapdConnection.getN)r4 r5 r6 r r3 r r r r r r rQ y s
rQ c @ r )
�
AsyncSnapdc C r9 r: )�
connection)r rr r r r r � r; zAsyncSnapd.__init__c � s2 �t t| jj|fi |���I d H }|�� |�� S r: )r r rr r r> r )r r r �responser r r r � s �
�zAsyncSnapd.getc � s8 �t t| jj||fi |���I d H }|�� |�� d S )Nr_ )r r rr r r> r )r r r r rs r r r r � s �
� zAsyncSnapd.postc � s^ �| j ||fi |��I d H }d�|�} | �|�I d H }|d d dkr&d S t�d�I d H q)Nz
v2/changes/{}T�resultr` �Doneg�������?)r r
r �asyncior$ )r r r r r_ � change_pathrt r r r �
post_and_wait� s �
�zAsyncSnapd.post_and_waitN)r4 r5 r6 r r r rx r r r r rq � s
rq )rv � functoolsr rm r �loggingr( rW � urllib.parser r �subiquitycore.async_helpersr �subiquitycore.utilsr r � getLoggerr& r r8 rC rE rQ rq r r r r �<module> s$
%
2