File: //lib/python3.9/site-packages/up2date_client/__pycache__/haltree.cpython-39.pyc
a
c8h" � @ sP zd dl mZmZ W n ey. eZeZY n0 G dd� d�ZG dd� d�ZdS )� )�
StringType�IntTypec @ s e Zd ZdZdd� Zdd� ZdS )� HalDevicez5An object containing its udi, properties and childrenc C s@ |d | _ || _g | _d | _d|v r0|d | _nd | _d | _d S )Nzinfo.udiz info.parent)�udi�
properties�children�classification�
parent_udi�parent)�selfr � r �:/usr/lib/python3.9/site-packages/up2date_client/haltree.py�__init__" s
zHalDevice.__init__c C s2 t | jd� | j�� D ]\}}t d|d|� qd S )N�:� z ==> )�printr r �items)r �property�valuer r r
�print_properties0 s zHalDevice.print_propertiesN)�__name__�
__module__� __qualname__�__doc__r r r r r r
r s r c @ sH e Zd Zdd� Zdd� Zdd� Zdd� Zed d
� �Zd d � Z d
d� Z
dS )�HalTreec C s d | _ g | _d S �N)�head�
no_parent_yet�r r r r
r 9 s zHalTree.__init__c C sL |j r8| �|j �}|r*|j�|� ||_q>| j�|� n|| _| �|� d S r )r �_HalTree__find_noder �appendr
r r �_HalTree__get_lost_children)r �
hal_devicer
r r r
�add> s z HalTree.addc C sN g }g }| j d d � }|D ].}|j|jkr||_|j�|� | j �|� qd S r )r r r r
r r �remove)r r" Z
found_listZindexesZno_parent_yet_copy�devr r r
Z__get_lost_childrenM s zHalTree.__get_lost_childrenc C sD | j r t�| j |�}|r |S | jD ] }t�||�}|r"| S q"dS )zo
This takes a node in the HalDevice tree and returns the HalDevice with
the given udi.
N)r r �_HalTree__find_node_workerr )r r �nodeZ
found_noder r r
Z __find_nodeW s
zHalTree.__find_nodec C s6 | j |kr| S | jD ] }t�||�}|r| S qd S r )r r r r& )r' r Zdevice�resr r r
Z__find_node_workerg s
zHalTree.__find_node_workerc C s | � | jd� d S )N� )�_HalTree__print_dev_treer r r r r
�
print_treeq s zHalTree.print_treec C s� t ||j� t |d|j� |j�� D ]�\}}t|�tkrx|�� r`t |d d|tt |��f � q�t |d d||f � q$t|�t
kr�t |d d|tt |��f � q$t |d d||f � q$t |j D ]}| � ||d � q�d S )NzCLASS:r z %-20s ==> %s)
r r r r r �typer �isdigit�hex�intr r r* )r r' �indent�namer �childr r r
Z__print_dev_treet s
zHalTree.__print_dev_treeN) r r r r r# r! r � staticmethodr& r+ r* r r r r
r 8 s
r N)�typesr r � ImportError�bytesr/ r r r r r r
�<module> s