
    g                     n    d dl Z d dlmZmZmZmZ d dlmZ d dlm	Z	 d dl
mZmZ d dlmZ  G d de      Zy)	    N)DictOptionalUnioncast)TracingGroupLocation)Artifact)ChannelOwnerfrom_nullable_channel)locals_to_paramsc                   <    e Zd Zdededededdf
 fdZ	 	 	 	 	 dded	ed
edededdfdZdd	ededdfdZ	deddfdZ
ddeej                  ef   ddfdZddeej                  ef   ddfdZddeej                  ef   ddfdZddZddededdfdZddZ xZS )TracingparenttypeguidinitializerreturnNc                     t         |   ||||       | j                  j                          d| _        d | _        d| _        d | _        y NF)super__init___channelmark_as_internal_type_include_sources
_stacks_id_is_tracing_traces_dir)selfr   r   r   r   	__class__s        N/var/www/openai/venv/lib/python3.12/site-packages/playwright/_impl/_tracing.pyr   zTracing.__init__   sG     	t[9++-&+)-!&*.    nametitle	snapshotsscreenshotssourcesc                 &  K   t        t                     }t        |      | _        | j                  j                  d|       d {    | j                  j                  d||d       d {   }| j                  |       d {    y 7 E7  7 	w)NtracingStarttracingStartChunk)r"   r!   )r   localsboolr   r   send_start_collecting_stacks)r   r!   r"   r#   r$   r%   params
trace_names           r   startzTracing.start#   s      "&(+ $Wmm  888==--5$!?
 

 ++J777	 	9
 	8s6   ABB&B,B-BBBBBc                    K   t        t                     }| j                  j                  d|       d {   }| j	                  |       d {    y 7 7 w)Nr(   )r   r)   r   r+   r,   )r   r"   r!   r-   r.   s        r   start_chunkzTracing.start_chunk4   sJ     !&(+==--.A6JJ
++J777 K7s!   3AAAAAAr.   c                    K   | j                   s"d| _         | j                  j                  d       | j                  j                  j	                  | j
                  |       d {   | _        y 7 
w)NT)r   _connectionset_is_tracinglocal_utilstracing_startedr   r   )r   r.   s     r   r,   z Tracing._start_collecting_stacks9   s^     #D++D1 $ 0 0 < < L Lj!
 
 
s   A"A1$A/%A1pathc                 B   K   | j                  |       d {    y 7 wN)_do_stop_chunkr   r7   s     r   
stop_chunkzTracing.stop_chunkA   s     !!$'''s   c                    K   | j                  |       d {    | j                  j                  d       d {    y 7 (7 w)NtracingStop)r:   r   r+   r;   s     r   stopzTracing.stopD   s;     !!$'''mm  /// 	(/s   AA "AAAA	file_pathc                   K   | j                          |sj| j                  j                  dddi       d {    | j                  r7| j                  j
                  j                  | j                         d {    y | j                  j                   }|rz| j                  j                  dddi       d {   }| j                  j
                  j                  t        |      |d   | j                  d| j                  d       d {    y | j                  j                  dddi       d {   }t        t        t           t        |j!                  d                  }|sD| j                  r7| j                  j
                  j                  | j                         d {    y |j#                  |       d {    |j%                          d {    | j                  j
                  j                  t        |      g | j                  d	| j                  d       d {    y 7 7 7 |7 ,7 7 7 ~7 h7 w)
NtracingStopChunkmodediscardentrieswrite)zipFilerE   stacksIdrC   includeSourcesarchiveartifactappend)_reset_stack_counterr   r+   r   r3   r5   trace_discarded	is_remotesend_return_as_dictzipstrr   r   r   r   r
   getsave_asdelete)r   r@   is_localresultrK   s        r   r:   zTracing._do_stop_chunkH   s    !!#--$$%7&)9LMMM&&22BB4??SSS''111==<<"VY$7 F ""..22"9~%i0 $#&*&;&;   }}88	
 
 X!&**Z"89
 &&22BB4??SSS y)))oo**..y> OO "&"7"7
 	
 	
Y NS
 T 	*	
s   4IH/AI9H2:?I9H5:AIH8&I3H;4A3I'H>(II IIAI)I*I2I5I8I;I>I IIIc                 b    | j                   r#d| _         | j                  j                  d       y y r   )r   r3   r4   r   s    r   rM   zTracing._reset_stack_counter   s,    $D++E2 r    locationc                 z   K   | j                   j                  dt        t                            d {    y 7 w)NtracingGroup)r   r+   r   r)   )r   r!   rZ   s      r   groupzTracing.group   s(     mm  1A&(1KLLLs   1;9;c                 V   K   | j                   j                  d       d {    y 7 w)NtracingGroupEnd)r   r+   rY   s    r   	group_endzTracing.group_end   s     mm  !2333s   )'))NNNNN)NNr9   )r   N)__name__
__module____qualname__r	   rR   r   r   r*   r/   r1   r,   r   pathlibPathr<   r?   r:   rM   r   r]   r`   __classcell__)r   s   @r   r   r      s;   /"/*-/58/GK/	/  88 8 	8
 8 8 
8"8s 8 8 8

 
 
(U7<<+<%= ( (0uW\\3%67 04 09
eGLL#4E.F 9
RV 9
v3
M M/C Mt M4r    r   )rd   typingr   r   r   r    playwright._impl._api_structuresr   playwright._impl._artifactr   playwright._impl._connectionr	   r
   playwright._impl._helperr   r    r    r   <module>rm      s*     . . A / L 5t4l t4r    