o h3@sddZddlZddlmZddlmZddlmZddl m Z m Z ddl m Z ddlmZmZmZddlmZd&dd Zd&d d Zd&d d Zd&ddZd&ddZdejfddZeeeejdZeeedejdZeeeejdZeedeejdZeeeeej dZ!eeeedej dZ"Gddde#Z$Gddde#Z%Gd d!d!e%Z&Gd"d#d#e%Z'd$d%Z(dS)'z pyexcel.internal.meta ~~~~~~~~~~~~~~~~~~~~~~ Annotate sheet and book class' attributes :copyright: (c) 2015-2017 by Onni Software Ltd. :license: New BSD License N)partial)PY2 append_doc)SOURCE) save_book save_sheetget_sheet_stream)make_a_propertycsfdd}d|_|S)z:make a custom presentation method for each file types csZt}||<di|}||z |}|}W|Sty,d}Y|Sw)/docstring is assigned a few lines down the lineN)rget_keyword_for_parameter write_data get_contentgetvalueAttributeError)selfkeywordskeyword memory_sourcecontent_streamcontent attribute source_getterr G/var/www/html/env/lib/python3.10/site-packages/pyexcel/internal/meta.pycustom_presenters    z(make_presenter..custom_presenterzGet data in %s format__doc__)rrrr rrmake_presenters rcCtj}t||S)z0make a custom presentation method for sheet )rget_writable_sourcerrr r rsheet_presenter, r!cCr)z/make a custom presentation method for book )rget_writable_book_sourcerrr r rbook_presenter3r"r$cfdd}d|_|S)z)make a custom input method for sheet cs|i}tjD] }||vr||||<qt}|dkr%||<||d<n|||<tdi|}|j|j|jfi|dS)r file_type file_contentNr ) constantsVALID_SHEET_PARAMETERSpoprr rinitpayloadname)rrr sheet_paramsfieldr named_contentrr rcustom_importer1>s   z"importer..custom_importer1Set data in %s formatr)rr2r r1rimporter:s  r4cr%)z(make a custom input method for book csTt}|dkr||<||d<n|||<tdi|\}}}|j|||ddS)r r&r')sheetsfilenamepathNr )rr _get_bookr+)rrrrr5r6r7r1r rcustom_book_importerUs  z+book_importer..custom_book_importerr3r)rr9r r1r book_importerQs  r:SheetcKs$|||}t|||fi|dS)z1 create custom attributes for each class N)formatr )clsr& instance_name descriptionr doc_stringr r rrds r) getter_funcr?Book)rAr>r?) setter_funcr?)r>rCr?)rArCr?)rArCr>r?c@s eZdZdZddZddZdS)StreamAttributez&Provide access to get_*_stream methodscCs ||_dSN)r=)rr=r r r__init__s zStreamAttribute.__init__cCst|jd}||dS)Nsave_to_memory)r&)getattrr=)rr-getterr r r __getattr__s  zStreamAttribute.__getattr__N)__name__ __module__ __qualname__rrFrJr r r rrDs rDc@sFeZdZdZeddZddZdddZd d Zd d Z d dZ dS) PyexcelObjectz/parent class for pyexcel.Sheet and pyexcel.BookcCst|S)aReturn a stream in which the content is properly encoded Example:: >>> import pyexcel as p >>> b = p.get_book(bookdict={"A": [[1]]}) >>> csv_stream = b.stream.texttable >>> print(csv_stream.getvalue()) A: +---+ | 1 | +---+ Where b.stream.xls.getvalue() is equivalent to b.xls. In some situation b.stream.xls is prefered than b.xls. Sheet examples:: >>> import pyexcel as p >>> s = p.Sheet([[1]], 'A') >>> csv_stream = s.stream.texttable >>> print(csv_stream.getvalue()) A: +---+ | 1 | +---+ Where s.stream.xls.getvalue() is equivalent to s.xls. In some situation s.stream.xls is prefered than s.xls. It is similar to :meth:`~pyexcel.Book.save_to_memory`. )rDrr r rstreams"zPyexcelObject.streamcKstd)aSave the content to memory :param file_type: any value of 'csv', 'tsv', 'csvz', 'tsvz', 'xls', 'xlsm', 'xlsm', 'ods' :param stream: the memory stream to be written to. Note in Python 3, for csv and tsv format, please pass an instance of StringIO. For xls, xlsx, and ods, an instance of BytesIO. z!save to memory is not implemented)NotImplementedError)rr&rr r rrGs zPyexcelObject.save_to_memorysvgcKs<|j|fi|}|dvrdd}t|d|t|||S)a Visualize the data Parameters: ----------------- file_type:string 'svg' by default. 'png', 'jpeg' possible depending on plugins chart_type:string 'bar' by default. other chart types are subjected to plugins. )pngrRjpegcSs|dS)Nutf-8)rdecoderOr r rrsz'PyexcelObject.plot..get_contentz _repr_%s_)rGsetattrr)rr&rmemory_contentrr r rplots zPyexcelObject.plotcCs|jSrE)htmlrOr r r _repr_html_szPyexcelObject._repr_html_cCs*trt}|dkr|j}|dS|jS)NasciirU)rsysgetdefaultencoding texttableencode)rdefault_encodingresultr r r__repr__s  zPyexcelObject.__repr__cCs|SrE)rcrOr r r__str__szPyexcelObject.__str__N)rR) rKrLrMrpropertyrPrGrYr[rcrdr r r rrNs #  rNc@s\eZdZdZeeZeeZee Z e e j ddZd ddZ d ddZ dd d ZdS) SheetMetazAnnotate sheet attributescKt|fd|i|S)z)Save the content to a named file file_namerrr6rr r rsave_asszSheetMeta.save_asNcKt|f||d|}|S)Nr& file_streamrirr&rPrr r rrG szSheetMeta.save_to_memorycCst|||||ddS)aYSave to database table through django model :param model: a database model :param initializer: a initialization functions for your model :param mapdict: custom map dictionary for your data columns :param batch_size: a parameter to Django concerning the size for bulk insertion )model initializermapdict batch_sizeNri)rrprqrrrsr r rsave_to_django_models  zSheetMeta.save_to_django_modelTcCt||||||ddS)aISave data in sheet to database table :param session: database session :param table: a database table :param initializer: a initialization functions for your table :param mapdict: custom map dictionary for your data columns :param auto_commit: by default, data is auto committed. )sessiontablerqrr auto_commitNri)rrvrwrqrrrxr r rsave_to_database%s  zSheetMeta.save_to_databaserE)NNNNNT)rKrLrMr classmethod REGISTER_IO register_ioREGISTER_PRESENTATIONregister_presentationREGISTER_INPUTregister_inputrdocsSAVE_AS_OPTIONSrkrGrtryr r r rrfs   rfc@s`eZdZdZeeZeeZee Z e e j ddZd ddZ d ddZ   dd d ZdS)BookMetazAnnotate book attributescKrg)z0 Save the content to a new file rhrrjr r rrkBszBookMeta.save_asNcKrl)aC Save the content to a memory stream :param file_type: what format the stream is in :param stream: a memory stream. Note in Python 3, for csv and tsv format, please pass an instance of StringIO. For xls, xlsx, and ods, an instance of BytesIO. rmrror r rrGIs zBookMeta.save_to_memorycKst|f|||d|dS)a Save to database table through django model :param models: a list of database models, that is accepted by :meth:`Sheet.save_to_django_model`. The sequence of tables matters when there is dependencies in between the tables. For example, **Car** is made by **Car Maker**. **Car Maker** table should be specified before **Car** table. :param initializers: a list of intialization functions for your tables and the sequence should match tables, :param mapdicts: custom map dictionary for your data columns and the sequence should match tables optional parameters: :param batch_size: django bulk_create batch size :param bulk_save: whether to use bulk_create or to use single save per record )models initializersmapdictsNr)rrrrrr r rsave_to_django_modelsWs zBookMeta.save_to_django_modelsTcCru)aJ Save data in sheets to database tables :param session: database session :param tables: a list of database tables, that is accepted by :meth:`Sheet.save_to_database`. The sequence of tables matters when there is dependencies in between the tables. For example, **Car** is made by **Car Maker**. **Car Maker** table should be specified before **Car** table. :param initializers: a list of intialization functions for your tables and the sequence should match tables, :param mapdicts: custom map dictionary for your data columns and the sequence should match tables :param auto_commit: by default, data is committed. )rvtablesrrrxNr)rrvrrrrxr r rryus zBookMeta.save_to_databaserE)NNrz)rKrLrMrr{REGISTER_BOOK_IOr}REGISTER_BOOK_PRESENTATIONrREGISTER_BOOK_INPUTrrrrrkrGrryr r r rr;s   "rcKs.tjdi|}|}|\}}|||fS)zGet an instance of :class:`Book` from an excel source Where the dictionary should have text as keys and two dimensional array as values. Nr )rget_book_sourceget_dataget_source_info)ra_sourcer5r6r7r r rr8s  r8rE))rr] functoolsrpyexcel.constantsr(pyexcel.docstrings docstringsrpyexcel._compactrrpyexcel.internalrpyexcel.internal.corerrrpyexcel.internal.utilsr rr!r$r4r:OUT_FILE_TYPE_DOC_STRINGrr~rIN_FILE_TYPE_DOC_STRINGrrIO_FILE_TYPE_DOC_STRINGr|robjectrDrNrfrr8r r r rsr             ]< ]