
    h                     ^    S SK r S SKJrJr  S SKJrJr  S SKJr  S SK	J
r
Jr   " S S\5      rg)    N)BaseCommandCommandError)get_commandsload_command_class)
connection)get_tenant_modelget_public_schema_namec                   2   ^  \ rS rSrSrU 4S jrS rSrU =r$ )Command	   z9Wrapper around django commands for use with an all tenantc                 l   > [         TU ]  U5        UR                  SSSSSS9  UR                  SSS	S
9  g )N--no-public?TFzExclude the public schema)nargsconstdefaulthelpcommand_name+z The command name you want to run)r   r   )superadd_argumentsadd_argument)selfparser	__class__s     h/var/www/html/env/lib/python3.13/site-packages/django_tenants/management/commands/all_tenants_command.pyr   Command.add_arguments   s@    f%MD%VqrN#<^_    c                    [        U5      S::  a  gSU;   nUS   /nUR                  U(       a  USS OUSS 5         [        5       US      n[        U[        5      (       a  UnO[        XCS   5      n[        R                  " 5       nUR                  U5      u  px[        U5        [        5       n	U	R                  R                  5       n
U(       a  U
R                  [!        5       S9n
U
 HR  nU R"                  R%                  S	UR&                  -  5        [(        R*                  " U5        UR-                  U5        MT     g! [         a    [	        SUS   -  5      ef = f)
zF
Changes the option_list to use the options from the wrapped command.
   Nr   r         zUnknown command: %r)schema_namezApplying command to: %s)lenextendr   KeyErrorr   
isinstancer   r   argparseArgumentParserparse_known_argsprintr   objectsallexcluder	   stdoutwriter#   r   
set_tenantrun_from_argv)r   argv	no_publiccommand_argsapp_nameklassschema_parserschema_namespaceargstenant_modeltenantstenants               r   r2   Command.run_from_argv   sE   
 t9>!T)	Qy	DHtABx@	H#~l1o6H h,,E&xaAE //1!.!?!?!Md')&&**,oo2H2JoKGFKK7&:L:LLM!!&)% #  	H4|AFGG	Hs   E E )	__name__
__module____qualname____firstlineno__r   r   r2   __static_attributes____classcell__)r   s   @r   r   r   	   s    FD`$& $&r   r   )r(   django.core.management.baser   r   django.core.managementr   r   	django.dbr   django_tenants.utilsr   r	   r   r?   r   r   <module>rJ      s      A C   I.&k .&r   