
    	]j                     n    d dl Z d dlmZ d dlmZ d dlmZ  G d de j                  j                        Z	y)    N)	load_func)settings)Organizationc                       e Zd Z ej                  d      Z ej                   eej                        d      Z
 G d d      Zed        Zej                  d        Zy)OrganizationFactorycompanyN)active_organizationc                       e Zd ZeZy)OrganizationFactory.MetaN)__name__
__module____qualname__r   model     T/root/env/lib/python3.12/site-packages/label_studio/organizations/tests/factories.pyMetar      s    r   r   c                 ,    t        j                  di |S )Nr   )r   create_organization)clsmodel_classargskwargss       r   _createzOrganizationFactory._create   s    //9&99r   c                 ~    |r| j                   sy | | j                   _        | j                   j                  dg       y )Nr	   )update_fields)
created_byr	   save)selfcreate	extractedr   s       r   created_by_active_organizationz2OrganizationFactory.created_by_active_organization   s3    T__.2+,A+BCr   )r   r   r   factoryFakertitle
SubFactoryr   r   USER_FACTORYr   r   classmethodr   post_generationr"   r   r   r   r   r      sl    GMM)$E###Ih.C.C$DZ^_J  : : D Dr   r   )
r#   core.utils.commonr   django.confr   organizations.modelsr   djangoDjangoModelFactoryr   r   r   r   <module>r/      s)     '   -D'..;; Dr   