U ‰d°:ã@sjddlZddlmZmZmZddlmZddlZddlmZddl m Z ddl m Z ddl mZddlmZmZd d „ZGd d „d ƒZGd d„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd „d eƒZGd!d"„d"eƒZGd#d$„d$eƒZGd%d&„d&eƒZ Gd'd(„d(eƒZ!dS))éN)ÚdateÚdatetimeÚtime)ÚDecimal)Úsettings)ÚObjectDoesNotExist)Útimezone)Úparse_duration)Ú force_strÚ smart_strcCs>tjddkr&tjj |¡}| |¡Stjj |¡ |¡SdS)Nré)ÚdjangoÚVERSIONÚutilsÚformatsZsanitize_strftime_formatÚstrftimeZ datetime_safeZ new_datetime)ÚvalueZdatetime_formatÚformat©rú9/tmp/pip-unpacked-wheel-gh2cwdms/import_export/widgets.pyÚformat_datetime s rc@s$eZdZdZddd„Zddd„ZdS) Ú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. r©ÚselfrÚrowÚkwargsrrrÚcleans 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 ©rrÚobjrrrÚrender,sz Widget.render)N)N©Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrrrrs rc@s"eZdZdZdd„Zddd„ZdS)Ú NumberWidgetz cCs"t|tƒr| ¡}|dkp |dkS©NÚ)Ú isinstanceÚstrÚstrip)rrrrrÚis_empty;s zNumberWidget.is_emptyNcCs|S©NrrrrrrAszNumberWidget.render)N)r!r"r#r$r+rrrrrr%7sr%c@seZdZdZddd„ZdS)Ú FloatWidgetz. Widget for converting floats fields. NcKs| |¡rdSt|ƒSr,)r+ÚfloatrrrrrJs zFloatWidget.clean)N©r!r"r#r$rrrrrr-Esr-c@seZdZdZddd„ZdS)Ú IntegerWidgetz/ Widget for converting integer fields. NcKs| |¡rdStt|ƒƒSr,)r+ÚintrrrrrrUs zIntegerWidget.clean)Nr/rrrrr0Psr0c@seZdZdZddd„ZdS)Ú DecimalWidgetz/ Widget for converting decimal fields. NcKs| |¡rdStt|ƒƒSr,)r+rr rrrrr`s zDecimalWidget.clean)Nr/rrrrr2[sr2c@seZdZdZdS)Ú CharWidgetz, Widget for converting text fields. N)r!r"r#r$rrrrr3fsr3c@sVeZdZdZddddddgZdd d d d d gZdddddddgZddd„Zddd„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) Ú1éTÚtrueÚTRUEÚTrueÚ0rFÚfalseÚFALSEÚFalser'NÚnullZNULLÚnoneÚNONEÚNonecCs&||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)rBrCrrrrr™s zBooleanWidget.clean)N)N) r!r"r#r$rCrDrBrrrrrrr4ms  r4c@s.eZdZdZd dd„Zd dd„Zd dd„ZdS) Ú DateWidgetzV Widget for converting date fields. Takes optional ``format`` parameter. NcCs,|dkrtjsd}q"tj}n|f}||_dS)N)z%Y-%m-%d)rZDATE_INPUT_FORMATSr©rrrrrrÚ__init__¦s zDateWidget.__init__c Ksb|sdSt|tƒr|S|jD]8}zt ||¡ ¡WSttfk rRYqYqXqtdƒ‚dS)NzEnter a valid date.)r(rrrÚstrptimeÚ ValueErrorÚ TypeError©rrrrrrrrr°s   zDateWidget.cleancCs|sdSt||jdƒS©Nr'r)rrrrrrr¼szDateWidget.render)N)N)N©r!r"r#r$rGrrrrrrrEŸs rEc@s.eZdZdZd dd„Zd dd„Zd dd„ZdS) Ú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_FORMATSrrFrrrrGÊs zDateTimeWidget.__init__c Ks„d}|s dSt|tƒr|}n:|jD]2}zt ||¡}Wq"ttfk rRYq"Yq"Xq"|rxtjrtt  |¡rtt  |¡}|Stdƒ‚dS)NzEnter a valid date/time.) r(rrrHrIrJrÚUSE_TZrZis_naiveZ make_aware)rrrrÚdtZformat_rrrrÔs    zDateTimeWidget.cleancCs(|sdStjrt |¡}t||jdƒSrL)rrOrÚ localtimerrrrrrræs  zDateTimeWidget.render)N)N)NrMrrrrrNÂs rNc@s.eZdZdZd dd„Zd dd„Zd dd„ZdS) Ú TimeWidgetzV Widget for converting time fields. Takes optional ``format`` parameter. NcCs,|dkrtjsd}q"tj}n|f}||_dS)N)z%H:%M:%S)rZTIME_INPUT_FORMATSrrFrrrrGõs zTimeWidget.__init__c Ksb|sdSt|tƒr|S|jD]8}zt ||¡ ¡WSttfk rRYqYqXqtdƒ‚dS)NzEnter a valid time.)r(rrrrHrIrJrKrrrrÿs   zTimeWidget.cleancCs|sdS| |jd¡SrL)rrrrrrr szTimeWidget.render)N)N)NrMrrrrrRîs rRc@s$eZdZdZddd„Zddd„ZdS) ÚDurationWidgetz5 Widget for converting time duration fields. Nc Ks8|sdSz t|ƒWSttfk r2tdƒ‚YnXdS)NzEnter a valid duration.)r rIrJrrrrrs  zDurationWidget.cleancCs|dkr dSt|ƒSr&©r)rrrrrszDurationWidget.render)N)Nr rrrrrSs rScs6eZdZdZd ‡fdd„ Zd dd„Zd dd„Z‡ZS) ÚSimpleArrayWidgetzv Widget for an Array field. Can be used for Postgres' Array field. :param separator: Defaults to ``','`` Ncs |dkr d}||_tƒ ¡dS)Nú,)Ú separatorÚsuperrG)rrW©Ú __class__rrrG,szSimpleArrayWidget.__init__cKs|r| |j¡SgSr,)ÚsplitrWrrrrr2szSimpleArrayWidget.cleancCs|j dd„|Dƒ¡S)Ncss|]}t|ƒVqdSr,rT)Ú.0ÚvrrrÚ 6sz+SimpleArrayWidget.render..)rWÚjoinrrrrr5szSimpleArrayWidget.render)N)N)N©r!r"r#r$rGrrÚ __classcell__rrrYrrU%s rUcs,eZdZdZd‡fdd„ Zddd„Z‡ZS) Ú 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 | dd¡¡YSXdS)Nú'ú")rXrÚjsonÚloadsÚdecoderÚJSONDecodeErrorÚreplace©rrrrÚvalrYrrrBs   zJSONWidget.cleancCs|rt |¡SdSr,)reÚdumpsrrrrrJszJSONWidget.render)N)N)r!r"r#r$rrrarrrYrrb9srbcsBeZdZdZd ‡fdd„ Zdd„Zd‡fd d „ Zdd d „Z‡ZS)Ú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 customised 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$||_||_||_tƒjf|ŽdSr,)ÚmodelÚfieldÚuse_natural_foreign_keysrXrG)rrorprqrrYrrrGyszForeignKeyWidget.__init__cOs |jj ¡S)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"] ) )roÚobjectsÚall)rrrÚargsrrrrÚ get_querysetszForeignKeyWidget.get_querysetNc sVtƒ |¡}|rN|jr.t |¡}|jjj|ŽS|j||f|Žj f|j |iŽSndSdSr,) rXrrqrerfrorrZget_by_natural_keyruÚgetrprjrYrrr–s   zForeignKeyWidget.cleanc Cs||dkr dS|j d¡}|D]Z}z*|jrÓsz*ManyToManyWidget.clean..z%s__in) rorrr?r(r.r1r[rWÚfilterrp)rrrrÚidsrrrrÌs    ÿzManyToManyWidget.cleancs"‡fdd„| ¡Dƒ}ˆj |¡S)Ncsg|]}tt|ˆjƒƒ‘qSr)r rxrp)r\r©rrrr}Ùsz+ManyToManyWidget.render..)rsrWr_)rrrrrr€rrØszManyToManyWidget.render)NN)N)Nr`rrrYrr{¸s  r{)"rerrrÚdecimalrr Z django.confrZdjango.core.exceptionsrZ django.utilsrZdjango.utils.dateparser Zdjango.utils.encodingr r rrr%r-r0r2r3r4rErNrRrSrUrbrmr{rrrrÚs0          2#,#i