o h/@sdZddlmZmZddlmZmZddlmZddl Z ddl Z ddl mZmZmZGdd d e jZGd d d eeZGd d d eeZGdddeZGdddeZdddZe eddl mZmZmZmZmZmZ dddZ dddZ dS)a django_excel ~~~~~~~~~~~~~~~~~~~ A django middleware that provides one application programming interface to read and write data in different excel file formats :copyright: (c) 2015 by Onni Software Ltd. :license: New BSD License )MemoryFileUploadHandlerTemporaryFileUploadHandler)InMemoryUploadedFileTemporaryUploadedFile) HttpResponseN)DJANGO_ONE_SIX PY2_VERSION urllib_quotec@sHeZdZdZddZd ddZ  dddZd d d Z  dd d ZdS) ExcelMixinzL Provide additional pyexcel-webio methods to Django's UploadedFiles cKsF|jdd}||d<|jd|j}|r||d<|Std)N. file_typer file_contentzNo content was uploaded.)namesplitfileseekreadIOError)selfkeywords extensioncontentrG/var/www/html/env/lib/python3.10/site-packages/django_excel/__init__.py get_paramss  zExcelMixin.get_paramsNcKs\|jd i|}d|vrd|d<d|vrd|d<||d<||d<||d<tjd i|dS) D Save data from a sheet to a nominated django model name_columns_by_rowrname_rows_by_columnr dest_modeldest_initializer dest_mapdictNr)rpesave_asrmodel initializermapdictrparamsrrrsave_to_database$szExcelMixin.save_to_databasecKL|jdi|}||d<||d<||d<||d<||d<tjdi|dSzD Save data from a book to a nominated django models dest_modelsdest_initializers dest_mapdictsdest_batch_sizedest_bulk_saveNr)rr# save_book_asrmodels initializersmapdicts batch_size bulk_saverr)rrrsave_book_to_database3z ExcelMixin.save_book_to_databasecKsD|jdi|}||d<||d<||d<tjdi||dS)rr r!r"Nr)rr#isave_asfree_resourcesr%rrrisave_to_databaseAs  zExcelMixin.isave_to_databasecKr+r,)rr# isave_book_asr3rrrisave_book_to_databaseMr:z!ExcelMixin.isave_book_to_database)NNN)NNNNN) __name__ __module__ __qualname____doc__rr*r9r=r?rrrrr s   r c@eZdZdZdS)ExcelInMemoryUploadedFilez> Mix-in pyexcel-webio methods in InMemoryUploadedFile Nr@rArBrCrrrrrE\rEc@rD)TemporaryUploadedExcelFilez? Mix-in pyexcel-webio methods in TemporaryUploadedFile NrFrrrrrHcrGrHc@seZdZdZddZdS)ExcelMemoryFileUploadHandlerzP Override MemoryFileUploadHandler to bring in ExcelInMemoryUploadedFile cCsP|jsdS|jdt|j|j|j|j||jd}ts!|j |d<t di|S)Nr)r field_namer content_typesizecharsetcontent_type_extrar) activatedrrdictrJ file_namerKrMrrNrE)r file_sizerrrr file_completens  z*ExcelMemoryFileUploadHandler.file_completeN)r@rArBrCrSrrrrrIjs rIcs eZdZdZfddZZS)TemporaryExcelFileUploadHandlerzT Override TemporaryFileUploadHandler to bring in TemporaryUploadedExcelFile csNtt|j|g|Ri||j|jd|jg}ts ||jt ||_ dS)zK Create the file object to append to as data is coming in. rN) superrnew_filerQrKrMrappendrNrHr)rrQargskwargs custom_args __class__rrrVs  z(TemporaryExcelFileUploadHandler.new_file)r@rArBrCrV __classcell__rrr[rrTsrTcCsFt|||d}|r!trt|tr|d}t|}d||f|d<|S)zB Custom response function that is called by pyexcel-webio )rKstatuszutf-8z+attachment; filename=%s;filename*=utf-8''%szContent-Disposition)rr isinstanceunicodeencoder )rrKr^rQresponseurl_encoded_file_namerrr_make_responses rd) make_responsemake_response_from_arraymake_response_from_dictmake_response_from_recordsmake_response_from_book_dictmake_response_from_query_setscK,tjdd|i|}t|||fd|i|S)z Produce a single sheet Excel book of *file_type* :param model: a Django model :param file_type: same as :meth:`~django_excel.make_response` :param status: same as :meth:`~django_excel.make_response` r&rQNr)r# get_sheetre)r&rr^rQrsheetrrrmake_response_from_a_tables  rocKrl)au Produce a multiple sheet Excel book of *file_type*. It becomes the same as :meth:`~django_excel.make_response_from_a_table` if you pass *tables* with an array that has a single table :param models: a list of Django models :param file_type: same as :meth:`~django_excel.make_response` :param status: same as :meth:`~django_excel.make_response` r4rQNr)r#get_bookre)r4rr^rQrbookrrrmake_response_from_tabless  rr)N)rkN)!rCdjango.core.files.uploadhandlerrrdjango.core.files.uploadedfilerr django.httprpyexcelr# pyexcel_webiowebio_compactrr r ExcelInputr rErHrIrTrd init_webiorerfrgrhrirjrorrrrrrs&  G