U
    àåfŠ<  ã                   @   sÐ   d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dl	m
Z
 d dlZd dlmZ d dlmZmZ d dlmZmZmZ d dlmZmZ d dlmZ d d	lmZ d d
lmZ G dd„ deƒZdS )é    N)Úimport_module)Úbuild_opener)Úsettings)ÚBaseCommandÚCommandError)Úfind_formattersÚhandle_extensionsÚrun_formatters)ÚContextÚEngine)Úarchive)Úparse_header_parameters)Úget_docs_versionc                   @   sv   e Zd ZdZg ZdddgZdZdd„ Zdd	d
„Zdd„ Z	ddd„Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ ZdS )ÚTemplateCommandaÛ  
    Copy either a Django application layout template or a Django project
    layout template into the specified directory.

    :param style: A color style object (see django.core.management.color).
    :param app_or_project: The string 'app' or 'project'.
    :param name: The name of the application or project.
    :param directory: The directory to which the template should be copied.
    :param options: The additional variables passed to project or app templates
    ÚhttpÚhttpsÚftp))z.py-tplz.pyc              	   C   sx   |j ddd |j dddd |j dd	d |j d
ddddgdd |j ddddg dd |j dddtjdddd d S )NÚnamez#Name of the application or project.)ÚhelpÚ	directoryú?zOptional destination directory)Únargsr   z
--templatez*The path or URL to load the template from.z--extensionz-eÚ
extensionsÚappendÚpyztThe file extension(s) to render (default: "py"). Separate multiple extensions with commas, or use -e multiple times.)ÚdestÚactionÚdefaultr   z--namez-nÚfilesz_The file name(s) to render. Separate multiple file names with commas, or use -n multiple times.z	--excludez-xÚ zbThe directory name(s) to exclude, in addition to .git and __pycache__. Can be used multiple times.)r   r   r   Úconstr   )Úadd_argumentÚargparseÚSUPPRESS)ÚselfÚparser© r&   úD/tmp/pip-unpacked-wheel-_jot26k8/django/core/management/templates.pyÚadd_arguments.   sF      ÿ ÿú
ú	øzTemplateCommand.add_argumentsNc           '      K   sÆ  || _ |dkrdnd| _g | _|d | _|  |¡ |d kr¦tj t ¡ |¡}zt 	|¡ W qî t
k
rz   td| ƒ‚Y qî tk
r¢ } zt|ƒ‚W 5 d }~X Y qîX nHtj tj |¡¡}|dkrÖ|  tj |¡d¡ tj |¡sîtd| ƒ‚tƒ }tt|d ƒƒ}g }	d	d
g}
|d D ] }|	 tdd„ | d¡ƒ¡ q| d¡ }rd|D ]}|
 | ¡ ¡ qN| jdkr¤| j d|d |¡f ¡ | j d|d |	¡f ¡ d| }d| }d| }d| }d dd„ | ¡ D ƒ¡}t|||||||dtƒ dtj i–dd}t!j"st! #¡  t $¡  |  %|d |¡}t&|ƒd  }t '|¡D ]\}}}||d … }| (||¡}|rˆtj ||¡}tj	|d!d" |d d … D ]F}d|krÄ| )d#¡s¸|d
