3
e�$g" � @ sR yd dl mZmZ W n ek
r0 eZeZY nX 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|kr0|d | _nd | _d | _d S )Nzinfo.udiz info.parent)�udi�
properties�children�classification�
parent_udi�parent)�selfr � r � /usr/lib/python3.6/haltree.py�__init__" s
zHalDevice.__init__c C s6 t | jd� x$| jj� D ]\}}t d|d|� qW d S )N�:z 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|j �}|r*|jj|� ||_q>| jj|� n|| _| j|� d S )N)r �_HalTree__find_noder �appendr
r r �_HalTree__get_lost_children)r �
hal_devicer
r r r
�add> s z HalTree.addc C sR g }g }| j d d � }x6|D ].}|j|jkr ||_|jj|� | j j|� q W d S )N)r r r r
r r �remove)r r Z
found_listZindexesZno_parent_yet_copyZdevr r r
Z__get_lost_childrenM s
zHalTree.__get_lost_childrenc C sD | j r tj| j |�}|r |S x"| jD ]}tj||�}|r$|S q$W 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 x"| jD ]}tj||�}|r|S qW d S )N)r r r r" )r# r Zdevice�resr r r
Z__find_node_workerg s
zHalTree.__find_node_workerc C s | j | 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� x�|jj� D ]�\}}t|�tkrz|j� rbt |d d|tt |��f � q�t |d d||f � q&t|�t
kr�t |d d|tt |��f � q&t |d d||f � q&W t x |j D ]}| j ||d � q�W d S )NzCLASS:z z %-20s ==> %s)
r r r r r �typer �isdigit�hex�intr r r& )r r# �indent�namer Zchildr 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