
    ?pjH                         S SK r S SKJr  SSKJrJrJrJr  SSKJ	r	J
r
Jr  SSKJrJrJrJrJrJr  SrSrS	rS
r " S S5      rS\S\S\4S jrS\S\S\4S jrg)    N)Any   )job_status_from_dataparse_batch_statusvalidate_result_json_urlvalidate_state)JobFailedErrorPollTimeoutErrorResultParseError)BatchStatusDocParsingPageDocParsingResult	JobStatusOCRPage	OCRResultg      @g      ?g      .@g     @c            	       p    \ rS rSr\\\\4S\S\S\S\4S jjr	S\
S\4S	 jrS\
S\4S
 jrS\
S\4S jrSrg)Poller,   initial_interval
multipliermax_intervalmax_wait_timec                 @    Xl         X l        X0l        X@l        XPl        g N)_http_initial_interval_multiplier_max_interval_max_wait_time)selfhttp_clientr   r   r   r   s         Y/var/www/html/pdf-tiff/venv/lib/python3.13/site-packages/paddleocr/_api_client/_poller.py__init__Poller.__init__-   s!     !
!1%)+    job_idreturnc                    U R                   n[        R                  " 5       nX0R                  -   n [        R                  " 5       nXT:  a  [	        XU-
  5      eU R
                  R                  U5      n[        U5      nUS:X  a)  [        U5      nU R
                  R                  U5      n	X4$ US:X  a  UR                  SS5      n
[        X5      eU[        R                  " 5       -
  nUS::  a"  [	        U[        R                  " 5       U-
  5      e[        R                  " [        X+5      5        [        X R                  -  U R                  5      nGM$  )NdonefailederrorMsgzUnknown errorr   )r   time	monotonicr   r
   r   get_job_statusr   r   fetch_jsonlgetr	   sleepminr   r   )r    r&   intervalstartdeadlinenowdatastatejson_url
jsonl_data	error_msg	remainings               r"   poll_until_donePoller.poll_until_done;   s   )) ....."C&vU{;;::,,V4D"4(E3D9!ZZ33H=
!''  HHZA	$V77 4>>#33IA~&vt~~/?%/GHHJJs8/08&6&668J8JKH- r%   c                 N    U R                   R                  U5      n[        X5      $ r   )r   r.   r   )r    r&   r7   s      r"   
get_statusPoller.get_statusX   s!    zz((0#F11r%   batch_idc                 N    U R                   R                  U5      n[        X5      $ r   )r   get_batch_statusr   )r    rB   r7   s      r"   rD   Poller.get_batch_status\   s!    zz**84!(11r%   )r   r   r   r   r   N)__name__
__module____qualname____firstlineno__DEFAULT_INITIAL_INTERVALDEFAULT_MULTIPLIERDEFAULT_MAX_INTERVALDEFAULT_MAX_WAIT_TIMEfloatr#   strr   r=   r   r@   r   rD   __static_attributes__ r%   r"   r   r   ,   sy     #;.24,  , 	,
 , ,Lc Lc L:2 2 22 2 2r%   r   r&   r:   r'   c                     / n0 nU H  nUS   n[        UR                  S5      [        5      (       a  UR                  US   5        US    HO  nUR	                  [        US   UR                  S5      UR                  S5      UR                  S5      US95        MQ     M     [        U UUS	9$ ! [        [        4 a  n[        S
U 35      UeS nAff = f)NresultdataInfo
ocrResultsprunedResultocrImagedocPreprocessingImage
inputImage)pruned_resultocr_image_urldoc_preprocessing_image_urlinput_image_urlrawr&   pages	data_infozMalformed OCR result payload: )

isinstancer0   dictupdateappendr   r   KeyError	TypeErrorr   )r&   r:   r`   ra   line_objrS   itemes           r"   parse_ocr_resultrk   a   s    L	"Hh'F&**Z0$77  
!34|,&*>&:&*hhz&:48HH=T4U(,(>  -	 # 
 	

 i  L!?sCD!KLs   B,B/ /C?CCc                     / n0 nU H  nUS   n[        UR                  S5      [        5      (       a  UR                  US   5        US    Hx  nUS   nUR	                  [        US   UR                  S0 5      UR                  S0 5      UR                  S5      UR                  S	5      UR                  S
0 5      UUS95        Mz     M     [        U UUS9$ ! [        [        4 a  n[        SU 35      UeS nAff = f)NrS   rT   layoutParsingResultsmarkdowntextimagesoutputImagesrV   rY   exports)markdown_textmarkdown_imagesoutput_imagesrZ   r]   rr   rn   r^   r_   z+Malformed document parsing result payload: )
rb   r0   rc   rd   re   r   r   rf   rg   r   )	r&   r:   r`   ra   rh   rS   ri   rn   rj   s	            r"   parse_doc_parsing_resultrv   |   s   Y	"Hh'F&**Z0$77  
!3456
+"&.v&6(0Xr(B&*hh~r&B&*hh~&>(,(> $B 7!) 	 7	 #$  
 	

 i  Y!LQCPQWXXYs   CC C<(C77C<)r,   typingr   _corer   r   r   r   errorsr	   r
   r   resultsr   r   r   r   r   r   rJ   rK   rL   rM   r   rO   listrk   rv   rQ   r%   r"   <module>r|      s       
      22 22jLS Ld Ly L6YS Yd Y?O Yr%   