U >Ke)@s2GdddZGdddeZGdddeZdS)c@s"eZdZdZdddZddZdS)BaseInstanceLoaderz: Base abstract implementation of instance loader. NcCs||_||_dSN)resourcedataset)selfrrrB/tmp/pip-unpacked-wheel-v_jw5did/import_export/instance_loaders.py__init__szBaseInstanceLoader.__init__cCstdSr)NotImplementedErrorrrowrrr get_instance szBaseInstanceLoader.get_instance)N)__name__ __module__ __qualname____doc__rr rrrrrs rc@s eZdZdZddZddZdS)ModelInstanceLoaderzc Instance loader for Django model. Lookup for model instance by ``import_id_fields``. cCs |jSr)r get_querysetrrrrrsz ModelInstanceLoader.get_querysetcCstzPi}|jD] }|jj|}||||j<q|rH|jf|WSWdSWn|jjjj k rnYdSXdSr) rget_import_id_fieldsfieldsclean attributergetZ_metamodelZ DoesNotExist)rr paramskeyfieldrrrr s  z ModelInstanceLoader.get_instanceN)r rrrrr rrrrrsrcs(eZdZdZfddZddZZS)CachedInstanceLoaderz Loads all possible model instances in dataset avoid hitting database for every ``get_instance`` call. This instance loader work only when there is one ``import_id_fields`` field. cstj||jd}jj|_i_jjrjj jjdkrfddjjD} j fdjj |i}fdd|D_dS)Ncsg|]}j|qSr)pk_fieldr).0r rrr 9sz1CachedInstanceLoader.__init__..z%s__incsi|]}j||qSr)r get_value)r instancerrr <s z1CachedInstanceLoader.__init__..) superrrrrr all_instancesrdictZ column_namerfilterr)rargskwargsZ pk_field_nameidsqs __class__rrr/s zCachedInstanceLoader.__init__cCs|jr|j|j|SdSr)r&rrrr rrrr @sz!CachedInstanceLoader.get_instance)r rrrrr __classcell__rrr-rr&s rN)rrrrrrrs