B
    ©0H[42  ã               @   s\   d dl Z d dlZd dlmZ d dlmZmZ d dlmZm	Z	 d dl
mZ G dd„ deƒZdS )é    N)ÚOrderedDict)ÚBaseCommandÚCommandError)ÚDEFAULT_DB_ALIASÚconnections)Ú
LOOKUP_SEPc               @   sL   e Zd ZdZdZdZdZdd„ Zdd„ Zd	d
„ Z	dd„ Z
dd„ Zdd„ ZdS )ÚCommandzXIntrospects the database tables in the given database and outputs a Django model module.F)Útable_name_filterz	django.dbc             C   s,   |j dddtdd� |j dddtdd	� d S )
NÚtableÚstoreÚ*z4Selects what tables or views should be introspected.)ÚactionÚnargsÚtypeÚhelpz
--databaseÚdatabasezMNominates a database to introspect. Defaults to using the "default" database.)r   ÚdestÚdefaultr   )Úadd_argumentÚstrr   )ÚselfÚparser© r   úmC:\Users\HIRONO~1\AppData\Local\Temp\pip-install-6bm3nxem\django\django\core\management\commands\inspectdb.pyÚadd_arguments   s    zCommand.add_argumentsc             K   sJ   y(x"|   |¡D ]}| j d| ¡ qW W n tk
rD   tdƒ‚Y nX d S )Nz%s
zPDatabase inspection isn't supported for the currently selected database backend.)Úhandle_inspectionÚstdoutÚwriteÚNotImplementedErrorr   )r   ÚoptionsÚliner   r   r   Úhandle   s
    zCommand.handlec             #   sÞ  t |d  }| d¡}dd„ }dd„ ‰ | ¡ ��¤}dV  dV  d	V  d
V  dV  dV  dV  d| j V  g }|d p~|j |¡}�xL|D �]B}|d k	r¨t|ƒr¨||ƒs¨qˆyŒy|j ||¡}	W n tk
rÔ   i }	Y nX y|j 	||¡}
W n tk
�r   i }
Y nX |j 
||¡}dd„ |
 ¡ D ƒ}|j ||¡}W n: tk
�rn } zd| V  d| V  wˆW d d }~X Y nX dV  dV  d||ƒ V  | ||ƒ¡ g }i }�x|D �]þ}g }tƒ }|d }||	k}|  |||¡\}}}| |¡ | |¡ | |¡ |||< ||k�rd|d< n||k�r(d|d< |�rr|	| d |k�rDdn||	| d ƒ}||k�rhd| }nd| }n0|  |||¡\}}}| |¡ | |¡ |d7 }|dk�rè|ddik�rè|d k�rÊ�q¨n|d!k�rè|jj�sè| d"¡ |d# �r|d$k�rd%}nd|d&< d|d'< d(|d)|k�r$dnd*|f }| d+¡�rB|d,7 }|�r|| d¡�s\|d-7 }|d- ‡ fd.d/„| ¡ D ƒ¡7 }|d07 }|�rœ|d1d2 |¡ 7 }d3| V  �q¨W x|  ||
|¡D ]}|V  �q¼W qˆW W d Q R X d S )4Nr   r	   c             S   s   t  dd|  ¡ ¡S )Nz[^a-zA-Z0-9]Ú )ÚreÚsubÚtitle)Ú
table_namer   r   r   Útable2model&   s    z.Command.handle_inspection.<locals>.table2modelc             S   s   |   d¡r| dd … S | S )Nzu'é   )Ú
startswith)Úsr   r   r   Ústrip_prefix)   s    z/Command.handle_inspection.<locals>.strip_prefixz0# This is an auto-generated Django model module.z<# You'll have to do the following manually to clean this up:z#   * Rearrange models' orderz>#   * Make sure each model has one field with primary_key=TruezL#   * Make sure each ForeignKey has `on_delete` set to the desired behavior.zh#   * Remove `managed = False` lines if you wish to allow Django to create, modify, and delete the tablezR# Feel free to rename the models, but don't rename db_table values or field names.zfrom %s import modelsr
   c             S   s0   g | ](}|d  rt |d ƒdkr|d d ‘qS )ÚuniqueÚcolumnsr(   r   )Úlen)Ú.0Úcr   r   r   ú
