
    	]j                         d dl Z d dlZd dlmZ ej                  j
                  Zd dlmZ d dl	m
Z
 d Zd Zd Zd Zd	 Zy)
    N)LABEL_CONFIG_AND_TASKS)LabelStudio)sdk_logsc                    t        | |j                        }|j                  j                  dt        d         }dddiig}|D ],  }|j
                  j                  |j                  |d          . |j
                  j                  |j                  	      D cg c]  }| }}t        |      d
k(  sJ |d   j                  x}sJ |d   j                  |d   d   k(  sJ |j
                  j                  |ddi       |j
                  j                  |j                  	      D cg c]  }| }}t        |      d
k(  sJ |d   j                  ddik(  sJ |j
                  j                  |       g }|j
                  j                  |j                  	      D ]  }|j                  |        t        |      dk(  sJ y c c}w c c}w )Nbase_urlapi_keyNew Projectlabel_configtitler   datamy_textz	Test taskprojectr   r      r   zUpdated task)idr   r   )r   r	   projectscreater   tasksr   listlenr   updatedeleteappend)django_live_urlbusiness_clientlsp	task_datataskr   task_ids           K/root/env/lib/python3.12/site-packages/label_studio/tests/sdk/test_tasks.pytest_task_CRUDr&      s   	o7N7N	OB
=STb=cdA9k234I 9
4<89 !hhmmADDm9:dT:E:u:??Qx{{"G"#"8==IaL0000HHOOwi%@OA hhmmADDm9:dT:E:u:??8==Y7777HHOOwOEadd+ Tu:?? ; ;s   	G+	Gc                    t        | |j                        }|j                  j                  dt        d         }t        d      D cg c]  }dddt        |      z   ii }}|D ],  }|j                  j                  |j                  |d   	       . |j                  j                  |j                  
      D cg c]  }| }}t        |      dk(  sJ |d d D cg c]  }|j                   }	}|j                  j                  |j                  dd|	d       t        |j                  j                  |j                  
      D cg c]  }| c}      dk(  sJ dd l}
|j                  j                  |j                  ddd|
j                  d|d   j                  gd      ii       |j                  j                  |j                  
      D cg c]  }| }}t        |      dk(  sJ |d   j                  d   dk(  sJ |j                  j                  |j                         d}|j                  j                  |j                  
      D ]  }d} |rJ y c c}w c c}w c c}w c c}w c c}w )Nr   r
   r   r   
   r   r   
Test task r   r      delete_tasksF)allincluded)r   r   selected_itemsr   additional_body_parametersselectedItemsT)r,   excluded)r   r   request_optionsr   zTest task 5r   )r   r	   r   r   r   rangestrr   r   r   r   actionsjsondumpsr   delete_all_tasks)r   r   r    r!   ir"   r#   r   ttasks_ids_to_deleter6   remaining_tasksany_task_founds                r%   test_delete_multi_tasksr>   '   sN   	o7N7N	OB
=STb=cdAGLRyQ!&9lSV&;<=QIQ 9
4<89 !hhmmADDm9:dT:E:u:).r3A14433 JJadd~eatFuvqtt!<==>!CCC JJ(DuQx{{m,T!U+
   )+add(CDtDOD1$$$1""9->>> HH&Nadd+ ~I R ; 4 >  Es   	I%=	I*I/	I4	I9c           	      T   t        | |j                        }|j                  j                  dt        d         }t        d      D cg c]  }dddt        |      z   ii }}|j                  j                  |j                  |	       d }t        |j                  j                  |j                  
            D ]  \  }}|dk(  s|j                  } n ddddddgidgi} |j                  j                  dd|i| |j                  j                  |      }	|	j                  d   dk(  sJ |	j                  dk(  sJ |	j                   d   d   |j"                  j                  k(  sJ |j                  j                  |j                  d      D cg c]  }|j                  s| }
}t%        |
      dk(  sJ |
d   j                  d   dk(  sJ |j                  j                  |j                  d      D cg c]  }| }
}t%        |
      dk(  sJ y c c}w c c}w c c}w )Nr   r
   r   r   r(   r   r   r)   r   requestr      resultlabelchoicesPositive)	from_nameto_nametypevaluer   r   zTest task 7r   r   user_idr,   r   fields )r   r	   r   r   r   r3   r4   import_tasksr   	enumerater   r   annotationsgetr   total_annotations
updated_byuserr   )r   r   r    r!   r9   r"   r$   r#   annotation_datasingle_taskexported_taskss              r%   test_export_tasksrY   R   s   	o7N7N	OB
=STb=cdAGLRyQ!&9lSV&;<=QIQKKi8GRXX]]144]89 46ggG 	Y	]fisht\uvwO BNN8W88 ((,,',*KI&-777((A---!!!$Y/?3G3G3J3JJJJ')xx}}QTT%}'PetTXTdTddeNe~!###!!!),===')xx}}QTT%}'PQtdQNQ~"$$$3 R( f Rs   	HH )H ?	H%c                 4   |j                  t        j                         t        | |j                        }|j
                  j                  dt        d         }|j
                  j                  |j                  t        d          |j                  j                  |j                  d      D cg c]  }| }}t        |      d	k(  sJ t        |d
   j                        d	k(  sJ t        |d
   j                        d	k(  sJ t        |      rJ y c c}w )Nr   r
   r   r   tasks_for_importr@   r,   rL   r   r   )	set_levelloggingERRORr   r	   r   r   r   rO   r   r   r   r   rQ   predictionsr   r   r   caplogr    r!   r#   r   s          r%   1test_upload_and_list_tasks_does_not_log_to_stderrrb   r   s    
W]]#	o7N7N	OB
=STb=cdAKK.DEW.XY hhmmADDmGHdTHEHu:??uQx##$)))uQx##$))) Is   2	Dc                 J   |j                  t        j                         t        | |j                        }|j
                  j                  dt        d         }|j                  j                  |j                        D cg c]  }| }}|rJ t        |      rJ y c c}w )Nr   r
   r   r   r   )r\   r]   r^   r   r	   r   r   r   r   r   r   r   r`   s          r%   +test_get_empty_tasks_does_not_log_to_stderrrd      s    
W]]#	o7N7N	OB
=STb=cdA hhmmADDm9:dT:E:9 ;s   	B )r]   pytestlabel_studio.tests.sdk.commonr   mark	django_db
pytestmarklabel_studio_sdk.clientr   label_studio.tests.sdk.utilsr   r&   r>   rY   rb   rd   rN       r%   <module>rm      s>      @[[""
 / 14(V%@ 	 rl   