o
    bCcj                     @   s   d dl Z d dlZd dlmZ d dlmZ dd ZG dd dZdd Z	d	d
 Z
dddZdd ZdddZdd Zdd Ze jdkrJd dlmZ dS G dd dejZdS )    N)Pathc                 C   s   | sd S g }|  dD ]G}d|v rI| d\}}t|d }t|d }||k r8|dd t||d D  q|dd t||d dD  q|t|d  q|S )N,-   c                 S      g | ]}|qS  r   .0ir   r   K/home/thesage/.local/lib/python3.10/site-packages/pypdfium2_cli/_parsers.py
<listcomp>       z!parse_numtext.<locals>.<listcomp>c                 S   r   r   r   r   r   r   r   r      r   )splitintextendrangeappend)numtextindicesnum_or_rangestartendr   r   r   parse_numtext
   s    "r   c                   @   s   e Zd Zdd Zdd ZdS )_Rangec                 C   s   || _ || _d S Nr   stop)selfr   r   r   r   r   __init__!   s   
z_Range.__init__c                 C   s   | j  d| j S )Nr   r   r   r   r   r   __repr__%   s   z_Range.__repr__N)__name__
__module____qualname__r   r!   r   r   r   r   r      s    r   c                 C   sn   d}d }g }| D ]"}|d |kr|s|  }n|r#|t|| d }|| |}q|r5|t|| |S )Nr   r   )popr   r   )pagenumsprevrange_startoutnr   r   r   pagenums_ranger(   s    
r+   c                    s   t |  fdd| D S )Nc                 3   s    | ]}t | V  qd S r   )round)r	   vn_digitsr   r   	<genexpr>A   s    zround_list.<locals>.<genexpr>)type)lstr/   r   r.   r   
round_list@   s   r3   Tc                 C   s<   | j dtdd | j ddd |r| j dd tdd	 d S d S )
NinputzInput PDF document)r1   helpz
--passwordz*A password to unlock the PDF, if encrypted)r5   z--pagesz"Page numbers and ranges to include)defaultr1   r5   )add_argumentr   r   )parserpagesr   r   r   	add_inputD   s"   
r:   c                 C   s   | j dtddd d S )Nz
--n-digits   z<Number of digits to which coordinates/sizes shall be rounded)r1   r6   r5   )r7   r   )r8   r   r   r   add_n_digitsX   s   
r<   Fc                 K   sP   t j| jfd| ji|}|r|  d| v r&| js&dd tt|D | _|S )Npasswordr9   c                 S   r   r   r   r   r   r   r   r   f   r   zget_input.<locals>.<listcomp>)pdfiumPdfDocumentr4   r=   
init_formsr9   r   len)argsr@   kwargspdfr   r   r   	get_inputa   s   rE   c                 c   s    | V  |E d H  d S r   r   )valueiteratorr   r   r   _postpeek_generatorm   s   rH   c                 C   s0   zt | }W n
 ty   Y dS w dt|| fS )N)FNT)nextStopIterationrH   )rG   first_valuer   r   r   iterator_hasvaluep   s   rL   )   	   )BooleanOptionalActionc                       s.   e Zd Z fddZdddZdd Z  ZS )	rO   c                    sZ   g }|D ]}| | |drd|dd   }| | qt jd||dd| d S )Nz----no-   r   )option_stringsdestnargsr   )r   
startswithsuperr   )r   rR   rS   rC   _option_stringsoption_string	__class__r   r   r      s   


zBooleanOptionalAction.__init__Nc                 C   s(   || j v rt|| j|d  d S d S )NrP   )rR   setattrrS   rU   )r   r8   	namespacevaluesrX   r   r   r   __call__   s   
zBooleanOptionalAction.__call__c                 C   s   d | jS )Nz | )joinrR   r    r   r   r   format_usage   s   z"BooleanOptionalAction.format_usager   )r"   r#   r$   r   r^   r`   __classcell__r   r   rY   r   rO   ~   s    
rO   )T)F)sysargparsepathlibr   pypdfium2._helpers_helpersr>   r   r   r+   r3   r:   r<   rE   rH   rL   version_inforO   Actionr   r   r   r   <module>   s    	

	
	