o eP@sddlZddlZddlZddlmZddlmZddlmZmZddl m Z m Z m Z m Z ddlmZddlmZddlmZdd lmZmZdd lmZdd lmZmZdd lmZdd lmZddl m!Z"ddl#m$Z$ddlm%Z%m&Z&m'Z'm(Z(m)Z)ddl*m+Z+m,Z,ddl-m.Z.ddl/m0Z0m1Z1ddl2m3Z3ddl4m5Z5e6e7Z8GdddZ9Gddde,e9Z:Gddde+e9Z;Gddde:e;ZGd!d"d"e;Z?Gd#d$d$e?ej=Z@Gd%d&d&e:e@ZAdS)'N)forms)settings)adminmessages)ADDITIONCHANGEDELETIONLogEntry)get_permission_codename) ContentType)PermissionDenied) HttpResponseHttpResponseRedirect)TemplateResponse)pathreverse)method_decorator) import_string) gettext_lazy) require_POST)ConfirmImportForm ExportFormImportExportFormBase ImportFormexport_action_form_factory)BaseExportMixinBaseImportMixin) RowResult) post_export post_import)TempFolderStorage)originalcs:eZdZfddZddZddZd fdd ZZS) ImportExportMixinBasecstj|i||dSN)super__init__init_change_list_template)selfargskwargs __class__F/var/www/html/venv/lib/python3.10/site-packages/import_export/admin.pyr&%s zImportExportMixinBase.__init__cCsft|ddr |j|_nd|_z t|dd|_Wnty%tdYnw|jdur1|j|_dSdS)Nchange_list_templatezadmin/change_list.html"import_export_change_list_templatez/failed to assign change_list_template attribute)getattrr/base_change_list_templateAttributeErrorloggerwarningr(r-r-r.r')s     z/ImportExportMixinBase.init_change_list_templatecCs|jjj}||jjjfSr$)model_meta app_label model_name)r(r9r-r-r.get_model_info>s z$ImportExportMixinBase.get_model_infoNcs |pi}|j|d<t||S)Nr2)r2r%changelist_viewr(request extra_contextr+r-r.r<Bs z%ImportExportMixinBase.changelist_viewr$)__name__ __module__ __qualname__r&r'r;r< __classcell__r-r-r+r.r#$s  r#cs0eZdZdZdZdZeZeZ dZ dZ dZ ddZ dd Zd d Zfd d ZeeddZddddZddZddZddZddZddZeddZed d!Zed"d#Zd$d%Zd&d'Zd(d)Z d*d+Z!d>d,d-Z"d.d/Z#d>d0d1Z$d2d3Z%d4d5Z&d6d7Z'd8d9Z(d:d;Z)d>fdttdd}|dur dS|j}t||}|jd|j|fS)zB Returns whether a request has import permission. $IMPORT_EXPORT_IMPORT_PERMISSION_CODENT%s.%sr1roptsr userhas_permr9)r(r>IMPORT_PERMISSION_CODErQcodenamer-r-r.has_import_permissionr z!ImportMixin.has_import_permissioncsRt}|}td|j|jd|dtd|j|jd|dg}||S)Nzprocess_import/z%s_%s_process_importnamezimport/z %s_%s_import)r%get_urlsr;r admin_site admin_viewprocess_import import_action)r(urlsinfomy_urlsr+r-r.rZs    zImportMixin.get_urlscOs||stt|jddr||}n |}||j}|rl|}|t|j d|j d}| r8dn|j }| } | |j d|| d} | } || } |j| ||g|Ri|} | || |SdS)z\ Perform the actual import action (after the user has confirmed the import) is_originalF input_format)encodingNimport_file_name)rYrd read_mode)rVr r1get_confirm_import_formcreate_confirm_formPOSTis_validget_import_formatsint cleaned_data from_encoding is_binaryrL get_read_modereadcreate_datasetprocess_datasetremoveprocess_result)r(r>r)r* confirm_form form_typeimport_formatsrcrdtmp_storage_cls tmp_storagedatadatasetresultr-r-r.r]s@      zImportMixin.process_importF)rollback_on_validation_errorsc Osr|j|g|Rd|i|}||di|}|j|g|Rd|i|} |j|fd|jd|jdd| S)NformForiginal_file_nameT)dry_run file_namerRr~r-)get_import_resource_kwargschoose_import_resource_classget_import_data_kwargs import_datarmgetrR) r(r|rvr>r~r)r* res_kwargsresource imp_kwargsr-r-r.rss:  zImportMixin.process_datasetcCsH||||||tjd|jdtd||jjd}t |S)Nsenderr7zadmin:%s_%s_changelist) current_app) generate_log_entriesadd_success_messager sendr7rr;r[rYr)r(r}r>urlr-r-r.rus   zImportMixin.process_resultc Cs|s[tjttjttjti}tj |j j }|D]C}|j |jkrZ|j |jkrZt'tjdtdtj j|jj ||j|j||j td|j dWdn1sUwYqdSdS)Nignorecategoryz%s through import_export)user_idcontent_type_id object_id object_repr action_flagchange_message)rGrIMPORT_TYPE_NEWrIMPORT_TYPE_UPDATErIMPORT_TYPE_DELETErr objects get_for_modelr7pk import_typeIMPORT_TYPE_ERRORIMPORT_TYPE_SKIPwarningscatch_warningsfilterwarningsPendingDeprecationWarningr log_actionrRrr_)r(r}r> logentry_maprrowr-r-r.rs6   z ImportMixin.generate_log_entriescCs<|jj}td|jtj|jtj|j}t ||dS)Nz/Import finished, with {} new and {} updated {}.) r7r8rformattotalsrrrverbose_name_pluralrsuccess)r(r}r>rQsuccess_messager-r-r.rs  zImportMixin.add_success_messagecK|jdi|SNr-get_context_datar(r*r-r-r.get_import_context_dataz#ImportMixin.get_import_context_datacKiSr$r-rr-r-r.rzImportMixin.get_context_datacCtjdtd|jS)z .. deprecated:: 3.0 Use :meth:`~import_export.admin.ImportMixin.get_import_form_class` or set the new :attr:`~import_export.admin.ImportMixin.import_form_class` attribute. zImportMixin.get_import_form() is deprecated and will be removed in a future release. Please use get_import_form_class() instead.r)rwarnDeprecationWarningimport_form_classr6r-r-r.get_import_form s zImportMixin.get_import_formcCr)z .. deprecated:: 3.0 Use :func:`~import_export.admin.ImportMixin.get_confirm_form_class` or set the new :attr:`~import_export.admin.ImportMixin.confirm_form_class` attribute. zImportMixin.get_confirm_import_form() is deprecated and will be removed in a future release. Please use get_confirm_form_class() instead.r)rrrconfirm_form_classr6r-r-r.rgs z#ImportMixin.get_confirm_import_formcOstjdtd|S)a  .. deprecated:: 3.0 Use :meth:`~import_export.admin.ImportMixin.get_import_form_kwargs` or :meth:`~import_export.admin.ImportMixin.get_confirm_form_kwargs` instead, depending on which form you wish to customize. zImportMixin.get_form_kwargs() is deprecated and will be removed in a future release. Please use get_import_form_kwargs() or get_confirm_form_kwargs() instead.r)rrr)r(rr)r*r-r-r.get_form_kwargs*s zImportMixin.get_form_kwargscCs\|}||}||}t|ts"tjdtd||fi|S||fd|i|S)aL .. versionadded:: 3.0 Return a form instance to use for the 'initial' import step. This method can be extended to make dynamic form updates to the form after it has been instantiated. You might also look to override the following: * :meth:`~import_export.admin.ImportMixin.get_import_form_class` * :meth:`~import_export.admin.ImportMixin.get_import_form_kwargs` * :meth:`~import_export.admin.ImportMixin.get_import_form_initial` * :meth:`~import_export.mixins.BaseImportMixin.get_import_resource_classes` |The ImportForm class must inherit from ImportExportFormBase, this is needed for multiple resource classes to work properly. r resources) rkget_import_form_classget_import_form_kwargs issubclassrrrrget_import_resource_classes)r(r>formats form_classr*r-r-r.create_import_form:s    zImportMixin.create_import_formcC*t|jddstjdtd|S|jS)z .. versionadded:: 3.0 Return the form class to use for the 'import' step. If you only have a single custom form class, you can set the ``import_form_class`` attribute to change this for your subclass. rbFzImportMixin.get_import_form() is deprecated and will be removed in a future release. Please use the new 'import_form_class' attribute to specify a custom form class, or override the get_import_form_class() method if your requirements are more complex.r)r1rrrrrr(r>r-r-r.rW z!ImportMixin.get_import_form_classcCs|jpd|jpd||dS)z .. versionadded:: 3.0 Return a dictionary of values with which to initialize the 'import' form (including the initial values returned by :meth:`~import_export.admin.ImportMixin.get_import_form_initial`). Nr{filesinitial)riFILESget_import_form_initialrr-r-r.rms z"ImportMixin.get_import_form_kwargscCr)z .. versionadded:: 3.0 Return a dictionary of initial field values to be provided to the 'import' form. r-rr-r-r.r{z#ImportMixin.get_import_form_initialcCs$||}|||}|di|S)a .. versionadded:: 3.0 Return a form instance to use for the 'confirm' import step. This method can be extended to make dynamic form updates to the form after it has been instantiated. You might also look to override the following: * :meth:`~import_export.admin.ImportMixin.get_confirm_form_class` * :meth:`~import_export.admin.ImportMixin.get_confirm_form_kwargs` * :meth:`~import_export.admin.ImportMixin.get_confirm_form_initial` Nr-)get_confirm_form_classget_confirm_form_kwargs)r(r> import_formrr*r-r-r.rhs zImportMixin.create_confirm_formcCr)z .. versionadded:: 3.0 Return the form class to use for the 'confirm' import step. If you only have a single custom form class, you can set the ``confirm_form_class`` attribute to change this for your subclass. rbFaImportMixin.get_confirm_import_form() is deprecated and will be removed in a future release. Please use the new 'confirm_form_class' attribute to specify a custom form class, or override the get_confirm_form_class() method if your requirements are more complex.r)r1rgrrrrrr-r-r.rrz"ImportMixin.get_confirm_form_classcCs6|rd}d}n |jp d}|jpd}|||||dS)z .. versionadded:: 3.0 Return a dictionary of values with which to initialize the 'confirm' form (including the initial values returned by :meth:`~import_export.admin.ImportMixin.get_confirm_form_initial`). Nr)rirget_confirm_form_initial)r(r>rr{rr-r-r.rs   z#ImportMixin.get_confirm_form_kwargscCs:|duriS|jdj|jdj|jd|jdddS)z .. versionadded:: 3.0 Return a dictionary of initial field values to be provided to the 'confirm' form. N import_filercr)rerrcr)rmtmp_storage_namerYr)r(r>rr-r-r.rs  z$ImportMixin.get_confirm_form_initialcOs |d}|r|d|SiS)z1 Prepare kwargs for import_data. r)rpop)r(r>r)r*rr-r-r.rs  z"ImportMixin.get_import_data_kwargscCsTd}|s |j}|}|||d}t}|D]}||7}q|||S)N)rdrf)rornrLrpbyteschunkssave)r(rrcrdryrzr{chunkr-r-r.write_to_tmp_storages   z ImportMixin.write_to_tmp_storagecCs.tt|j}tdd|i}|d|dS)Nzk%(exc_name)s encountered while trying to read file. Ensure you have chosen the correct format for the file.exc_namer)reprtyper@r add_error)r(rermsgr-r-r. add_data_read_fail_error_to_formsz,ImportMixin.add_data_read_fail_error_to_formc s||st|}|}t|jddr||}nB||}|j|g|Ri|}t|t rG|||j p8d|j psz-ImportMixin.import_action..ImporttitlerQmediacSs&g|]}|dd|DfqS)cSg|]}|jqSr- column_name)rfr-r-r.rsz8ImportMixin.import_action...)get_display_nameget_user_visible_fields)rrr-r-r.rs  fields_listr-)8rVr rrkr1rrrrrrirrrrrlistrjrlrmrornrdrrrrr Exceptionrerrorsrs has_errorshas_validation_errorsrurrYrrqrrrrrRrhrrupdater[ each_contextrr7r8rrrimport_template_name)r(r>r)r*contextrxrr form_kwargsrrcrr{rr|rr}rzrrrrresource_classesr-rr.r^s&                   zImportMixin.import_actionc(|duri}|||d<t||S)NrV)rVr%r<r=r+r-r.r<zImportMixin.changelist_viewr$),r@rArB__doc__r0rrrrrrnrFrIrGrLrVrZrrr]rsrurrrrr"rrgrrrrrrhrrrrrrr^r<rCr-r-r+r.rDHsT  *         "rDcseZdZdZdZdZdZeZfddZ ddZ d d Z d d Z d dZ ddZeddZddZddZdfdd ZfddZZS) ExportMixinz Export mixin. This is intended to be mixed with django.contrib.admin.ModelAdmin https://docs.djangoproject.com/en/dev/ref/contrib/admin/ z+admin/import_export/change_list_export.htmlzadmin/import_export/export.htmlNcs4t}td|j|jd|dg}||S)Nzexport/z %s_%s_exportrX)r%rZrr[r\ export_actionr;)r(r_rar+r-r.rZs   zExportMixin.get_urlscCrM)zB Returns whether a request has export permission. $IMPORT_EXPORT_EXPORT_PERMISSION_CODENTrOrP)r(r>EXPORT_PERMISSION_CODErQrUr-r-r.has_export_permissionrWz!ExportMixin.has_export_permissionc Cs||}|||}||}||}||}||r&dgt|}||}||j||||j |||j |j |j |d }|j |d<tjdkrN|j|d<Gddd|} | d i|} t| dre| jS| |S) zo Returns export queryset. Default implementation respects applied search and filters. action_checkbox) r>r7 list_displaylist_display_links list_filterdate_hierarchy search_fieldslist_select_related list_per_pagelist_max_show_all list_editable model_admin sortable_by)rsearch_help_textc@seZdZddZdS)z9ExportMixin.get_export_queryset..ExportChangeListcSsdS)a+ Overrides ChangeList.get_results() to bypass default operations like pagination and result counting, which are not needed for export. This prevents executing unnecessary COUNT queries during ChangeList initialization. Nr-rr-r-r. get_resultsrzEExportMixin.get_export_queryset..ExportChangeList.get_resultsN)r@rArBrr-r-r-r.ExportChangeLists rquerysetNr-)get_list_displayget_list_display_linksget_list_select_relatedget_list_filterget_search_fields get_actionsrget_changelistr7rrrrrdjangoVERSIONrhasattrr get_queryset) r(r>r r rr r ChangeListchangelist_kwargsrclr-r-r.get_export_querysets:            zExportMixin.get_export_querysetc Osj|d}||s t|j||g|Ri|}|j||j|jd}|d}|s3|r3| |}|S)zH Returns file_format representation for given queryset. r>) escape_htmlescape_formulaerd) rr r get_data_for_export export_datashould_escape_htmlshould_escape_formulaerroencode) r( file_formatrr)r*r>r{r-rdr-r-r.get_export_datas     zExportMixin.get_export_datacKrrrrr-r-r.get_export_context_datarz#ExportMixin.get_export_context_datacKrr$r-rr-r-r.rrzExportMixin.get_context_datacCr)z .. deprecated:: 3.0 Use :meth:`~import_export.admin.ExportMixin.get_export_form_class` or set the new :attr:`~import_export.admin.ExportMixin.export_form_class` attribute. zExportMixin.get_export_form() is deprecated and will be removed in a future release. Please use the new 'export_form_class' attribute to specify a custom form class, or override the get_export_form_class() method if your requirements are more complex.r)rrrexport_form_classr6r-r-r.get_export_forms zExportMixin.get_export_formcCs|jS)zD Get the form class used to read the export format. )r4r6r-r-r.get_export_form_class(sz!ExportMixin.get_export_form_classc s8|sttjddr}n}}|||jp!dd}|re|t |j d} |}j |||j |d} |} t| | d} d|||f| d<tjdjd | S} | j|td | d <|| d <jj| d <fddD| d<jj|_t|jg| S)NrbF)rr1)r>rd export_form content_typeattachment; filename="%s"Content-DispositionrExportrrrQcs.g|]}|dd|jdDfqS)cSrr-r)rfieldr-r-r.rXsz8ExportMixin.export_action...)r7)rr7r)rresr6r-r.rUsz-ExportMixin.export_action..r)r r r1r5r6get_export_formatsriget_export_resource_classesrjrlrmr)r2 to_encodingget_content_typer get_export_filenamerrr7r3rr[rrr8rYrrexport_template_name) r(r>r)r*rwrrr1rr-r9responserr-r6r.r.sH          zExportMixin.export_actioncr)Nr )r r%r<r=r+r-r.r<brzExportMixin.changelist_viewcs t|Sr$)r%rC)r(r>rr1r+r-r.rChs zExportMixin.get_export_filenamer$)r@rArBrr0rDrArr4rZr r)r2r3rr"r5r6rr<rCrCr-r-r+r.rs$  5 4rc@seZdZdZdZdS)ImportExportMixinz" Import and export mixin. z2admin/import_export/change_list_import_export.htmlN)r@rArBrr0r-r-r-r.rFlsrFc@eZdZdZdS)ImportExportModelAdminzB Subclass of ModelAdmin with import/export functionality. Nr@rArBrr-r-r-r.rHwrHcsHeZdZdZdZfddZddZfddZefd d Z Z S) ExportActionMixinzI Mixin with export functionality implemented as an admin action. Ncsrg}|}|rt|D]\}}|t||fq t|dkr)|ddt||_t j |i|dS)zb Adds a custom action form initialized with the available export formats. rr)rz---N) r? enumerateappendrK get_titleleninsertr action_formr%r&)r(r)r*choicesrirr+r-r.r&s   zExportActionMixin.__init__c Cs~|jd}|st|tddS|}|t|}|j||||jd}| }t ||d}d| |||f|d<|S)z> Exports the selected rows using file_format. r1z!You must select an export format.)r>rdr8r:r;N) rirrr5rr?rlr2rArBr rC) r(r>r export_formatrr1r-r9rEr-r-r.export_admin_actions     z%ExportActionMixin.export_admin_actioncs,t|}|jt|jdtdfd|S)zJ Adds the export action to the list of available actions. rUz'Export selected %(verbose_name_plural)s)rU)r%r rrrUr)r(r>actionsr+r-r.r s zExportActionMixin.get_actionscs tj}tj|jdg|jdS)Nzimport_export/action_formats.js)jscss)r%rrMedia_js_css)r( super_mediar+r-r.rs  zExportActionMixin.media) r@rArBrr0r&rUr propertyrrCr-r-r+r.rK}s  rKc@rG)ExportActionModelAdminz^ Subclass of ModelAdmin with export functionality implemented as an admin action. NrIr-r-r-r.r^rJr^c@rG)ImportExportActionModelAdminz Subclass of ExportActionModelAdmin with import/export functionality. Export functionality is implemented as an admin action. NrIr-r-r-r.r_rJr_)Bloggingrr"r django.confrdjango.contribrrdjango.contrib.admin.modelsrrrr django.contrib.authr "django.contrib.contenttypes.modelsr django.core.exceptionsr django.httpr rdjango.template.responser django.urlsrrdjango.utils.decoratorsrdjango.utils.module_loadingrdjango.utils.translationrrdjango.views.decorators.httprrrrrrmixinsrrresultsrsignalsrr tmp_storagesr!utilsr" getLoggerr@r4r#rDrrF ModelAdminrHrKr^r_r-r-r-r.sJ              $ZO G