o wh@spdZddlmZddlmZmZmZddlmZm Z Gddde Z Gdd d e Z Gd d d e Z d d ZdS)z pyexcel_io.book ~~~~~~~~~~~~~~~~~~~ The io interface to file extensions :copyright: (c) 2014-2017 by Onni Software Ltd. :license: New BSD License, see LICENSE for more details N) OrderedDictisstreamPY2)MESSAGE_ERROR_03MESSAGE_WRONG_IO_INSTANCEc@sTeZdZdZdZddZddZddZd d Zd d Z d dZ ddZ ddZ dS) RWInterfacez7 The common methods for book reader and writer NcCs d|_dSN _file_typeselfrA/var/www/html/env/lib/python3.10/site-packages/pyexcel_io/book.py__init__s zRWInterface.__init__cKtd)zopen a file for read or writePlease implement this methodNotImplementedErrorr file_namekeywordsrrropenzRWInterface.opencKr)z$open a file stream for read or writerrr file_streamrrrr open_streamrzRWInterface.open_streamcKr)%open a file content for read or writerrrrrr open_content!rzRWInterface.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 typeNr )r file_typerrrset_type%s zRWInterface.set_typecCsdS)z4 close the file handle if necessary Nrr rrrclose-szRWInterface.closecCs|Sr rr rrr __enter__5szRWInterface.__enter__cCs |dSr )r!)r a_typevalue tracebackrrr__exit__8s zRWInterface.__exit__) __name__ __module__ __qualname____doc__ stream_typerrrrr r!r"r&rrrrrs rcs`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__rrrAs  zBookReader.__init__cK||_||_dS)zk open a file with unlimited keywords keywords are passed on to individual readers N)r.r0rrrrrH zBookReader.opencKst|rAtrt|dr|dn(t||j}nddlm}z|dWn|y8t||j}Ynw||_ ||_ dSt t )zw open a file with unlimited keywords for reading keywords are passed on to individual readers seekr)UnsupportedOperationN) rrhasattrr6_convert_content_to_streamreadr ior7r/r0IOErrorr)r rrr7rrrrQs$       zBookReader.open_streamcKs"t||j}|j|fi|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)r9r r)r file_contentrrrrrrrs zBookReader.open_contentcsDfdd|jD}t|dkr|dj||diStd)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 %s)r1lenr> read_sheet ValueError)r rBnamed_contentsrrArread_sheet_by_name|s   zBookReader.read_sheet_by_namecCs6z|j|}|j||iWSty|w)z> read an indexed sheet from a excel data book )r1r>rE IndexErrorr!)r sheet_indexsheetrrrread_sheet_by_indexs  zBookReader.read_sheet_by_indexcCs&t}|jD] }||||j<q|Sz8 read everything from a excel data book )rr1rEr>)r resultrKrrrread_alls zBookReader.read_allcCs@t}|D]}t|tr|||q|||q|SrM)r isinstanceintupdaterLrH)r sheetsrNrKrrr read_manys  zBookReader.read_manycCr)zE Return a context specific sheet from a native sheet rr)r native_sheetrrrrEzBookReader.read_sheet)r'r(r)r*rrrrrHrLrOrTrE __classcell__rrr2rr,<s  !    r,csHeZdZdZfddZddZddZdd Zd d Zd d Z Z S) BookWriterz Standard book writer cstt|d|_d|_dSr )r-rXr_file_alike_objectr0r r2rrrs zBookWriter.__init__cKr4)zw open a file with unlimited keywords for writing keywords are passed on to individual writers N)rYr0rrrrrr5zBookWriter.opencKs&t|stt|j|fi|dS)z~ open a file stream with unlimited keywords for writing keywords are passed on to individual writers N)rr<rrrrrrrszBookWriter.open_streamcKr)rz+Normal writer would not need this interface) ExceptionrrrrrrzBookWriter.open_contentcCs:|D]}||}|r||||qtddS)z7 write a dictionary into an excel file zCannot create a sheet writer!N) create_sheet write_arrayr!rZ)r incoming_dictrB sheet_writerrrrwrites  zBookWriter.writecCr)z: implement this method for easy extension zPlease implement create_sheet()r)r rBrrrr[rVzBookWriter.create_sheet) r'r(r)r*rrrrr_r[rWrrr2rrXs    rXcCs"t|}|||d|S)Nr)managerget_ior_r6)r=rstreamrrrr9s   r9)r*pyexcel_io.managerr`pyexcel_io._compactrrr constantsrrobjectrr,rXr9rrrrs -w 5