krØ| *|¡ n||
kr”| *|¡ q”|D ]z}| +d$¡röqàtj ||¡}tj ||| (||¡¡}| j,D ]2\} }!| +| ¡r"|d t&| ƒ … |! } qVq"tj |¡rxtd%|| j|f ƒ‚| +|¡sŽ||	krêt-|d&d'}"|" .¡ }#W 5 Q R X t/ƒ  0|#¡}$|$ 1|¡}#t-|d(d&d'}%|% |#¡ W 5 Q R X nt2 3||¡ | jdkr| j d)| ¡ z|  4||¡ |  5|¡ W n, tk
rX   | j6 d*| | j7j8¡ Y nX qàqB| jr´| jdkr‚| j d+¡ | jD ]*}&tj 9|&¡r¦t *|&¡ n
t2 :|&¡ qˆt;|gf|Ž d S ),NÚappÚanÚaÚ	verbosityz'%s' already existsr   zBDestination directory '%s' does not exist, please create it first.r   z.gitÚ__pycache__r   c                 S   s   |   ¡ S )N)Ústrip)Úxr&   r&   r'   Ú<lambda>y   ó    z(TemplateCommand.handle.<locals>.<lambda>ú,Úexcludeé   z/Rendering %s template files with extensions: %sz, z.Rendering %s template files with filenames: %sz%s_namez%s_templatez%s_directoryzcamel_case_%s_namer   c                 s   s   | ]}|d kr|V  qdS )Ú_Nr&   )Ú.0r/   r&   r&   r'   Ú	<genexpr>Š   s      z)TemplateCommand.handle.<locals>.<genexpr>Zdocs_versionZdjango_versionF)Z
autoescapeÚtemplateé   T)Úexist_okÚ.)z.pyoz.pycz	.py.classz_%s already exists. Overlaying %s %s into an existing directory won't replace conflicting files.zutf-8)ÚencodingÚwzCreating %szkNotice: Couldn't set permission bits on %s. You're probably using an uncommon filesystem setup. No problem.zCleaning up temporary files.)<Úapp_or_projectÚa_or_anÚpaths_to_remover,   Úvalidate_nameÚosÚpathÚjoinÚgetcwdÚmakedirsÚFileExistsErrorr   ÚOSErrorÚabspathÚ
expanduserÚbasenameÚexistsr   Útupler   ÚextendÚmapÚsplitÚgetr   r.   ÚstdoutÚwriteÚtitler
   r   ÚdjangoÚ__version__r   Z
configuredÚ	configureÚsetupÚhandle_templateÚlenÚwalkÚreplaceÚ
startswithÚremoveÚendswithÚrewrite_template_suffixesÚopenÚreadr   Zfrom_stringÚrenderÚshutilÚcopyfileÚapply_umaskÚmake_writeableÚstderrÚstyleZNOTICEÚisfileÚrmtreer	   )'r$   r>   r   ÚtargetÚoptionsZtop_dirÚeZformatter_pathsr   Zextra_filesZexcluded_directoriesÚfiler3   r   Ú	base_nameZbase_subdirZbase_directoryZcamel_case_nameZcamel_case_valueÚcontextZtemplate_dirZprefix_lengthÚrootÚdirsr   Ú	path_restZrelative_dirÚ
target_dirÚdirnameÚfilenameÚold_pathÚnew_pathÚ
old_suffixÚ
new_suffixZtemplate_fileÚcontentr8   Znew_fileZpath_to_remover&   r&   r'   ÚhandleV   s   

ÿÿÿÿÿÿ     ú÷


  
ÿ
ýþÿ
þü
zTemplateCommand.handlec                 C   sª   |dkrt j tjd d|¡S | d¡r4|dd… }t j |¡}t j |¡}t j |¡r\|S |  	|¡rr|  
|¡}nt j |¡}t j |¡r”|  |¡S td| j|f ƒ‚dS )z«
        Determine where the app or project templates are.
        Use django.__path__[0] as the default because the Django install
        directory isn't known.
        Nr   Úconfzfile://é   zcouldn't handle %s template %s.)rB   rC   rD   rU   Ú__path__r]   rJ   ÚnormpathÚisdirÚis_urlÚdownloadrI   rL   Úextractr   r>   )r$   r8   ÚsubdirZexpanded_templateZabsolute_pathr&   r&   r'   rY   ê   s     


ÿzTemplateCommand.handle_templater   c                 C   s|   |d krt dj| j| jdƒ‚| ¡ s<t dj|| j|dƒ‚zt|ƒ W n tk
r\   Y nX t dj|| j| j|dƒ‚d S )Nz you must provide {an} {app} name)r*   r)   zX'{name}' is not a valid {app} {type}. Please make sure the {type} is a valid identifier.)r   r)   Útypez'{name}' conflicts with the name of an existing Python module and cannot be used as {an} {app} {type}. Please try another {type}.)r   r*   r)   r‡   )r   Úformatr?   r>   Úisidentifierr   ÚImportError)r$   r   Zname_or_dirr&   r&   r'   rA     s6    þÿüÿ	úÿzTemplateCommand.validate_namec              
   C   sš  dd„ }d| j  }tj|dd}| j |¡ ||ƒ\}}| jdkrR| j d| ¡ tj	 
||¡}tƒ }dd	tj› fg|_zF| |¡2}	t|d
ƒ}
|	 ¡ }|
 |	 ¡ ¡ W 5 Q R X W 5 Q R X W n4 tk
