U
    a±dÁ'  ã                   @   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 ddddg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r¨eejd ƒ nedejd  d ƒ 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ÚparseStringZUserÚGroupZ
PermissionÚMessageZ	TextFieldZ	DateFieldZ	CharFieldZIntegerFieldZ
FloatFieldZ	AutoFieldZBooleanFieldZDateTimeFieldZ	TimeField)ÚtextÚdateZvarcharÚintÚfloatÚserialÚbooleanÚnumericÚ	timestampZ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)ÚfksZid_ÚiÚ_© r   úF/tmp/pip-unpacked-wheel-portabvm/django_extensions/utils/dia2django.pyÚ
find_index+   s    
r   c                 C   sh   | D ]^}t ||d ƒ}t ||d ƒ}|| d  d|¡|| d< ||| d kr|| 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)Zrelsr   ÚjZ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r¬|	 d
¡}
|	j	D ]&}|j
tjkr€| ¡ r€| d¡dkrÞ| d¡d j	d jdd… }d| }g |
|dg||< | d¡dkr€| d¡D ]®}| d¡dkrö| d¡D ]\}| d¡dkrL| d¡d j	d jdd… }q| d¡dkr|| d¡d j	d jdd… }nô| d¡dkrº| d¡d j	d jdd… }|dkrpd}n¶| d¡dkr| d¡d  d¡dkr| dd¡ ¡  d¡r*| dd¡dd… }t |¡dkr*t |¡ | dd¡ ¡  d¡r| dd¡dd… }t |¡dkrt |¡ qt |¡}| dd¡ d¡r | dd¡dd… }||kr¾| |d ¡}n4|| d  |¡dkrò|tkrò|| d  |¡ d!| }t|ƒdkrx| d"d#| d" ¡}nX| d$¡dkrx| d%¡dkrht|ƒdkrhd&| d"d#| d" ¡ }nd'||f }n | dd¡ d(¡r| dd¡dd… }||kr¸| |d ¡}n4|| d  |¡dkrì|tkrì|| d  |¡ d!| }t|ƒdkrx| d"d#| d" ¡}n`|d krDd!t| ¡  ¡   d) | d" }n4d*| d¡ d" }t|ƒdkrx| d"d+| d, ¡}|d
krŠ|d-ksö|| d.  d/||f 7  < qöq€qZ|	 d¡d0kr d1d1g}|	 d2¡}|D ].}t| d3¡ƒrÒ| d3¡|t| d4¡ƒ< qÒt|ƒ d1|kr~| |¡ qZ|	 d¡d5krZ|	 d¡}|D ]@}t|j	d jdd… ƒr<|td6|j	d jdd…  ƒ7 }q<qZt||ƒ g }| ¡ D ]X\}}|d.  d77  < |d D ]$}|tkr¶|| d8  d7  < q¶| |g| ¡ q–d}	|	t|ƒk rè|	}|	d }|t|ƒk rF||	 d || d kr:|}|d7 }q||	krZ|	d7 }	nt||	 d || d kr²|| d ||	 d kr²td9||	 d || d f ƒ qè||	 }|| ||	< |||< |	t|ƒd krôqèqô|  ¡  |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:compositeZumlattributezdia:attributeÚvaluez##Z
visibilityzdia:enumÚvalÚ2ú zmanytomanyfield(é   zforeignkey(é   zManyToManyField(z'self'zmodels.ú)ú,Z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   ZgetElementsByTagNameÚstrÚgetAttributeÚ
childNodesZnodeTyper   ZELEMENT_NODEZhasAttributesÚdatar   ÚlowerÚ
startswithÚdependclassesÚcountr   Úv2cÚsearchÚlenÚfindÚtsdÚstripÚgroupr   Úprintr   r   Úreverse)ZarchivoZ
models_txtÚfr<   ZppalZdatosZclasesZheritZimportsr   Zmyidr   ZactclasZmynameZllÚkZncZtcr(   ZnewcZvarchZmyforZmyconsÚaZorderedZfkÚmarkr   r   r   Ú
dia2djangoF   sè    

" 
*






""


&



4 rN   Ú__main__r   r   z Use:
 
   z diagram.dia

)Ú__doc__r4   r6   ÚreÚsysZxml.dom.minidomr   r   r?   rE   ÚcompilerA   r   r   rN   Ú__name__rC   ÚargvrH   r   r   r   r   Ú<module>   s:   ó
 