U g?@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-bv_uwtkt/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 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 ||_dSN)coerce_to_string)rr(rrr__init__AszNumberWidget.__init__cCs"t|tr|}|dkp |dkSN) isinstancestrstrip)rrrrris_emptyDs zNumberWidget.is_emptyNcCs|jr|dkrdSt|S|Sr*)r(r rrrrr JszNumberWidget.render)F)N)r"r#r$r%r)r/r rrrrr&9s r&c@seZdZdZdddZdS) FloatWidgetz- Widget for converting float fields. NcKs||rdSt|Sr')r/floatrrrrrUs zFloatWidget.clean)Nr"r#r$r%rrrrrr0Psr0c@seZdZdZdddZdS) IntegerWidgetz/ Widget for converting integer fields. NcKs||rdStt|Sr')r/intrrrrrr`s zIntegerWidget.clean)Nr2rrrrr3[sr3c@seZdZdZdddZdS) DecimalWidgetz/ Widget for converting decimal fields. NcKs||rdStt|Sr')r/rr rrrrrks zDecimalWidget.clean)Nr2rrrrr5fsr5cs,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(r8rrrr){szCharWidget.__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)rMrNrrrrrs zBooleanWidget.clean)N)N) r"r#r$r%rNrOrMr 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 rrrrrPs rPc@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_FORMATSrrQrrrr)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,rrrRrSrTrUSE_TZrZis_naiveZ make_aware)rrrrdtZformat_rrrrs    zDateTimeWidget.cleancCs(|sdStjrt|}t||jdSrV)rrYr localtimerrrrrrr s  zDateTimeWidget.render)N)N)NrWrrrrrXs rXc@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_FORMATSrrQrrrr)s zTimeWidget.__init__c Ksb|sdSt|tr|S|jD]8}zt||WSttfk rRYqYqXqtddS)NzEnter a valid time.)r,rrrrRrSrTrUrrrr s   zTimeWidget.cleancCs|sdS||jdSrV)rrrrrrr ,szTimeWidget.render)N)N)NrWrrrrr\s r\c@s$eZdZdZdddZdddZdS) DurationWidgetz5 Widget for converting time duration fields. Nc Ks8|sdSz t|WSttfk r2tdYnXdS)NzEnter a valid duration.)r rSrTrrrrr7s  zDurationWidget.cleancCs|dkr dSt|Sr*r-rrrrr @szDurationWidget.render)N)Nr!rrrrr]2s 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,) separatorr9r))rrar<rrr)MszSimpleArrayWidget.__init__cKs|r||jSgSr')splitrarrrrrSszSimpleArrayWidget.cleancCs|jdd|DS)Ncss|]}t|VqdSr'r^).0vrrr Wsz+SimpleArrayWidget.render..)rajoinrrrrr VszSimpleArrayWidget.render)N)N)Nr"r#r$r%r)rr r>rrr<rr_Fs 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'")r9rjsonloadsdecoderJSONDecodeErrorreplacer:r<rrrds   zJSONWidget.cleancCs|rt|SdSr')rkdumpsrrrrr lszJSONWidget.render)N)N)r"r#r$r%rr r>rrr<rrhZs rhcsBeZdZdZd 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_keysr9r))rrsrtrurr<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"] ) )rsobjectsall)rrrargsrrrr get_querysetszForeignKeyWidget.get_querysetNc sVt|}|rN|jr.t|}|jjj|S|j||f|j f|j |iSndSdSr') r9rrurkrlrsrvZget_by_natural_keyrygetrtr:r<rrrs   zForeignKeyWidget.cleanc Cs||dkr dS|jd}|D]Z}z*|jrrrr<rrqqs * rqcs6eZdZdZd 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<|dkr d}|dkrd}||_||_||_tjf|dS)Nr`rr)rsrartr9r))rrsrartrr<rrr)szManyToManyWidget.__init__cKsd|s|jjSt|ttfr*t|g}n ||j}tddd|D}|jjjfd|j |iS)NcSsg|] }|qSr)r.)rcirrr sz*ManyToManyWidget.clean..z%s__in) rsrvrJr,r1r4rbrafilterrt)rrrridsrrrrs   zManyToManyWidget.cleancs"fdd|D}j|S)Ncsg|]}tt|jqSr)r r|rt)rcrrrrrsz+ManyToManyWidget.render..)rwrarf)rrrrrrrr szManyToManyWidget.render)NN)N)Nrgrrr<rrs  r)$rkrrrdecimalrrZ django.confrZdjango.core.exceptionsrZ django.utilsrZdjango.utils.dateparser Zdjango.utils.encodingr r Zdjango.utils.formatsr rrr&r0r3r5r6r?rPrXr\r]r_rhrqrrrrrs2       !   3$,$j