3
�h>`P? �
@ s d dl mZ dddddddd d
d d d
dg
Zd dlZd dlT d dlZd dlZd dlmZ d dl T d dl
T d dl m Z m
Z
edd�Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd d � Zd d � Zd d
� ZG d!d"� d"e�Z e e �G d#d� de ��Z dS )$� )�absolute_import�string_to_xmlnode�string_to_cdata_xmlnode�validate_database_doc�boolean�xml_attributes�xml_attribute_dict�xml_child_elements_iter�xml_child_elements� xml_get_child_element_by_name� xml_get_child_elements_by_name�xml_child_element_names�xml_has_child_elements� XmlSerializeN)�*)�
get_config)�
add_metaclass� string_typesZgeneral�
i18n_encodingc C sZ | d krt d� dS | j� }|d kr0t d� dS |jd�}|d krNt d� dS t|�S d S )Nz3validate_database_doc: doc is empty, validate failsFz4validate_database_doc: root is empty, validate fails�versionz7validate_database_doc: version is empty, validate fails)� log_debug�getRootElement�propZdatabase_version_compatible)�doc� root_noder � r �#/usr/lib/python3.6/xml_serialize.pyr > s
c C sh t | t�r| S t | t�rF| j� } | d kr,dS | d kr8dS td | ��n t | t�rXt| �S td | ��d
S )
zconvert value to bool�t�true�1T�f�false�0Fz cannot convert (%s) to booleanN)r r r )r r! r" )�
isinstance�boolr �lower�
ValueError�int)�valuer r r r N s
c C s t jt|��S )N)�libxml2ZnewText�str)r r( r r r r ` s c C s | j |t|��S )N)Z
newCDataBlock�len)r r( r r r r d s c c s0 | j � }x"|r*|j� |j� fV |j� }q
W d S )N)Zget_properties�get_nameZ get_content�get_next)�noder r r r r t s c C s& i }x t | �D ]\}}|||< qW |S )N)r )r. Zprops�namer( r r r r { s c c s0 | j � }x"|r*|j� dkr |V |j� }q
W d S )N�element)� get_children�get_typer- )r. �childr r r r � s
c C s: | j � }x,|r4|j� dkr*|j� |kr*|S |j� }q
W d S )Nr0 )r1 r2 r, r- )r. r/ r3 r r r r � s c C sD g }| j � }x2|r>|j� dkr4|j� |kr4|j|� |j� }qW |S )Nr0 )r1 r2 r, �appendr- )r. r/ �elementsr3 r r r r � s
c C s t t| ��S )N)�listr )r. r r r r
� s c C s dd� t | �D �S )Nc S s g | ] }|j � �qS r )r, )�.0�er r r �
<listcomp>� s z+xml_child_element_names.<locals>.<listcomp>)r )r. r r r r
� s c C s. | j � }x |r(|j� dkr dS |j� }q
W dS )Nr0 TF)r1 r2 r- )r. r3 r r r r � s c s$ e Zd Zdd� Z� fdd�Z� ZS )�XmlSerializeMetaDatac s� |dkrt j| |||�S |jd��|jdd ��|jdd �� �d krjx&|D ] }|jjdd ���d k rHP qHW n � ��fdd�}||d< t j| |||�S )Nr �__init__�init_from_xml_node�_init_postprocessc sJ t | �dkr<t| d tj�r<�| |� � d k rF� | d � n
�| |� d S )N� � r )r+ r# r) ZxmlNode)�args�kwds)r= � normal_init�xml_initr r � wrapped_init� s
z2XmlSerializeMetaData.__new__.<locals>.wrapped_init)�type�__new__�get�__dict__)�cls� classname�bases� classdictZbase_clsrD r )r= rB rC r rF � s
z XmlSerializeMetaData.__new__c s� t t| �j|||� |jd�� � s&d S � dkr6d| _njd| _� fdd�t� j� �D �| _� fdd�t� j� �D �| _| j| j | _ | jj
� | jj
� | j j
� d S )N� _xml_info� unstructuredTFc s g | ]}� | d dkr|�qS )�XMLFormr0 r )r7 �x)�xml_infor r r9 � s z1XmlSerializeMetaData.__init__.<locals>.<listcomp>c s g | ]}� | d dkr|�qS )rO Z attributer )r7 rP )rQ r r r9 � s ) �superr: r; rG �
_unstructuredr6 �keys� _elements� _attributes�_names�sort)rI rJ rK rL )� __class__)rQ r r; � s
z XmlSerializeMetaData.__init__)�__name__�
__module__� __qualname__rF r; �
__classcell__r r )rY r r: � s r: c @ sr e Zd Zdd� Zdd� Zdd� Zdd� Zdd
d �Zdd d
�Zddd�Z ddd�Z
d dd�Z d dd�Z d dd�Z
d S ) r c C s | j � d S )N)�_init_defaults)�selfr r r r; � s zXmlSerialize.__init__c C s | j � S )N)�get_xml_text_doc)r_ r r r �__str__� s zXmlSerialize.__str__c C st | j dkrd S x`| jD ]V}| j | }|jdd �}|d k rHt| ||� � q|jd�r`t| |g � qt| |d � qW d S )NrN �defaultr6 )rM rW rG �setattr)r_ r/ � name_inforb r r r r^ � s
zXmlSerialize._init_defaultsc C s8 | j r$dd� t| jj� �D �}g }n | j}| j}||fS )Nc S s g | ]}|j d �s|�qS )�_)�
startswith)r7 rP r r r r9 s z<XmlSerialize.get_elements_and_attributes.<locals>.<listcomp>)rS r6 rH rT rU rV )r_ r5 �
attributesr r r �get_elements_and_attributes� s z(XmlSerialize.get_elements_and_attributesNc C s$ t jd�}| j||�}|j|� |S )Nz1.0)r) ZnewDoc�
get_xml_nodesZsetRootElement)r_ �obj_namer �rootr r r � get_xml_doc s
zXmlSerialize.get_xml_docc C s< d }}z | j |�}|jtdd�}W d |d k r6|j� X |S )Nr? )�encoding�format)rl Z serializer �freeDoc)r_ rj r Ztext_docr r r r` s
z XmlSerialize.get_xml_text_docc C s| d }z`y&t j|j� �}|j� }| j||� W n4 t jk
r` } ztjtjd| � d S d }~X nX W d |d k rv|j� X d S )Nz"read_xml() libxml2.parserError: %s) r) ZparseDoc�stripr r< � parserError�syslog�LOG_ERRro )r_ Zbufrj r r r8 r r r �read_xml s zXmlSerialize.read_xmlc C s� d }z�y*t j|�}|r"||�s"dS | j||� W nd t jk
rd } ztjtjd| � dS d }~X n2 tk
r� } ztjtjd| � dS d }~X nX W d |d k r�|j� X dS )NFz'read_xml_file() libxml2.parserError: %szread_xml_file() error: %sT)r) Z parseFiler< rq rr rs � Exceptionro )r_ Zxmlfilerj Z validate_docr r8 r r r �
read_xml_file& s"
zXmlSerialize.read_xml_filec C s� yfd}|d krt j}n2t|t�r0t|d�}d}nt|t�r<n td| ��|j| j|�� |rd|j � W n8 t
k
r� } z t j t j d||f � W Y d d }~X nX d S )NF�wTzbad file parameter %szcould not write %s: %s)
�sys�stdoutr# r �openZFileTyper& �writer` �closeru rr rs )r_ rj r Z
need_to_closer8 r r r � write_xml: s
zXmlSerialize.write_xmlc C s� | j � \}}|d kr | jj}tj|�}xD|D ]<}| j| }|jdt�}t| |�} | d k r,|j ||| �� q,W �xv|D �]l}�y$| jdkr�t
}d }
n | j| }|jdt
�}|jd�}
t| |�} | d ks�t | t �r�t
| �dkr�wt|
�rRtj|�} |j| � x�| D ]P} t | t��r$| j||
�}
| j|
� q�tj|
�}| j|� ||| �}
|j|
� q�W nLt | t��rv| j||�}
|j|
� n(tj|�} |j| � ||| �}
| j|
� W qt tk
�r� } z"tjtjd| jj|| f � W Y d d }~X qtX qtW |S )NZexport_typecastrN r6 r z%s.%s value=%s)rh rY rZ r) ZnewNoderM rG r* �getattrZsetPropr r# r6 r+ ZaddChildr ri ru rr rs )r_ r rj r5 rg rk r/ rd �typecastr( �list_item_name� element_node�itemr3 Zlist_item_noder8 r r r ri M sT
2zXmlSerialize.get_xml_nodesc C s� | j � \}}| j� |d kr"|}n&t||�}|d krHtd||j� f ��xzt|�D ]n\}}||krxtd||j� f � qR| j| }|jdt �} t
| t �r�t | t
�r�td| ��qR| j|| |�� qRW �x0t|�D �]"}
|
j� }| jr�|
j� }| j||� q�||k�rtd||j� f � q�| j| }|jdt �} |jd�} | �r�t| |g �} t|
| �}
t
| t ��r�t | t
��r�x@|
D ]}| j| |�� �qtW n"xd|
D ]}| j| |j� �� �q�W q�t
| t ��r�t | t
��r�| j|| |
�� q�|
j� }| j|| |�� q�W d S )Nz+xml child element (%s) not found in node %sz0unknown attribute (%s) found in xml element (%s)Zimport_typecastz-Illegal use of substructure in attribute (%s)z.unknown element (%s) found in xml element (%s)r6 )rh r^ r �KeyErrorr, r r rM rG r* r# rE �
issubclassr r& � __setattr__r rS Z
getContentr~ r r4 )r_ Zxml_noderj r5 rg rk r/ r( rd r r� r� �attrZ
list_nodesZ list_noder r r r< � sN
z XmlSerialize.init_from_xml_node)N)N)N)NN)NN)N)N)rZ r[ r\ r; ra r^ rh rl r` rt rv r} ri r< r r r r r � s
7) Z
__future__r �__all__rx �typesr) rr Zsetroubleshoot.configr Zsetroubleshoot.errcodeZsetroubleshoot.utilZsixr r r r r r r r r r r r r
r
r rE r: �objectr r r r r �<module> sJ
5