OwlCyberSecurity - MANAGER
Edit File: xml_serializer.cpython-39.pyc
a ���e�G � @ s d Z ddlZddlmZ ddlmZ ddlmZ ddl m Z ddl m Z ddl mZ dd lmZ dd lmZmZ dd lmZmZ G d d � d ej�ZG dd� dej�Zdd� ZG dd� de�ZG dd� de�Z G dd� de �Z G dd� de �Z G dd� de �Z dS ) z XML serializer. � N)�pulldom)�handler)� ExpatParser)�apps)�settings)�ObjectDoesNotExist)�base)�DEFAULT_DB_ALIAS�models)�SimplerXMLGenerator�UnserializableContentErrorc @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dS )� Serializerz Serialize a QuerySet to XML.c C s4 | j �d�d ur0| j�dd| j �d� | � d S )N�indent� � )�options�get�xmlZignorableWhitespace)�self�level� r �g/home/gouroczh/virtualenv/pat/3.9/lib/python3.9/site-packages/django/core/serializers/xml_serializer.pyr s �zSerializer.indentc C s: t | j| j�dtj��| _| j�� | j�dddi� dS )zT Start serialization -- open the XML document and the root element. �encoding�django-objects�versionz1.0N) r �streamr r r �DEFAULT_CHARSETr Z startDocument� startElement�r r r r �start_serialization s � z Serializer.start_serializationc C s$ | � d� | j�d� | j�� dS )z8 End serialization -- end the document. r r N)r r � endElementZ endDocumentr r r r �end_serialization$ s z Serializer.end_serializationc C sp t |d�s t�dt|� ��| �d� dt|j�i}| jrDt |d�s^|j}|dur^t|�|d< | j � d|� dS ) z3 Called as each object is handled. �_metaz6Non-model object (%s) encountered during serialization� �model� natural_keyN�pk�object) �hasattrr ZSerializationError�typer �strr" Zuse_natural_primary_keysr& r r )r �obj�attrs�obj_pkr r r � start_object, s � zSerializer.start_objectc C s | � d� | j�d� dS )zA Called after handling all fields for an object. r# r' N)r r r )r r+ r r r � end_object> s zSerializer.end_objectc C s� | � d� | j�d|j|�� d�� t||j�dur�|�|�}|�� dkrZtj||j d�}z| j� |� W q� t y� t d|j j|j|jf ��Y q�0 n | j�d� | j�d� dS ) zg Handle each field on an object (except for ForeignKeys and ManyToManyFields). � �field)�namer) N� JSONField��clsz0%s.%s (pk:%s) contains unserializable characters�None)r r r r2 �get_internal_type�getattrZvalue_to_string�json�dumps�encoder� charactersr � ValueError� __class__�__name__r& �addQuickElementr )r r+ r1 �valuer r r � handle_fieldE s* �� �� zSerializer.handle_fieldc C s� | � |� t||�� �}|dur�| jr~t|jjd�r~t||j�}|�� }|D ].}| j � di � | j � t |�� | j � d� qLq�| j � t |�� n | j �d� | j � d� dS )zq Handle a ForeignKey (they need to be treated slightly differently from regular fields). Nr% �naturalr6 r1 )�_start_relational_fieldr8 Z get_attname�use_natural_foreign_keysr( � remote_fieldr$ r2 r% r r r< r* r r@ )r r+ r1 Z related_att�related� key_valuer r r �handle_fk_fieldf s � zSerializer.handle_fk_fieldc s� |j jjjr�� �|� � jr@t|j jd�r@� fdd�}dd� }n� fdd�}dd� }t|di �� |j |||��}|D ] }||� qt� j � d � d S ) z� Handle a ManyToManyField. Related objects are only serialized as references to the object's PK (i.e. the related *data* is not dumped, just the relation). r% c sZ | � � }� j�di � |D ].}� j�di � � j�t|�� � j�d� q� j�d� d S )Nr' rC )r% r r r< r* r )rA rC rH r r r � handle_m2m� s z/Serializer.handle_m2m_field.<locals>.handle_m2mc S s t | |j��� S �N)r8 r2 �iterator�r+ r1 r r r �queryset_iterator� s z6Serializer.handle_m2m_field.<locals>.queryset_iteratorc s � j jddt| j�id� d S )Nr' r& )r, )r r@ r* r& )rA r r r rJ � s c S s t | |j��d ��d��� S �Nr& )r8 r2 Zselect_related�onlyrL rM r r r rN � s ���Z_prefetched_objects_cacher1 N) rF Zthroughr" Z auto_createdrD rE r( r$ r8 r r2 r r )r r+ r1 rJ rN Zm2m_iterZrelobjr r r �handle_m2m_field s � � zSerializer.handle_m2m_fieldc C s6 | � d� | j�d|j|jjjt|jjj �d�� dS )z1Output the <field> element for relational fields.r0 r1 )r2 �rel�toN) r r r r2 rF r> r? r* r$ r" )r r1 r r r rD � s ��z"Serializer._start_relational_fieldN) r? � __module__� __qualname__�__doc__r r r! r. r/ rB rI rQ rD r r r r r s !/r c sX e Zd ZdZedd�� fdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z � Z S )� DeserializerzDeserialize XML.F)�using�ignorenonexistentc sF t � j|fi |�� |�dd�| _t�| j| �� �| _|| _ || _ d S )N�handle_forward_referencesF) �super�__init__�poprZ r �parser � _make_parser� event_stream�db�ignore)r Zstream_or_stringrX rY r �r> r r r\ � s zDeserializer.__init__c C s t � S )z;Create a hardened XML parser (no custom/external entities).)�DefusedExpatParserr r r r r_ � s zDeserializer._make_parserc C sD | j D ]4\}}|dkr|jdkr| j �|� | �|� S qt�d S )NZ START_ELEMENTr' )r` �nodeNameZ expandNode�_handle_object� StopIteration)r �event�noder r r �__next__� s zDeserializer.__next__c C s� | � |d�}i }|�d�r8|jj�|�d��||jjj< i }i }dd� |j�� D �}|�d�D �]*}|�d�}|s|t � d��| j r�||vr�q^|j� |�} | j r�t| j tj�r�| �|| �} | t jkr�dd � |�d �D �|| < n | || j< q^| j �r>t| j tj��r>| �|| �} | t jk�r2d d � |�d �D �|| < n | || j< q^|�d ��rPd} n0| �t|��� �} | �� dk�r�tj| | jd�} | || j< q^t � ||| j �} t � | ||�S )z1Convert an <object> node to a DeserializedObject.r$ r& c S s h | ] }|j �qS r )r2 )�.0�fr r r � <setcomp>� � z.Deserializer._handle_object.<locals>.<setcomp>r1 r2 z,<field> node is missing the 'name' attributec S s g | ]}d d� |� d�D ��qS )c S s g | ]}t |��� �qS r �� getInnerText�strip)rk Znat_noder r r � <listcomp> s �z:Deserializer._handle_object.<locals>.<listcomp>.<listcomp>rC )�getElementsByTagName)rk Zobj_noder r r rr s ��z/Deserializer._handle_object.<locals>.<listcomp>r' c S s g | ]}t |��� �qS r ro �rk �kr r r rr s �rC r6 Nr3 r4 ) �_get_model_from_nodeZ hasAttributer" r& � to_python� getAttributeZattnameZ get_fieldsrs r �DeserializationErrorrb � get_fieldrF � isinstancer Z ManyToManyRel�_handle_m2m_field_node� DEFER_FIELDr2 Z ManyToOneRel�_handle_fk_field_noderp rq r7 r9 �loads�decoderZbuild_instancera ZDeserializedObject) r ri ZModel�dataZm2m_dataZdeferred_fields� field_namesZ field_node� field_namer1 rA r+ r r r rf � sX � � � � � � zDeserializer._handle_objectc C s� |� d�rdS |jj}t|jd�r�|� d�}|r�dd� |D �}z|j�| j�j|� }W n$ tyz | j rtt j Y S � Y n0 t ||jj �}|jjjjjr�|j}n"t|��� }|j�|jj ��|�}|S t|��� }|j�|jj ��|�S dS )z8 Handle a <field> node for a ForeignKey r6 N�get_by_natural_keyrC c S s g | ]}t |��� �qS r ro rt r r r rr 1 rn z6Deserializer._handle_fk_field_node.<locals>.<listcomp>)rs rF r$ r( �_default_manager� db_managerra r� r rZ r r} r8 r� r" r& rp rq rz rw )r ri r1 r$ �keys� field_valuer+ r- r r r r~ $ s> �� �� �z"Deserializer._handle_fk_field_nodec s� |j j��j� t� d�r*� ��fdd�}n �fdd�}g }z"|�d�D ]}|�||�� qFW nR ty� } z:t|t�r��j r�t j W Y d}~S t � ||��W Y d}~nd}~0 0 |S dS )z> Handle a <field> node for a ManyToManyField. r� c sJ | � d�}|r2dd� |D �}� ��j�j|� j}n�jj�| �d��}|S )NrC c S s g | ]}t |��� �qS r ro rt r r r rr Y rn zLDeserializer._handle_m2m_field_node.<locals>.m2m_convert.<locals>.<listcomp>r& )rs r� ra r� r& r" rw rx )�nr� r� r- �Zdefault_managerr$ r r r � m2m_convertU s ��z8Deserializer._handle_m2m_field_node.<locals>.m2m_convertc s � j j�| �d��S rO )r" r& rw rx )r� )r$ r r r� f s r' N) rF r$ r� r( rs �append� Exceptionr{ r rZ r r} ZM2MDeserializationError)r ri r1 r� �values�c�er r� r r| M s "z#Deserializer._handle_m2m_field_nodec C s^ |� |�}|s"t�d|j|f ��z t�|�W S ttfyX t�d|j|f ��Y n0 dS )ze Look up a model from a <object model=...> or a <field rel=... to=...> node. z0<%s> node is missing the required '%s' attributez,<%s> node has invalid model identifier: '%s'N)rx r ry re r Z get_model� LookupError� TypeError)r ri �attrZmodel_identifierr r r rv u s �� ��z!Deserializer._get_model_from_node) r? rT rU rV r r\ r_ rj rf r~ r| rv � __classcell__r r rc r rW � s �M)(rW c C s\ g }| j D ]F}|j|jks&|j|jkr4|�|j� q |j|jkr |�t|�� q q d� |�S )z3Get all the inner text of a DOM node (recursively).� ) Z childNodesZnodeTypeZ TEXT_NODEZCDATA_SECTION_NODEr� r� Z ELEMENT_NODE�extendrp �join)ri Z inner_text�childr r r rp � s � � rp c sH e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd d � Zd d � Z � Z S )rd zi An expat parser hardened against XML bomb attacks. Forbid DTDs, external entity references c s2 t � j|i |�� | �tjd� | �tjd� d S )NF)r[ r\ Z setFeaturer Zfeature_external_gesZfeature_external_pes)r �args�kwargsrc r r r\ � s zDefusedExpatParser.__init__c C s t |||��d S rK )� DTDForbidden)r r2 �sysid�pubidZhas_internal_subsetr r r �start_doctype_decl� s z%DefusedExpatParser.start_doctype_declc C s t ||||||��d S rK ��EntitiesForbidden)r r2 �is_parameter_entityrA r r� r� � notation_namer r r � entity_decl� s z DefusedExpatParser.entity_declc C s t |d ||||��d S rK r� )r r2 r r� r� r� r r r �unparsed_entity_decl� s z'DefusedExpatParser.unparsed_entity_declc C s t ||||��d S rK )�ExternalReferenceForbidden�r �contextr r� r� r r r �external_entity_ref_handler� s z.DefusedExpatParser.external_entity_ref_handlerc C s4 t �| � | j}| j|_| j|_| j|_| j |_ d S rK ) � _ExpatParser�reset�_parserr� �StartDoctypeDeclHandlerr� �EntityDeclHandlerr� �UnparsedEntityDeclHandlerr� �ExternalEntityRefHandler)r �parserr r r r� � s zDefusedExpatParser.reset) r? rT rU rV r\ r� r� r� r� r� r� r r rc r rd � s rd c @ s e Zd ZdZdd� ZdS )�DefusedXmlExceptionzBase exception.c C s t | �S rK )r* r r r r �__repr__� s z DefusedXmlException.__repr__N)r? rT rU rV r� r r r r r� � s r� c s( e Zd ZdZ� fdd�Zdd� Z� ZS )r� z&Document type definition is forbidden.c s t � �� || _|| _|| _d S rK )r[ r\ r2 r� r� )r r2 r� r� rc r r r\ � s zDTDForbidden.__init__c C s d}|� | j| j| j�S )Nz7DTDForbidden(name='{}', system_id={!r}, public_id={!r})��formatr2 r� r� �r Ztplr r r �__str__� s zDTDForbidden.__str__�r? rT rU rV r\ r� r� r r rc r r� � s r� c s( e Zd ZdZ� fdd�Zdd� Z� ZS )r� z Entity definition is forbidden.c s2 t � �� || _|| _|| _|| _|| _|| _d S rK )r[ r\ r2 rA r r� r� r� )r r2 rA r r� r� r� rc r r r\ � s zEntitiesForbidden.__init__c C s d}|� | j| j| j�S )Nz<EntitiesForbidden(name='{}', system_id={!r}, public_id={!r})r� r� r r r r� � s zEntitiesForbidden.__str__r� r r rc r r� � s r� c s( e Zd ZdZ� fdd�Zdd� Z� ZS )r� z-Resolving an external reference is forbidden.c s& t � �� || _|| _|| _|| _d S rK )r[ r\ r� r r� r� r� rc r r r\ � s z#ExternalReferenceForbidden.__init__c C s d}|� | j| j�S )Nz8ExternalReferenceForbidden(system_id='{}', public_id={}))r� r� r� r� r r r r� � s z"ExternalReferenceForbidden.__str__r� r r rc r r� � s r� ) rV r9 Zxml.domr Zxml.saxr Zxml.sax.expatreaderr r� Z django.appsr Z django.confr Zdjango.core.exceptionsr Zdjango.core.serializersr Z django.dbr r Zdjango.utils.xmlutilsr r r rW rp rd r= r� r� r� r� r r r r �<module> s( + O$