Current File : //snap/core20/2599/usr/share/subiquity/console_conf/models/__pycache__/systems.cpython-38.pyc
U
K��^_ � @ sJ d dl Z d dlZd dlZd dlZe �d�Zdddddgddddd d
gd d idddd
dgd d id d id�d�d�dddddgd d id d id d id d id�d�ddd
ddgd d id d id d id�d�d�d�d�id�ZG dd � d �ZejG d d � d ��Z ejG d d � 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�objectTZarray�label�brand�model�type�string�title�mode�r r
)r �additionalProperties�required�
properties)r �items�id�username� display-name)r r r �
validation�brand-id)r r r )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 s2 || _ d | _d | _dd� |D �}|r.|d | _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)�selfZ systems_dataZcsr 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 s | j S r& r' r( r r r � selectiond s z RecoverySystemsModel.selectionc C s | j S r& )r r( r r r r h s z RecoverySystemsModel.currentc
C s� g }| D ]�}|d }|d }t |d |d |d d�}t|d |d |d |�dd �d
�}g }|�d g �D ] }|�t|d |d
d�� qjt|�dd�|d |||d�} |�| � qt|�S )Nr r r r �r �brand_id� display_namer r r Zunproven��IDr r- r r r r
r r Fr �r r r r r )� SystemModel�Brand�get�append� SystemAction�RecoverySystemr )
Zrecovery_systemsr Zsyst�m�br r r �ar r r r � from_systemsl s6 �
�
� z!RecoverySystemsModel.from_systemsc C st z&t �| �}t�|t� |�dg �}W nB t jk
rH t�d� � Y n" tj k
rh t�d� � Y nX t
� |�S )z4Deserialize recovery systems from input JSON stream.r z#cannot decode recovery systems infoz%cannot validate recovery systems data) �json�load�
jsonschemaZvalidate�_RECOVERY_SYSTEMS_SCHEMAr3 ZJSONDecodeError�logZ exceptionZValidationErrorr r: )Z
chooser_inputZdecr r r r �from_systems_stream� s
z(RecoverySystemsModel.from_systems_streamc C sF t | t�s td�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 r; �dump)�objZchooser_outputZchoicer r r �to_response_stream� s
��z'RecoverySystemsModel.to_response_streamN)�__name__�
__module__� __qualname__�__doc__r r% r) �propertyr* r � staticmethodr: r@ rG r r r r r P s
r c @ s< e Zd Ze�� Ze�� Ze�� Ze�� Ze�� Z dd� Z
dS )r6 c C s< | j |j ko:| j|jko:| j|jko:| j|jko:| j|jkS r& r0 �r �otherr r r �__eq__� s
�
�
�
�zRecoverySystem.__eq__N) rH rI rJ �attr�ibr r r r r rP r r r r r6 � s r6 c @ s4 e Zd Ze�� Ze�� Ze�� Ze�� Zdd� Z dS )r2 c C s0 | j |j ko.| j|jko.| j|jko.| j|jkS r& r. rN r r r rP � s
�
�
�z Brand.__eq__N)
rH rI rJ rQ rR r/ r r- r rP r r r r r2 � s
r2 c @ s, e Zd Ze�� Ze�� Ze�� Zdd� ZdS )r1 c C s$ | j |j ko"| j|jko"| j|jkS r& r+ rN r r r rP � s
�
�zSystemModel.__eq__N) rH rI rJ rQ rR r r, r- rP r r r r r1 � s r1 c @ s$ e Zd Ze�� Ze�� Zdd� ZdS )r5 c C s | j |j ko| j|jkS r& r rN r r r rP � s
�zSystemAction.__eq__N)rH rI rJ rQ rR r r
rP r r r r r5 � s r5 c @ s$ e Zd Ze�� Ze�� Zdd� ZdS )r$ c C s | j |j ko| j|jkS r& r! rN r r r rP � s
�zSelectedSystemAction.__eq__N)rH rI rJ rQ rR r" r# rP r r r r r$ � s r$ )Zloggingr; rQ r= Z getLoggerr? r> r r r6 r2 r1 r5 r$ r r r r �<module> sp
���
�� �������6]
Mini Shell