Current File : //snap/core22/current/usr/share/subiquity/console_conf/models/__pycache__/systems.cpython-310.pyc
o
K��^_ � @ sD d dl Z d dlZd dlZd dlZe �d�Zdddddgdddg d�d d
idddd d gd d
id d
id
�d�d�ddg d�d d
id d
id d
id d
id�d�ddg d�d d
id d
id d
id�d�d�d�d�id�ZG dd� d�ZejG dd� d��Z ejG dd� d��Z
ejG d d � d ��Z ejG d d � d ��Z ejG d d!� d!��Z
dS )"� Nzconsole_conf.models.systemsz'http://json-schema.org/draft-04/schema#�systems�objectT�array)�label�brand�model�type�string�title�mode�r
r )r �additionalProperties�required�
properties)r �items)�id�username� display-name)r r r �
validation)r �brand-idr )r �actionsr r )r r r )z$schemar
r r
r r c @ sd e Zd ZdZdd� Zdd� Zdd� Zedd � �Zed
d � �Z e
d d
� �Z e
dd� �Z e
dd� �Z
dS )�RecoverySystemsModelzRecovery chooser datac C s6 || _ d | _d | _dd� |D �}|r|d | _d S d S )Nc S s g | ]}|j r|�qS � )�current)�.0�sr r �3/usr/share/subiquity/console_conf/models/systems.py�
<listcomp>Z s z1RecoverySystemsModel.__init__.<locals>.<listcomp>r )r �
_selection�_current)�self� systems_data�csr r r �__init__S s �z RecoverySystemsModel.__init__c C s t ||d�| _d S )N��system�action)�SelectedSystemActionr )r r% r& r r r �select^ s zRecoverySystemsModel.selectc C s
d | _ d S �N�r �r r r r �unselecta s
z RecoverySystemsModel.unselectc C � | j S r) r* r+ r r r � selectiond � z RecoverySystemsModel.selectionc C r- r) )r r+ r r r r h r/ z RecoverySystemsModel.currentc
C s� g }| D ]U}|d }|d }t |d |d |d d�}t|d |d |d |�dd �d
�}g }|�d g �D ]}|�t|d |d
d�� q5t|�dd�|d |||d�} |�| � qt|�S )Nr r r r �r �brand_id� display_namer r r �unproven��IDr r2 r r r
r r r Fr �r r r r r )� SystemModel�Brand�get�append� SystemAction�RecoverySystemr )
�recovery_systemsr �syst�m�br r r �ar r r r � from_systemsl s6 �
�
� z!RecoverySystemsModel.from_systemsc C sh zt �| �}t�|t� |�dg �}W n t jy! t�d� � tj y. t�d� � w t
� |�S )z4Deserialize recovery systems from input JSON stream.r z#cannot decode recovery systems infoz%cannot validate recovery systems data) �json�load�
jsonschema�validate�_RECOVERY_SYSTEMS_SCHEMAr9 �JSONDecodeError�log� exception�ValidationErrorr rB )�
chooser_input�decr r r r �from_systems_stream� s
�
z(RecoverySystemsModel.from_systems_streamc C sF t | t�std�t| ����| jj| jj| jj d�d�}t
j ||d�S )z]Serialize an object with selected action as JSON to the given output
stream.
zunexpected type: {})r r
)r r& )�fp) �
isinstancer' � TypeError�formatr r% r r& r r
rC �dump)�obj�chooser_output�choicer r r �to_response_stream� s
��z'RecoverySystemsModel.to_response_streamN)�__name__�
__module__� __qualname__�__doc__r# r( r, �propertyr. r � staticmethodrB rN rW r r r r r P s
r c @ s< e Zd Ze�� Ze�� Ze�� Ze�� Ze�� Z dd� Z
dS )r< c C s< | j |j ko | j|jko | j|jko | j|jko | j|jkS r) r6 �r �otherr r r �__eq__� s
�
�
�
�zRecoverySystem.__eq__N) rX rY rZ �attr�ibr r r r r r` r r r r r< � s r< c @ s4 e Zd Ze�� Ze�� Ze�� Ze�� Zdd� Z dS )r8 c C s0 | j |j ko| j|jko| j|jko| j|jkS r) r4 r^ r r r r` � s
�
�
�z Brand.__eq__N)
rX rY rZ ra rb r5 r r2 r r` r r r r r8 � s r8 c @ s, e Zd Ze�� Ze�� Ze�� Zdd� ZdS )r7 c C s$ | j |j ko| j|jko| j|jkS r) r0 r^ r r r r` � s
�
�zSystemModel.__eq__N) rX rY rZ ra rb r r1 r2 r` r r r r r7 � s
r7 c @ �$ e Zd Ze�� Ze�� Zdd� ZdS )r; c C � | j |j ko | j|jkS r) r r^ r r r r` � �
�zSystemAction.__eq__N)rX rY rZ ra rb r
r r` r r r r r; � � r; c @ rc )r' c C rd r) r$ r^ r r r r` � re zSelectedSystemAction.__eq__N)rX rY rZ ra rb r% r& r` r r r r r' � rf r' )�loggingrC ra rE � getLoggerrI rG r r r< r8 r7 r; r' r r r r �<module> sr
���
�� �������6]
Mini Shell