o e?@svddlZddlmZmZmZddlmZddlZddlmZddl m Z ddl m Z ddl mZddlmZmZdd lmZd d ZGd d d ZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGd d!d!eZGd"d#d#eZ Gd$d%d%eZ!Gd&d'd'eZ"Gd(d)d)eZ#dS)*N)datedatetimetime)Decimal)settings)ObjectDoesNotExist)timezone)parse_duration) force_str smart_str) number_formatcCs:tjddkrtjj|}||Stjj||S)Nr)djangoVERSIONutilsformatssanitize_strftime_formatstrftime datetime_safe new_datetime)valuedatetime_formatformatrH/var/www/html/venv/lib/python3.10/site-packages/import_export/widgets.pyformat_datetimes rc@$eZdZdZdddZdddZdS)Widgetz A Widget takes care of converting between import and export representations. This is achieved by the two methods, :meth:`~import_export.widgets.Widget.clean` and :meth:`~import_export.widgets.Widget.render`. NcKs|S)a Returns an appropriate Python object for an imported value. For example, if you import a value from a spreadsheet, :meth:`~import_export.widgets.Widget.clean` handles conversion of this value into the corresponding Python object. Numbers or dates can be *cleaned* to their respective data types and don't have to be imported as Strings. rselfrrowkwargsrrrclean!s z Widget.cleancCst|S)a Returns an export representation of a Python value. For example, if you have an object you want to export, :meth:`~import_export.widgets.Widget.render` takes care of converting the object's field to a value that can be written to a spreadsheet. )r rrobjrrrrender.sz Widget.renderN__name__ __module__ __qualname____doc__r"r%rrrrrs  rc@s,eZdZdZd ddZddZd dd ZdS) NumberWidgetz Widget for converting numeric fields. :param coerce_to_string: If True, render will return a string representation of the value (None is returned as ""), otherwise the value is returned. FcCs ||_dSr&)coerce_to_string)rr-rrr__init__As zNumberWidget.__init__cCs"t|tr |}|dup|dkSN) isinstancestrstrip)rrrrris_emptyDs zNumberWidget.is_emptyNcCs|jr |dur dSt|S|Sr/)r-r r#rrrr%JszNumberWidget.render)Fr&)r(r)r*r+r.r4r%rrrrr,9s  r,c@eZdZdZdddZdS) FloatWidgetz- Widget for converting float fields. NcKs||rdSt|Sr&)r4floatrrrrr"Us zFloatWidget.cleanr&r(r)r*r+r"rrrrr6Pr6c@r5) IntegerWidgetz/ Widget for converting integer fields. NcK||rdStt|Sr&)r4intrrrrrr"`  zIntegerWidget.cleanr&r8rrrrr:[r9r:c@r5) DecimalWidgetz/ Widget for converting decimal fields. NcKr;r&)r4rr rrrrr"kr=zDecimalWidget.cleanr&r8rrrrr>fr9r>cs,eZdZdZdddZd fdd ZZS) CharWidgeta Widget for converting text fields. :param coerce_to_string: If True, the value returned by clean() is cast to a string. :param allow_blank: If True, and if coerce_to_string is True, then clean() will return null values as empty strings, otherwise as null. FcCs||_||_dS) N)r- allow_blank)rr-rArrrr.{s zCharWidget.__init__Nc sFtj||fi|}|jdur!|dur|jdurdS|St|S|S)NTr0)superr"r-rAr rrr r!val __class__rrr"s  zCharWidget.clean)FFr&)r(r)r*r+r.r" __classcell__rrrErr?qs r?c@s<eZdZdZgdZgdZgdZd ddZd dd ZdS) BooleanWidgeta Widget for converting boolean fields. The widget assumes that ``True``, ``False``, and ``None`` are all valid values, as to match Django's `BooleanField `_. That said, whether the database/Django will actually accept NULL values will depend on if you have set ``null=True`` on that Django field. While the BooleanWidget is set up to accept as input common variations of "True" and "False" (and "None"), you may need to munge less common values to ``True``/``False``/``None``. Probably the easiest way to do this is to override the :func:`~import_export.resources.Resource.before_import_row` function of your Resource class. A short example:: from import_export import fields, resources, widgets class BooleanExample(resources.ModelResource): warn = fields.Field(widget=widgets.BooleanWidget()) def before_import_row(self, row, row_number=None, **kwargs): if "warn" in row.keys(): # munge "warn" to "True" if row["warn"] in ["warn", "WARN"]: row["warn"] = True return super().before_import_row(row, row_number, **kwargs) )1TtrueTRUETrue)0rFfalseFALSEFalse)r0NnullNULLnoneNONENoneNcCs&||jvrdS|r|jdS|jdS)z On export, ``True`` is represented as ``1``, ``False`` as ``0``, and ``None``/NULL as a empty string. Note that these values are also used on the import confirmation view. r0r) NULL_VALUES TRUE_VALUES FALSE_VALUESr#rrrr%s zBooleanWidget.rendercKs ||jvrdS||jvrdSdS)NTF)rWrXrrrrr"s zBooleanWidget.cleanr&) r(r)r*r+rXrYrWr%r"rrrrrHs  rHc@.eZdZdZd ddZd ddZd ddZdS) DateWidgetz Widget for converting date fields. Takes optional ``format`` parameter. If none is set, either ``settings.DATE_INPUT_FORMATS`` or ``"%Y-%m-%d"`` is used. NcC,|durtjs d}ntj}n|f}||_dS)N)z%Y-%m-%d)rDATE_INPUT_FORMATSrrrrrrrr.  zDateWidget.__init__c KV|sdSt|tr |S|jD]}z t||WSttfy&Yqwtd)NzEnter a valid date.)r1rrrstrptime ValueError TypeErrorrrr r!rrrrr"  zDateWidget.cleancCs|sdSt||jdSNr0r)rrr#rrrr%zDateWidget.renderr&r(r)r*r+r.r"r%rrrrr[    r[c@rZ) DateTimeWidgetz Widget for converting date fields. Takes optional ``format`` parameter. If none is set, either ``settings.DATETIME_INPUT_FORMATS`` or ``"%Y-%m-%d %H:%M:%S"`` is used. NcCr\)N)z%Y-%m-%d %H:%M:%S)rDATETIME_INPUT_FORMATSrr^rrrr.r_zDateTimeWidget.__init__c Ksxd}|sdSt|tr|}n|jD]}zt||}Wqttfy&Yqw|r8tjr6t |r6t |}|Std)NzEnter a valid date/time.) r1rrrarbrcrUSE_TZris_naive make_aware)rrr r!dtformat_rrrr"s    zDateTimeWidget.cleancCs(|sdStjr t|}t||jdSrf)rrlr localtimerrr#rrrr%s  zDateTimeWidget.renderr&rhrrrrrjs   rjc@rZ) TimeWidgetz Widget for converting time fields. Takes optional ``format`` parameter. If none is set, either ``settings.DATETIME_INPUT_FORMATS`` or ``"%H:%M:%S"`` is used. NcCr\)N)z%H:%M:%S)rTIME_INPUT_FORMATSrr^rrrr.r_zTimeWidget.__init__c Kr`)NzEnter a valid time.)r1rrrrarbrcrdrrrr" rezTimeWidget.cleancCs|sdS||jdSrf)rrr#rrrr%,rgzTimeWidget.renderr&rhrrrrrrrirrc@r)DurationWidgetz5 Widget for converting time duration fields. Nc Ks.|sdSzt|WSttfytdw)NzEnter a valid duration.)r rbrcrrrrr"7s zDurationWidget.cleancCs|durdSt|Sr/r2r#rrrr%@szDurationWidget.renderr&r'rrrrrt2s  rtcs6eZdZdZd fdd Zd ddZd ddZZS) SimpleArrayWidgetzv Widget for an Array field. Can be used for Postgres' Array field. :param separator: Defaults to ``','`` Ncs |durd}||_tdS)N,) separatorrBr.)rrxrErrr.MszSimpleArrayWidget.__init__cKs|r||jSgSr&)splitrxrrrrr"SszSimpleArrayWidget.cleancCs|jdd|DS)Ncss|]}t|VqdSr&ru).0vrrr Wsz+SimpleArrayWidget.render..)rxjoinr#rrrr%VszSimpleArrayWidget.renderr&r(r)r*r+r.r"r%rGrrrErrvFs  rvcs,eZdZdZdfdd ZdddZZS) JSONWidgeta$ Widget for a JSON object (especially required for jsonb fields in PostgreSQL database.) :param value: Defaults to JSON format. The widget covers two cases: Proper JSON string with double quotes, else it tries to use single quotes and then convert it to proper JSON. Nc sJt|}|r#zt|WStjjy"t|ddYSwdS)N'")rBr"jsonloadsdecoderJSONDecodeErrorreplacerCrErrr"ds  zJSONWidget.cleancCs|rt|SdSr&)rdumpsr#rrrr%ls zJSONWidget.renderr&)r(r)r*r+r"r%rGrrrErrZs rcsBeZdZdZd fdd ZddZdfd d Zdd d ZZS)ForeignKeyWidgetu Widget for a ``ForeignKey`` field which looks up a related model using either the PK or a user specified field that uniquely identifies the instance in both export and import. The lookup field defaults to using the primary key (``pk``) as lookup criterion but can be customized to use any field on the related model. Unlike specifying a related field in your resource like so… :: class Meta: fields = ('author__name',) …using a :class:`~import_export.widgets.ForeignKeyWidget` has the advantage that it can not only be used for exporting, but also importing data with foreign key relationships. Here's an example on how to use :class:`~import_export.widgets.ForeignKeyWidget` to lookup related objects using ``Author.name`` instead of ``Author.pk``:: from import_export import fields, resources from import_export.widgets import ForeignKeyWidget class BookResource(resources.ModelResource): author = fields.Field( column_name='author', attribute='author', widget=ForeignKeyWidget(Author, 'name')) class Meta: fields = ('author',) :param model: The Model the ForeignKey refers to (required). :param field: A field on the related model used for looking up a particular object. :param use_natural_foreign_keys: Use natural key functions to identify related object, default to False pkFc s(||_||_||_tjdi|dSNr)modelfielduse_natural_foreign_keysrBr.)rrrrr!rErrr.szForeignKeyWidget.__init__cOs |jjS)a% Returns a queryset of all objects for this Model. Overwrite this method if you want to limit the pool of objects from which the related object is retrieved. :param value: The field's value in the datasource. :param row: The datasource's current row. As an example; if you'd like to have ForeignKeyWidget look up a Person by their pre- **and** lastname column, you could subclass the widget like so:: class FullNameForeignKeyWidget(ForeignKeyWidget): def get_queryset(self, value, row, *args, **kwargs): return self.model.objects.filter( first_name__iexact=row["first_name"], last_name__iexact=row["last_name"] ) )robjectsall)rrr argsr!rrr get_querysets zForeignKeyWidget.get_querysetNc sXt|}|r*|jrt|}|jjj|S|j||fi|j di|j |iSdSr) rBr"rrrrrget_by_natural_keyrgetrrCrErrr"s  &zForeignKeyWidget.cleanc Csz|durdS|jd}|D],}z|jrt|WSt||d}Wn ttfy2YdSw|dur:dSq|S)Nr0__) rryrrr natural_keygetattrrbr)rrr$attrsattrrrrr%s zForeignKeyWidget.render)rFr&) r(r)r*r+r.rr"r%rGrrrErrqs * rcs6eZdZdZd fdd Zd ddZd ddZZS) ManyToManyWidgeta8 Widget that converts between representations of a ManyToMany relationships as a list and an actual ManyToMany field. :param model: The model the ManyToMany field refers to (required). :param separator: Defaults to ``','``. :param field: A field on the related model. Default is ``pk``. Nc s@|durd}|dur d}||_||_||_tjdi|dS)Nrwrr)rrxrrBr.)rrrxrr!rErrr.szManyToManyWidget.__init__cKsh|s|jjSt|ttfrt|g}n||j}tddd|D}|jjjdid|j |iS)NcSsg|]}|qSr)r3)rzirrr sz*ManyToManyWidget.clean..z%s__inr) rrrTr1r7r<ryrxfilterr)rrr r!idsrrrr"s   zManyToManyWidget.cleancs"fdd|D}j|S)Ncsg|] }tt|jqSr)r rr)rzr$rrrrsz+ManyToManyWidget.render..)rrxr})rrr$rrrrr%s zManyToManyWidget.render)NNr&r~rrrErrs   r)$rrrrdecimalrr django.confrdjango.core.exceptionsr django.utilsrdjango.utils.dateparser django.utils.encodingr r django.utils.formatsr rrr,r6r:r>r?rHr[rjrrrtrvrrrrrrrs4       !   3$,$j