o
    /h/                     @   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$   ]/var/www/html/env_mimamsha/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rd}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| qt|}|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rt|dkrd&|d"d#| d"  }nd'||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rd1d1g}|	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 }qq-t|| g }| D ],\}}|d.  d77  < |d D ]}|tvr|| d8  d7  < q||g|  qd}	|	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|   |rt|}|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   st  











	u







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>   
 G