U
    ®[Öe?  ã                   @   s@   d dl Z d dlmZ d dlmZ d dlmZ G dd„ deƒZdS )é    N)Ú	constants)ÚIWriter)ÚCSVZipSheetWriterc                   @   s(   e Zd ZdZdd„ Zdd„ Zdd„ ZdS )	ÚCsvZipWriterzÄ
    csvz writer

    It is better to store csv files as a csvz as it saves your disk space.
    Pyexcel-io had the facility to unzip it for you or you could use
    any other unzip software.
    c                 K   s8   || _ t |dtj¡| _|| _|tjkr4tj| jd< d S )NÚwÚdialect)Ú
_file_typeÚzipfileÚZipFileÚZIP_DEFLATEDÚ	_keywordsr   ZFILE_FORMAT_TSVZZKEYWORD_TSV_DIALECT)ÚselfÚ	file_nameÚ	file_typeÚkeywords© r   ú`/var/www/html/services/stratfitenv/lib/python3.8/site-packages/pyexcel_io/writers/csvz_writer.pyÚ__init__   s
    
zCsvZipWriter.__init__c                 C   s4   |}|d krt j}t| j|| jd d… f| jŽ}|S )Né   )r   ZDEFAULT_SHEET_NAMEr   r	   r   r   )r   ÚnameZ
given_nameÚwriterr   r   r   Úcreate_sheet   s      ÿÿzCsvZipWriter.create_sheetc                 C   s   | j r| j  ¡  d S )N)r	   Úclose)r   r   r   r   r   !   s    zCsvZipWriter.closeN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r   r   r   r   r      s   	r   )r	   Z
pyexcel_ior   Zpyexcel_io.plugin_apir   Zpyexcel_io.writers.csvz_sheetr   r   r   r   r   r   Ú<module>   s   