o
    &zh                     @  s\   d dl mZ d dlZd dlmZ d dlmZmZ d dlm	Z	m
Z
 eeZG dd dZdS )    )annotationsN)BytesIO)	PdfReader	PdfWriter)getFilepisaTempFilec                   @  sP   e Zd ZddddZdddZeZd	d
 Zdd Zdd ZdddZ	e	Z
e	ZdS )pisaPDFreturnNonec                 C  s   || _ g | _d S N)capacityfiles)selfr    r   H/var/www/html/kangema/venv/lib/python3.10/site-packages/xhtml2pdf/pdf.py__init__   s   
zpisaPDF.__init__Nc                 C  s.   t ||}| }|r| jt| d S d S r   )r   getFileContentr   appendr   )r   urlbasepathobjdatar   r   r   
addFromURI   s
   
zpisaPDF.addFromURIc                 C  s(   t |dr| j| d S | | d S Nread)hasattrr   r   r   )r   fr   r   r   addFromFile'   s   
zpisaPDF.addFromFilec                 C  s.   t | | jd }|r| j| d S d S N)r   )r   encoder   r   r   r   )r   r   r   r   r   r   addFromString-   s   zpisaPDF.addFromStringc                 C  s"   t |jdr| j|j d S d S r   )r   destr   r   )r   docr   r   r   addDocument2   s   zpisaPDF.addDocumentc                 C  sr   t  }| jD ]}t|}tt|jD ]
}||j|  qq|d ur*|| |S t| j	d}|| |
 S r   )r   r   r   rangelenpagesadd_pagewriter   r   getvalue)r   fileoutputpdffilepdf
pageNumberoutr   r   r   join6   s   


zpisaPDF.join)r	   )r
   r   r   )__name__
__module____qualname__r   r   addFromFileNamer   r!   r$   r1   r*   __str__r   r   r   r   r      s    

r   )
__future__r   loggingior   pypdfr   r   xhtml2pdf.filesr   r   	getLoggerr2   logr   r   r   r   r   <module>   s   