<listcomp>J   s    z-Command.handle_inspection.<locals>.<listcomp>z# Unable to inspect table '%s'z# The error was: %sr"   zclass %s(models.Model):r   TZprimary_keyr,   r(   r   zForeignKey(%szForeignKey('%s'ú(Úidz
AutoField(zIntegerField(z
AutoField?é   zBooleanField(zNullBooleanField(ZblankÚnullz	%s = %s%sÚ.zmodels.zForeignKey(z, models.DO_NOTHINGz, c             3   s&   | ]\}}d |ˆ t |ƒƒf V  qdS )z%s=%sN)Úrepr)r/   ÚkÚv)r+   r   r   ú	<genexpr>�   s   z,Command.handle_inspection.<locals>.<genexpr>ú)z  # ú z    %s)r   ÚgetÚcursorÚ	db_moduleÚintrospectionZtable_namesÚcallableZget_relationsr   Zget_constraintsZget_primary_key_columnÚvaluesZget_table_descriptionÚ	ExceptionÚappendr   Únormalize_col_nameÚupdateÚextendÚget_field_typeÚfeaturesZcan_introspect_autofieldr)   ÚendswithÚjoinÚitemsÚget_meta)r   r   Ú
connectionr	   r'   r>   Zknown_modelsZtables_to_introspectr&   Z	relationsÚconstraintsZprimary_key_columnZunique_columnsZtable_descriptionÚeÚused_column_namesÚcolumn_to_field_nameÚrowZcomment_notesZextra_paramsZcolumn_nameÚis_relationZatt_nameÚparamsZnotesZrel_toÚ
field_typeÚfield_paramsÚfield_notesZ
field_descZ	meta_liner   )r+   r   r   !   s¼    




















zCommand.handle_inspectionc       	      C   s’  i }g }|  ¡ }||kr"| d¡ |rF| d¡r>|dd… }n||d< t dd|¡\}}|dkrj| d	¡ | t¡dkr´x| t¡dkr–| td¡}qzW |  ¡  t¡dkr´| d
¡ | d¡rÐd| }| d¡ | d¡rìd| }| d¡ t	 
|¡�r
|d7 }| d¡ |d  ¡ �r*d| }| d¡ ||k�rpd}x d||f |k�rX|d7 }�q:W d||f }| d¡ ||k�rˆ|�rˆ||d< |||fS )zU
        Modify the column name to make it Python-compatible as a field name
        zField name made lowercase.Ú_idNéýÿÿÿZ	db_columnz\WÚ_r   z.Field renamed to remove unsuitable characters.z>Field renamed because it contained more than one '_' in a row.zfield%sz*Field renamed because it started with '_'.z%sfieldz(Field renamed because it ended with '_'.Z_fieldz4Field renamed because it was a Python reserved word.z	number_%sz:Field renamed because it wasn't a valid Python identifier.z%s_%dr(   z'Field renamed because of name conflict.)ÚlowerrD   rJ   r#   ÚsubnÚfindr   Úreplacer)   ÚkeywordÚ	iskeywordÚisdigit)	r   Zcol_namerQ   rT   rW   rX   Únew_nameZnum_replÚnumr   r   r   rE   ¦   sL    











zCommand.normalize_col_namec             C   s  t ƒ }g }y|j |d |¡}W n" tk
rB   d}| d¡ Y nX t|ƒtkrb|\}}| |¡ |dkr‚|d r‚t|d ƒ|d< |dkrþ|d d	ks¢|d
 d	kræ| d¡ |d d	k	rÀ|d nd|d< |d
 d	k	rÜ|d
 nd
|d< n|d |d< |d
 |d< |||fS )zê
        Given the database connection, the table name, and the cursor row
        description, this routine will return the given field type name, as
        well as any additional keyword parameters and notes for the field.
        r(   Z	TextFieldzThis field type is a guess.Z	CharFieldé   Ú
max_lengthZDecimalFieldé   Né   zamax_digits and decimal_places have been guessed, as this database handles decimal fields as floaté
   Z
max_digitsZdecimal_places)	r   r@   rH   ÚKeyErrorrD   r   ÚtuplerF   Úint)r   rN   r&   rS   rW   rX   rV   Ú
new_paramsr   r   r   rH   Þ   s*    
zCommand.get_field_typec       
         s˜   g }xZ|  ¡ D ]N\}}|d r|d }t|ƒdkrdd ‡ fdd„|D ƒ¡ d }| |¡ qW d	d
dd| g}	|r”dd |¡ d }|	d| g7 }	|	S )zº
        Return a sequence comprising the lines of code necessary
        to construct the inner Meta class for the model corresponding
        to the given database table name.
        r,   r-   r(   r2   z, c             3   s   | ]}d ˆ |  V  qdS )z'%s'Nr   )r/   r0   )rR   r   r   r:     s    z#Command.get_meta.<locals>.<genexpr>r;   r"   z    class Meta:z        managed = Falsez        db_table = '%s'z,)z        unique_together = %s)rL   r.   rK   rD   )
r   r&   rO   rR   Zunique_togetherÚindexrU   r-   ÚtupÚmetar   )rR   r   rM     s     
zCommand.get_metaN)Ú__name__Ú
__module__Ú__qualname__r   Zrequires_system_checksZstealth_optionsr?   r   r!   r   rE   rH   rM   r   r   r   r   r   
   s   
 8&r   )r`   r#   Úcollectionsr   Zdjango.core.management.baser   r   Z	django.dbr   r   Zdjango.db.models.constantsr   r   r   r   r   r   Ú<module>   s   