
    h                     h    S SK Jr  SSKJr  S SKrS SKJr  S SKrS SK	r	SSK
Jr  S r " S S	\5      rg)
    )ArgumentParser   )BaseInsightFaceCLICommandN   )downloadc                 X    [        U R                  U R                  U R                  5      $ N)ModelDownloadCommandmodelrootforce)argss    ]/var/www/fran/franai/venv/lib/python3.13/site-packages/insightface/commands/model_download.pymodel_download_command_factoryr      s    

DIItzzBB    c                   H    \ rS rSr\S\4S j5       rS\S\S\4S jr	S r
S	rg
)r
      parserc                     U R                  S5      nUR                  S[        SSS9  UR                  SSSS	9  UR                  S
[        SS9  UR                  [        S9  g )Nzmodel.downloadz--rootz~/.insightfacez$Path to location to store the models)typedefaulthelpz--force
store_truez:Force the model to be download even if already in root-dir)actionr   r   zName of the model to download)r   r   )func)
add_parseradd_argumentstrset_defaultsr   )r   download_parsers     r   register_subcommand(ModelDownloadCommand.register_subcommand   sx     ++,<=$$3(8?e 	% 	
 	$$l1m 	% 	
 	$$W3=\$]$$*H$Ir   r   r   r   c                 (    Xl         X l        X0l        g r	   )_model_root_force)selfr   r   r   s       r   __init__ModelDownloadCommand.__init__   s    
r   c                 X    [        SU R                  U R                  U R                  S9  g )Nmodels)r   r   )r   r$   r&   r%   )r'   s    r   runModelDownloadCommand.run"   s    4;;dkk

Kr   )r&   r$   r%   N)__name__
__module____qualname____firstlineno__staticmethodr   r!   r   boolr(   r,   __static_attributes__ r   r   r
   r
      s>    	JN 	J 	Jc  T 
Lr   r
   )argparser    r   osos.pathpathospzipfileglobutilsr   r   r
   r5   r   r   <module>r?      s/    # ' 	    CL4 Lr   