3
g'                 @   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j
dZdd Zdd Zdd Zedkreejdkre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.   x(| j  D ]\}}| | d |kr
|S q
W 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   `/var/www/tester-filtro-web/env/lib/python3.6/site-packages/django_extensions/utils/dia2django.py
find_index+   s    r   c             C   sl   xf| D ]^}t ||d }t ||d }|| d jd||| d< ||| d kr|| d j| qW 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   sX  d}t j| d}tj|dj }t|}|jdd jdd jd}i }g }td}xL|D ]B}	|	jdd	kr|	jd
}
x |	j	D ]8}|j
tjko|j r|jddkr|jdd j	d jdd: }d| }g |
|dg||< |jddkrx|jdD ]}|jddkrxl|jdD ]\}|jddkr^|jdd j	d jdd; }q&|jddkr|jdd j	d jdd< }n|jddkr|jdd j	d jdd= }|dkrd}n|jddkr&|jdd jddkr&|jddj jdr<|jdddd> }tj|dkr<tj| |jddj jdr&|jdddd? }tj|dkr&tj| q&W tj|}|jddjdr4|jdddd@ }||kr|j|d}n4|| d j|dkr|tkr|| d j| d | }t|dkr|jd!d"| d! }nX|jd#dAkr|jd$dkr|t|dkr|d%|jd!d"| d!  }nd&||f }n |jddjd'r,|jddddB }||kr|j|d}n4|| d j|dkr |tkr || d j| d | }t|dkr|jd!d"| d! }n`|d krXd t|j j   d( | d! }n4d)|jd d! }t|dkr|jd!d*| d+ }|d
ko|d,ks|| d-  d.||f 7  < qW qW q^|	jdd/kr@d0d0g}|	jd1}x6|D ].}t|jd2r|jd2|t|jd3< qW t| d0|kr|j| q^|	jdd4kr^|	jd}xH|D ]@}t|j	d jddC r^|td5|j	d jddD  7 }q^W q^W t|| g }xh|j D ]\\}}|d-  d67  < x0|d D ]$}|tkr|| d7  d7  < qW |j|g|  qW d}	x|	t|k r|	}|	d }x:|t|k rx||	 d || d krl|}|d7 }q@W ||	kr|	d7 }	nr||	 d || d kr|| d ||	 d krtd8||	 d || d f  P ||	 }|| ||	< |||< |	t|d kr$P q$W |j   |r0t|}x"|D ]}	|d9t|	d7  7 }q6W |S )EN 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:connectionZ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
r2   r2   r2   r2   r2   r2   r2   r2   r2   r2   )!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 
rM   __main__r   r   z Use:
 
   z diagram.dia

)__doc__r3   r5   resysZxml.dom.minidomr   r   r>   rD   compiler@   r   r   rM   __name__rB   argvrG   r   r   r   r   <module>   s8   
 