o
    ~#iÁ'  ã                   @   sÄ   d Z ddlZddlZddlZddlZddlmZmZ g d¢Zddddd	d
dd	ddddddœZ	e 
d¡Zdd„ Zdd„ Zdd„ Zedkr`eejƒdkrSeejd ƒ dS edejd  d ƒ dS dS )u„  
Author Igor TÃ¡mara igor@tamarapatino.org
Use this little program as you wish, if you
include it in your work, let others know you
are using it preserving this note, you have
the right to make derivative works, Use it
at your own risk.
Tested to work on(etch testing 13-08-2007):
  Python 2.4.4 (#2, Jul 17 2007, 11:56:54)
  [GCC 4.1.3 20070629 (prerelease) (Debian 4.1.2-13)] on linux2
é    N)ÚNodeÚparseString)ÚUserÚGroupÚ
PermissionÚMessageÚ	TextFieldÚ	DateFieldÚ	CharFieldÚIntegerFieldÚ
FloatFieldÚ	AutoFieldÚBooleanFieldÚDateTimeFieldÚ	TimeField)ÚtextÚdateÚvarcharÚintÚfloatÚserialÚbooleanÚnumericÚ	timestampÚbigintÚdatetimeÚtimeÚboolzvarchar\((\d+)\)c                 C   s.   |   ¡ D ]\}}| | d |kr|  S qdS )zÙ
    Look for the id on fks, fks is an array of arrays, each array has on [1]
    the id of the class in a dia diagram.  When not present returns None, else
    it returns the position of the class with id on fks
    é   N)Úitems)ÚfksÚid_ÚiÚ_© r$   úi/var/www/GraceOrthoHospital/venv_grace/lib/python3.10/site-packages/django_extensions/utils/dia2django.pyÚ
find_index+   s
   ÿr&   c                 C   sh   | D ]/}t ||d ƒ}t ||d ƒ}|| d  d|¡|| d< ||| d vr1|| d  |¡ qdS )a  
    Get a list of relations, between parents and sons and a dict of
    clases named in dia, and modifies the fks to add the parent as fk to get
    order on the output of classes and replaces the base class of the son, to
    put the class parent name.
    r   r   é   zmodels.ModelN)r&   ÚreplaceÚappend)Úrelsr    ÚjÚsonÚparentr$   r$   r%   Úaddparentstofks7   s   €ûr.   c              	   C   s,  d}t  | d¡}tj|d ¡ }t|ƒ}| d¡d  d¡d  d¡}i }g }tdƒ}|D ]Ž}	|	 d¡d	krR|	 d
¡}
|	j	D ]}|j
tjkrP| ¡ rP| d¡dkrq| d¡d j	d jdd… }d| }g |
|dg||< | d¡dkrP| d¡D ]Ð}| d¡dkrO| d¡D ]¨}| d¡dkr§| d¡d j	d jdd… }qŽ| d¡dkr¾| d¡d j	d jdd… }qŽ| d¡dkrÛ| d¡d j	d jdd… }|dkrÚd}qŽ| d¡dkr6| d¡d  d¡dkr6| dd¡ ¡  d¡r| dd¡dd… }t |¡dkrt |¡ | dd¡ ¡  d¡r6| dd¡dd… }t |¡dkr6t |¡ qŽt |¡}| dd¡ d¡rŒ| dd¡dd… }||kr\| |d ¡}n|| d  |¡dkrv|tvrv|| d  |¡ d!| }t|ƒdkr‹| d"d#| d" ¡}n«| d$¡dkr·| d%¡dkr°t|ƒdkr°d&| d"d#| d" ¡ }n‡d'||f }n€| dd¡ d(¡r| dd¡dd… }||kr×| |d ¡}n|| d  |¡dkrñ|tvrñ|| d  |¡ d!| }t|ƒdkr| d"d#| d" ¡}n0|d u rd!t| ¡  ¡   d) | d" }nd*| d¡ d" }t|ƒdkr7| d"d+| d, ¡}|d
krA|d-ksO|| d.  d/||f 7  < q~q@q-|	 d¡d0krŒd1d1g}|	 d2¡}|D ]}t| d3¡ƒr{| d3¡|t| d4¡ƒ< qet|ƒ d1|vr‹| |¡ q-|	 d¡d5kr¼|	 d¡}|D ] }t|j	d jdd… ƒrº|td6|j	d jdd…  ƒ7 }q›q-t||ƒ g }| ¡ D ],\}}|d.  d77  < |d D ]}|tvré|| d8  d7  < qØ| |g| ¡ qÈd}	|	t|ƒk ry|	}|	d }|t|ƒk r%||	 d || d v r|}|d7 }|t|ƒk s||	kr/|	d7 }	n9||	 d || d v rZ|| d ||	 d v rZtd9||	 d || d f ƒ n||	 }|| ||	< |||< |	t|ƒd krrn|	t|ƒk sþ|  ¡  |r„t|ƒ}|D ]}	|d:t|	d8 ƒ 7 }q†|S );NÚ Úrb)Úfileobjzdia:diagramr   z	dia:layerz
dia:objectÚtypezUML - ClassÚidÚnamez
dia:stringr   éÿÿÿÿz
class %s(models.Model) :
Ú
attributeszdia:compositeÚumlattributezdia:attributeÚvaluez##Ú
visibilityzdia:enumÚvalÚ2ú zmanytomanyfield(é   zforeignkey(é   zManyToManyField(z'self'zmodels.ú)ú,ÚFieldz()z	models.%szmodels.%s(%s)zForeignKey(ú(zmodels.CharField(max_length=z, z )zAutoField()r'   z    %s = %s
zUML - GeneralizationÚAzdia:connectionÚtoÚhandlezUML - SmallPackagezfrom %s.models import *z+
    def __str__(self):
        return u""
é   z:Not able to resolve circular ForeignKeys between %s and %sz%s
)!ÚcodecsÚopenÚgzipÚGzipFileÚreadr   ÚgetElementsByTagNameÚstrÚgetAttributeÚ
childNodesÚnodeTyper   ÚELEMENT_NODEÚhasAttributesÚdatar(   ÚlowerÚ
startswithÚdependclassesÚcountr)   Úv2cÚsearchÚlenÚfindÚtsdÚstripÚgroupr   Úprintr.   r   Úreverse)ÚarchivoÚ
models_txtÚfrS   ÚppalÚdatosÚclasesÚheritÚimportsr"   Úmyidr+   ÚactclasÚmynameÚllÚkÚncÚtcr:   ÚnewcÚvarchÚmyforÚmyconsÚaÚorderedÚfkÚmarkr$   r$   r%   Ú
dia2djangoF   s   

  €*

€


€

€
"€¹H
€

€
 €

€ý

4 írx   Ú__main__r'   r   z Use:
 
   z diagram.dia

)Ú__doc__rG   rI   ÚreÚsysÚxml.dom.minidomr   r   rV   r\   ÚcompilerX   r&   r.   rx   Ú__name__rZ   Úargvr_   r$   r$   r$   r%   Ú<module>   s>   ó
 ü