U [eq@sxdZddlZddlmZddlmZmZddlmZm Z dZ Gddde Z Gd d d e Z Gd d d e Zd dZdS)z pyexcel_io.book ~~~~~~~~~~~~~~~~~~~ The io interface to file extensions :copyright: (c) 2014-2022 by Onni Software Ltd. :license: New BSD License, see LICENSE for more details N) OrderedDictisstream)MESSAGE_ERROR_03MESSAGE_WRONG_IO_INSTANCEzxDeprecated since v0.6.0! Although backward compatibility is preserved, it is recommended to upgrade to get new features.c@sTeZdZdZdZddZddZddZd d Zd d Z d dZ ddZ ddZ dS) RWInterfacez7 The common methods for book reader and writer NcCsttd|_dSN)warningswarnDEPRECATED_SINCE_0_6_0 _file_typeselfrQ/var/www/html/services/stratfitenv/lib/python3.8/site-packages/pyexcel_io/book.py__init__s zRWInterface.__init__cKs tddS)zopen a file for read or writePlease implement this methodNNotImplementedErrorr file_namekeywordsrrropen#szRWInterface.opencKs tddS)z$open a file stream for read or writerNrr file_streamrrrr open_stream'szRWInterface.open_streamcKs tddS)%open a file content for read or writerNrrrrr open_content+szRWInterface.open_contentcCs ||_dS)z set the file type for the instance file type is needed when a third party library could handle more than one file typeN)r )r file_typerrrset_type/szRWInterface.set_typecCsdS)z4 close the file handle if necessary Nrr rrrclose7szRWInterface.closecCs|Srrr rrr __enter__?szRWInterface.__enter__cCs |dSr)r )ra_typevalue tracebackrrr__exit__BszRWInterface.__exit__) __name__ __module__ __qualname____doc__Z stream_typerrrrrr r!r%rrrrrsrcs`eZdZdZfddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ Z S) BookReaderz Standard book reader cs*tt|d|_d|_d|_d|_dSr)superr*r _file_name _file_stream _keywords _native_bookr  __class__rrrKs zBookReader.__init__cKs||_||_dS)zk open a file with unlimited keywords keywords are passed on to individual readers N)r,r.rrrrrRszBookReader.opencKsbt|rVddlm}z|dWn$|k rFt||j}YnX||_||_nt t dS)zw open a file with unlimited keywords for reading keywords are passed on to individual readers r)UnsupportedOperationN) rior2seek_convert_content_to_streamreadr r-r.IOErrorr)rrrr2rrrr  zBookReader.open_streamcKst||j}|j|f|dS)z read file content as if it is a file stream with unlimited keywords for reading keywords are passed on to individual readers N)r5r r)r file_contentrrrrrrqs zBookReader.open_contentcsHfdd|jD}t|dkr8|dj||diStddS)z; read a named sheet from a excel data book csg|]}|jkr|qSr)name).0content sheet_namerr s z1BookReader.read_sheet_by_name..rrzCannot find sheet %sN)r/lenr9 read_sheet ValueError)rr=Znamed_contentsrr<rread_sheet_by_name{s   zBookReader.read_sheet_by_namecCs@z|j|}|j||iWStk r:|YnXdS)z> read an indexed sheet from a excel data book N)r/r9r@ IndexErrorr )rZ sheet_indexsheetrrrread_sheet_by_indexs  zBookReader.read_sheet_by_indexcCs&t}|jD]}||||j<q |Sz8 read everything from a excel data book )rr/r@r9)rresultrDrrrread_alls zBookReader.read_allcCs@t}|D]0}t|tr*|||q |||q |SrF)r isinstanceintupdaterErB)rZsheetsrGrDrrr read_manys  zBookReader.read_manycCs tddS)zE Return a context specific sheet from a native sheet rNr)rZ native_sheetrrrr@szBookReader.read_sheet)r&r'r(r)rrrrrBrErHrLr@ __classcell__rrr0rr*Fs      r*csHeZdZdZfddZddZddZdd Zd d Zd d Z Z S) BookWriterz Standard book writer cstt|d|_d|_dSr)r+rNr_file_alike_objectr.r r0rrrszBookWriter.__init__cKs||_||_dS)zw open a file with unlimited keywords for writing keywords are passed on to individual writers N)rOr.rrrrrszBookWriter.opencKs"t|stt|j|f|dS)z~ open a file stream with unlimited keywords for writing keywords are passed on to individual writers N)rr7rrrrrrrszBookWriter.open_streamcKs tddS)rz+Normal writer would not need this interfaceN) ExceptionrrrrrszBookWriter.open_contentcCs<|D]2}||}|r.||||qtdqdS)z7 write a dictionary into an excel file zCannot create a sheet writer!N) create_sheet write_arrayr rP)rZ incoming_dictr=Z sheet_writerrrrwrites   zBookWriter.writecCs tddS)z: implement this method for easy extension zPlease implement create_sheet()Nr)rr=rrrrQszBookWriter.create_sheet) r&r'r(r)rrrrrSrQrMrrr0rrNs    rNcCspt|}t|}|dko&t|t }|dko8t|t}|rJ|d}n|rX|d}|||d|S)Nbytesstringzutf-8r) managerZget_ioZ get_io_typerIrTencodedecoderSr4)r8rstreamZtarget_content_typeZ needs_encodeZ needs_decoderrrr5s"        r5)r)r Zpyexcel_io.managerrVZpyexcel_io._compactrr constantsrrr objectrr*rNr5rrrrs  .l5