U [e3@sddZddlZddlmZddlmZddlmZddlm Z m Z ddl m Z ddl mZmZmZdd lmZd(d d Zd)d d Zd*ddZd+ddZd,ddZdejfddZeeeejdZeeedejdZeeeejdZeedeejdZeeeeejdZ eeeedejdZ!Gddde"Z#Gd d!d!e"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-2022 by Onni Software Ltd. :license: New BSD License N)partial) constants) docstrings)PY2 append_doc)SOURCE) save_book save_sheetget_sheet_stream)make_a_propertycsfdd}d|_|S)z5make a custom presentation method for each file typescsXt}||<f|}||z|}|}Wntk rRd}YnX|S)/docstring is assigned a few lines down the lineN)rget_keyword_for_parameter write_data get_contentgetvalueAttributeError)selfkeywordskeywordZ memory_sourceZcontent_streamcontent attribute source_getterW/var/www/html/services/stratfitenv/lib/python3.8/site-packages/pyexcel/internal/meta.pycustom_presenters     z(make_presenter..custom_presenterzGet data in %s format__doc__)rrrrrrmake_presenters rcCstj}t||S)z+make a custom presentation method for sheet)rZget_writable_sourcerrrrrsheet_presenter+srcCstj}t||S)z*make a custom presentation method for book)rZget_writable_book_sourcerrrrrbook_presenter1sr csfdd}d|_|S)z$make a custom input method for sheetcsti}tjD]}||kr ||||<q t}|dkrJ||<||d<n|||<tf|}|j|j|jf|dS)r file_type file_contentN) rZVALID_SHEET_PARAMETERSpoprr r initpayloadname)rrrZ sheet_paramsfieldrZ named_contentrrrcustom_importer1:s    z"importer..custom_importer1Set data in %s formatr)rr)rr(rimporter7s  r+csfdd}d|_|S)z#make a custom input method for bookcsPt}|dkr$||<||d<n|||<tf|\}}}|j|||ddS)r r!r")sheetsfilenamepathN)rr _get_bookr$)rrrrr,r-r.r(rrcustom_book_importerPs  z+book_importer..custom_book_importerr*r)rr0rr(r book_importerMs r1ZSheetcKs |||}t|||f|dS)z1 create custom attributes for each class N)formatr )clsr! instance_name descriptionrZ doc_stringrrrr_s r) getter_funcr5ZBook)r6r4r5) setter_funcr5)r4r7r5)r6r7r5)r6r7r4r5c@s eZdZdZddZddZdS)StreamAttributez&Provide access to get_*_stream methodscCs ||_dSN)r3)rr3rrr__init__szStreamAttribute.__init__cCst|jd}||dS)Nsave_to_memory)r!)getattrr3)rr&getterrrr __getattr__s zStreamAttribute.__getattr__N)__name__ __module__ __qualname__rr:r>rrrrr8sr8c@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`. )r8rrrrstreams"zPyexcelObject.streamcKs tddS)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 implementedN)NotImplementedError)rr!rrrrr;s zPyexcelObject.save_to_memorysvgcKs8|j|f|}|dkr4dd}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. )ZpngrFZjpegcSs|dS)Nutf-8)rdecoderCrrrrsz'PyexcelObject.plot..get_contentz _repr_%s_)r;setattrr)rr!rZmemory_contentrrrrplots zPyexcelObject.plotcCs|jSr9)htmlrCrrr _repr_html_szPyexcelObject._repr_html_cCs*tr$t}|dkr$|j}|dS|jS)NasciirG)rsysgetdefaultencodingZ texttableencode)rZdefault_encodingresultrrr__repr__s  zPyexcelObject.__repr__cCs|Sr9)rRrCrrr__str__szPyexcelObject.__str__N)rF) r?r@rArpropertyrDr;rJrLrRrSrrrrrBs #  rBc@sXeZdZdZeeZeeZee Z e e j ddZd ddZd ddZdd d ZdS) SheetMetazAnnotate sheet attributescKst|fd|i|S)z Save the content to a named file file_namer rr-rrrrsave_asszSheetMeta.save_asNcKst|f||d|}|S)Nr!Z file_streamrWrr!rDrrrrr;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_sizeNrW)rr\r]r^r_rrrsave_to_django_model s zSheetMeta.save_to_django_modelTcCst||||||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. )sessiontabler]r^ auto_commitNrW)rrarbr]r^rcrrrsave_to_databases zSheetMeta.save_to_database)N)NNN)NNT)r?r@rAr classmethod REGISTER_IO register_ioREGISTER_PRESENTATIONregister_presentationREGISTER_INPUTregister_inputrdocsSAVE_AS_OPTIONSrYr;r`rdrrrrrUs   rUc@sXeZdZdZeeZeeZee Z e e j ddZd ddZd ddZdd d ZdS)BookMetazAnnotate book attributescKst|fd|i|S)z0 Save the content to a new file rVrrXrrrrY<szBookMeta.save_asNcKst|f||d|}|S)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. rZror[rrrr;Cs 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 initializersmapdictsNro)rrprqrrrrrrsave_to_django_modelsQszBookMeta.save_to_django_modelsTcCst||||||ddS)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. )ratablesrqrrrcNro)rrartrqrrrcrrrrdoszBookMeta.save_to_database)N)NN)NNT)r?r@rArreREGISTER_BOOK_IOrgREGISTER_BOOK_PRESENTATIONriREGISTER_BOOK_INPUTrkrrlrmrYr;rsrdrrrrrn5s   "rncKs*tjf|}|}|\}}|||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. )rZget_book_sourceget_dataZget_source_info)rZa_sourcer,r-r.rrrr/s  r/)N)N)N)N)N)(rrN functoolsrZpyexcelrrrlZpyexcel._compactrrZpyexcel.internalrZpyexcel.internal.corerr r Zpyexcel.internal.utilsr rrr r+r1ZOUT_FILE_TYPE_DOC_STRINGrrhrvZIN_FILE_TYPE_DOC_STRINGrjrwZIO_FILE_TYPE_DOC_STRINGrfruobjectr8rBrUrnr/rrrrsp              ];]