U >Ke?@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>tjddkr&tjj|}||Stjj||SdS)Nr)djangoVERSIONutilsformatsZsanitize_strftime_formatstrftimeZ datetime_safeZ new_datetime)valueZdatetime_formatformatr9/tmp/pip-unpacked-wheel-v_jw5did/import_export/widgets.pyformat_datetimes rc@s$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.render)N)N__name__ __module__ __qualname____doc__rr rrrrrs rc@s,eZdZdZd ddZddZd dd ZdS) NumberWidgetz Takes optional ``coerce_to_string`` parameter, set to ``True`` the :meth:`~import_export.widgets.Widget.render` method will return a string else it will return a value. FcCs ||_dSN)coerce_to_string)rr(rrr__init__@szNumberWidget.__init__cCs"t|tr|}|dkp |dkSN) isinstancestrstrip)rrrrris_emptyCs zNumberWidget.is_emptyNcCs|jrt|S|Sr')r(r rrrrr IszNumberWidget.render)F)N)r"r#r$r%r)r/r rrrrr&9s r&c@seZdZdZdddZdS) FloatWidgetz. Widget for converting floats fields. NcKs||rdSt|Sr')r/floatrrrrrRs zFloatWidget.clean)Nr"r#r$r%rrrrrr0Msr0c@seZdZdZdddZdS) IntegerWidgetz/ Widget for converting integer fields. NcKs||rdStt|Sr')r/intrrrrrr]s zIntegerWidget.clean)Nr2rrrrr3Xsr3c@seZdZdZdddZdS) DecimalWidgetz/ Widget for converting decimal fields. NcKs||rdStt|Sr')r/rr rrrrrhs zDecimalWidget.clean)Nr2rrrrr5csr5cs,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||_||_dSr')r( allow_blank)rr(r7rrrr)xszCharWidget.__init__Nc s@tj||f|}|jdkr<|dkr4|jdkr`_. 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) 1TtrueTRUETrue0rFfalseFALSEFalser+NnullZNULLnoneNONENonecCs&||jkrdS|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. r+r) NULL_VALUES TRUE_VALUES FALSE_VALUESrrrrr s zBooleanWidget.rendercKs ||jkrdS||jkrdSdS)NTF)rLrMrrrrrs zBooleanWidget.clean)N)N) r"r#r$r%rMrNrLr rrrrrr>s  r>c@s.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. NcCs,|dkrtjsd}q"tj}n|f}||_dS)N)z%Y-%m-%d)rZDATE_INPUT_FORMATSrrrrrrrr)s zDateWidget.__init__c Ksb|sdSt|tr|S|jD]8}zt||WSttfk rRYqYqXqtddS)NzEnter a valid date.)r,rrrstrptime ValueError TypeErrorrrrrrrrrrs   zDateWidget.cleancCs|sdSt||jdSNr+r)rrrrrrr szDateWidget.render)N)N)Nr"r#r$r%r)rr rrrrrOs rOc@s.eZdZdZd ddZd ddZd ddZdS) 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. NcCs,|dkrtjsd}q"tj}n|f}||_dS)N)z%Y-%m-%d %H:%M:%S)rZDATETIME_INPUT_FORMATSrrPrrrr)s zDateTimeWidget.__init__c Ksd}|s dSt|tr|}n:|jD]2}zt||}Wq"ttfk rRYq"Yq"Xq"|rxtjrtt |rtt |}|StddS)NzEnter a valid date/time.) r,rrrQrRrSrUSE_TZrZis_naiveZ make_aware)rrrrdtZformat_rrrrs    zDateTimeWidget.cleancCs(|sdStjrt|}t||jdSrU)rrXr localtimerrrrrrr s  zDateTimeWidget.render)N)N)NrVrrrrrWs rWc@s.eZdZdZd ddZd ddZd ddZdS) 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. NcCs,|dkrtjsd}q"tj}n|f}||_dS)N)z%H:%M:%S)rZTIME_INPUT_FORMATSrrPrrrr)s zTimeWidget.__init__c Ksb|sdSt|tr|S|jD]8}zt||WSttfk rRYqYqXqtddS)NzEnter a valid time.)r,rrrrQrRrSrTrrrrs   zTimeWidget.cleancCs|sdS||jdSrU)rrrrrrr (szTimeWidget.render)N)N)NrVrrrrr[ s r[c@s$eZdZdZdddZdddZdS) DurationWidgetz5 Widget for converting time duration fields. Nc Ks8|sdSz t|WSttfk r2tdYnXdS)NzEnter a valid duration.)r rRrSrrrrr3s  zDurationWidget.cleancCs|dkr dSt|Sr*r-rrrrr <szDurationWidget.render)N)Nr!rrrrr\.s r\cs6eZdZdZd fdd Zd ddZd ddZZS) SimpleArrayWidgetzv Widget for an Array field. Can be used for Postgres' Array field. :param separator: Defaults to ``','`` Ncs |dkr d}||_tdS)N,) separatorr8r))rr`r;rrr)IszSimpleArrayWidget.__init__cKs|r||jSgSr')splitr`rrrrrOszSimpleArrayWidget.cleancCs|jdd|DS)Ncss|]}t|VqdSr'r]).0vrrr Ssz+SimpleArrayWidget.render..)r`joinrrrrr RszSimpleArrayWidget.render)N)N)Nr"r#r$r%r)rr r=rrr;rr^Bs r^cs,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 sLt|}|rHz t|WStjjk rFt|ddYSXdS)N'")r8rjsonloadsdecoderJSONDecodeErrorreplacer9r;rrr`s   zJSONWidget.cleancCs|rt|SdSr')rjdumpsrrrrr hszJSONWidget.render)N)N)r"r#r$r%rr r=rrr;rrgVs rgcsBeZdZdZd 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$||_||_||_tjf|dSr')modelfielduse_natural_foreign_keysr8r))rrrrsrtrr;rrr)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"] ) )rrobjectsall)rrrargsrrrr get_querysetszForeignKeyWidget.get_querysetNc sVt|}|rN|jr.t|}|jjj|S|j||f|j f|j |iSndSdSr') r8rrtrjrkrrruZget_by_natural_keyrxgetrsr9r;rrrs   zForeignKeyWidget.cleanc Cs||dkr dS|jd}|D]Z}z*|jrsz*ManyToManyWidget.clean..z%s__in) rrrurIr,r1r4rar`filterrs)rrrridsrrrrs   zManyToManyWidget.cleancs"fdd|D}j|S)Ncsg|]}tt|jqSr)r r{rs)rbrrrrrsz+ManyToManyWidget.render..)rvr`re)rrrrrrrr szManyToManyWidget.render)NN)N)Nrfrrr;rr~s  r~)$rjrrrdecimalrrZ django.confrZdjango.core.exceptionsrZ django.utilsrZdjango.utils.dateparser Zdjango.utils.encodingr r Zdjango.utils.formatsr rrr&r0r3r5r6r>rOrWr[r\r^rgrpr~rrrrs2       !   3$,$j