
    xhQ                         S SK Jr  S SKJr  S SKJr  S SKJr  S SKJ	r	   " S S\R                  5      r " S S	\	5      rg
)    )StrEnum)Union)settings)models)PostgresAddDefaultPartitionc                   6    \ rS rSr\S\\\4   4S j5       rSr	g)FromStringIntegerChoices   stringc                 >    U  H  nUR                   U:X  d  M  Us  $    g N)label)clsr   statuss      /code/glitchtip/model_utils.pyfrom_string$FromStringIntegerChoices.from_string   s    F||v%      N)
__name__
__module____qualname____firstlineno__classmethodr   strr   r   __static_attributes__r   r   r   r	   r	      s$    sG|!4  r   r	   c                   R   ^  \ rS rSrSrS\4S jrU 4S jrU 4S jrU 4S jr	Sr
U =r$ )	TestDefaultPartition   z.Create default partition only on test databasereturnc                 P    [         R                  =(       d    [         R                  $ r   )r   TESTINGENABLE_TEST_API)selfs    r   is_testing_database(TestDefaultPartition.is_testing_database   s    ;8#;#;;r   c                 R   > U R                  5       (       a  [        TU ]  " U0 UD6  g g r   )r%   superstate_forwardsr$   argskwargs	__class__s      r   r)   #TestDefaultPartition.state_forwards   s)    ##%%G"D3F3 &r   c                 R   > U R                  5       (       a  [        TU ]  " U0 UD6  g g r   )r%   r(   database_forwardsr*   s      r   r0   &TestDefaultPartition.database_forwards   s)    ##%%G%t6v6 &r   c                 R   > U R                  5       (       a  [        TU ]  " U0 UD6  g g r   )r%   r(   database_backwardsr*   s      r   r3   'TestDefaultPartition.database_backwards!   s)    ##%%G&77 &r   r   )r   r   r   r   __doc__boolr%   r)   r0   r3   r   __classcell__)r-   s   @r   r   r      s&    8<T <478 8r   r   N)enumr   typingr   django.confr   	django.dbr   Bpsql_partition.backend.migrations.operations.add_default_partitionr   IntegerChoicesr	   r   r   r   r   <module>r>      s5        
v44 86 8r   