rô } ztd|||f ƒ‚W 5 d}~X Y nX | d¡d }|d }|r0t|ƒ\}}| d¡p,|}n|}|  |¡d }|d }|sn|rnt |¡}|rn||7 }||kr–tj	 
||¡}t ||¡ |S |S )zB
        Download the given URL and return the file name.
        c                 S   s8   |   d¡}| d¡d }|  d¡r,|d }n| }||fS )Nú/éÿÿÿÿ)ÚrstriprP   r_   )ÚurlÚtmprw   Údisplay_urlr&   r&   r'   Úcleanup_url-  s    


z-TemplateCommand.download.<locals>.cleanup_urlúdjango_%s_template_Z	_download©ÚprefixÚsuffixr4   zDownloading %sz
User-AgentzDjango/Úwbz"couldn't download URL %s to %s: %sNr‹   rŒ   zcontent-dispositionrw   r9   zcontent-type)r>   ÚtempfileÚmkdtempr@   r   r,   rR   rS   rB   rC   rD   r   rU   rV   Ú
addheadersra   Úinforb   rH   r   rP   r   rQ   ÚsplitextÚ	mimetypesÚguess_extensionrd   Úmove)r$   rŽ   r‘   r”   Útempdirrw   r   Úthe_pathÚopenerÚsourcerl   Úheadersrn   Z	used_nameÚcontent_dispositionr5   ÚparamsZguessed_filenameÚextÚcontent_typeZguessed_pathr&   r&   r'   r„   (  sF    	

&ÿ

zTemplateCommand.downloadc                 C   s@   t  |¡\}}| ¡  d¡r8|dd… | }|dd… }||fS )z@
        Like os.path.splitext, but takes off .tar, too
        z.taréüÿÿÿN)Ú	posixpathr›   Úlowerr_   )r$   r    Úbaser¦   r&   r&   r'   r›   f  s
    zTemplateCommand.splitextc              
   C   s   d| j  }tj|dd}| j |¡ | jdkr>| j d| ¡ zt 	||¡ |W S  tj
tfk
rŠ } ztd|||f ƒ‚W 5 d}~X Y nX dS )zŠ
        Extract the given file to a temporary directory and return
        the path of the directory with the extracted content.
        r’   Z_extractr“   r4   zExtracting %sz"couldn't extract file %s to %s: %sN)r>   r—   r˜   r@   r   r,   rR   rS   r   r…   ZArchiveExceptionrH   r   )r$   rw   r”   rŸ   rn   r&   r&   r'   r…   p  s    

ÿzTemplateCommand.extractc                 C   s*   d|krdS |  dd¡d  ¡ }|| jkS )z)Return True if the name looks like a URL.ú:Fr9   r   )rP   rª   Úurl_schemes)r$   r8   Úschemer&   r&   r'   rƒ   ‚  s    zTemplateCommand.is_urlc                 C   s<   t  d¡}t  |¡ t t  |¡j¡}t  ||| @ ¡ d S )Nr   )rB   ÚumaskÚstatÚS_IMODEÚst_modeÚchmod)r$   rx   ry   Úcurrent_umaskZcurrent_moder&   r&   r'   rf   ‰  s    

zTemplateCommand.apply_umaskc                 C   s:   t  |t j¡s6t  |¡}t |j¡tjB }t  ||¡ dS )zb
        Make sure that the file is writeable.
        Useful if our source is read-only.
        N)rB   ÚaccessÚW_OKr°   r±   r²   ÚS_IWUSRr³   )r$   rw   ÚstZnew_permissionsr&   r&   r'   rg     s    
zTemplateCommand.make_writeable)N)r   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__Zrequires_system_checksr­   r`   r(   r}   rY   rA   r„   r›   r…   rƒ   rf   rg   r&   r&   r&   r'   r      s   
(
 
#>
r   )r"   rœ   rB   r©   rd   r°   r—   Ú	importlibr   Úurllib.requestr   rU   Zdjango.confr   Zdjango.core.management.baser   r   Zdjango.core.management.utilsr   r   r	   Zdjango.templater
   r   Zdjango.utilsr   Zdjango.utils.httpr   Zdjango.utils.versionr   r   r&   r&   r&   r'   Ú<module>   s"   