o hs"@sddlmZddlmZddlmZddlmZddlm Z m Z ddl m Z ddlmZzddlmZWn ey?d ZYnwGd d d Zd d Zd S))ObjectDoesNotExist) force_str)capfirst) DEFAULT_ALIAS) NotHandled SpatialError)log)haystack_get_model)distanceNc@seZdZdZddZddZddZdd Zd d Zd d Z e e Z ddZ ddZ e e e ZddZddZe eeZddZddZe eeZddZe eZddZe eZddZd d!Zd"d#Zd$d%Zd&d'Zd(S)) SearchResultaB A single search result. The actual object is loaded lazily by accessing object; until then this object only stores the model, pk, and score. Note that iterating over SearchResults and getting the object for each result will do O(N) database queries, which may not fit your needs for performance. cKs|||_|_||_||_d|_d|_d|_g|_|dd|_ |dd|_ d|_ | |_ |D]\}}||jvrG||j|<|j|q3dS)N_point_of_origin _distance) app_label model_namepkscore_object_model _verbose_name_additional_fieldspopr r stored_fields_get_logritems__dict__append)selfrrrrkwargskeyvaluer A/var/www/html/env/lib/python3.10/site-packages/haystack/models.py__init__s"    zSearchResult.__init__cCs tdS)Nhaystack)logging getLoggerrr r r!r/ zSearchResult._get_logcCsd|j|j|jfS)Nz)rrrr&r r r!__repr__2s zSearchResult.__repr__cCs t|SN)rr(r&r r r!__str__9s zSearchResult.__str__cCs|dkrt|j|dS)N__getnewargs__)AttributeErrorrget)rattrr r r! __getattr__<szSearchResult.__getattr__cCs ddlm}|t|jSNr connections)r#r2rget_unified_index get_indexmodel)rr2r r r!_get_searchindexBs zSearchResult._get_searchindexcCs|jdurZ|jdur|jd|dSz1z|jj|jd|_WW|jStyD|j d|j |j |jj j|jd|_YW|jSwt yY|jd|d|_Y|jSw|jS)N/Model could not be found for SearchResult '%s'.)rz)Model '%s.%s' not handled by the routers.zr))rr?r r r! _set_modelsr'zSearchResult._set_modelcCsddlm}|jdurUtdurtd|jstdt||jds+td|jd|jdj\}}t||jd}|durAdS|j\}}|t ||f||fj d|_|jS) Nr)DistancezThe backend doesn't have 'DISTANCE_AVAILABLE' enabled & the 'geopy' library could not be imported, so distance information is not available.z$The original point is not available.fieldz[The field '%s' was not included in search results, so the distance could not be calculated.point)km) django.contrib.gis.measurerEr geopy_distancerr hasattrcoordsgetattrr rH)rrEpo_lngpo_latlocation_fieldlf_lnglf_latr r r! _get_distancexs.   zSearchResult._get_distancecCr>r))r )rdistr r r! _set_distancer'zSearchResult._set_distancecC.|jdur|jd|dStt|jjjSNr7)r5rr8rr_meta verbose_namer&r r r!_get_verbose_name zSearchResult._get_verbose_namecCrVrW)r5rr8rrrYverbose_name_pluralr&r r r!_get_verbose_name_pluralr\z%SearchResult._get_verbose_name_pluralcCs(|jdur|jd|dSt|jjS)z9Returns the content type for the result's model instance.Nr7rX)r5rr8strrYr&r r r! content_types  zSearchResult.content_typecCs"i}|jD] }t||||<q|S)z Returns a dictionary of all of the fields from the raw result. Useful for serializing results. Only returns what was seen from the search engine, so it may have extra fields Haystack's indexes aren't aware of. )rrM)radditional_fields fieldnamer r r!get_additional_fieldss z"SearchResult.get_additional_fieldscCs|jdur>ddlm}z |t|j}Wn ty"iYSwi|_|j D]\}}|j dur=t ||d|j|<q+|jS)z Returns a dictionary of all of the stored fields from the SearchIndex. Useful for serializing results. Only returns the fields Haystack's indexes are aware of as being 'stored'. Nrr1TrX) _stored_fieldsr#r2rr3r4r5rfieldsrstoredrM)rr2indexrbrFr r r!get_stored_fieldss    zSearchResult.get_stored_fieldscCs|j}|d=|S)zp Returns a dictionary representing the ``SearchResult`` in order to make it pickleable. r)rcopy)rret_dictr r r! __getstate__s zSearchResult.__getstate__cCs|j|||_dS)zU Updates the object's attributes according to data passed by pickle. N)rupdaterr)r data_dictr r r! __setstate__s zSearchResult.__setstate__N)__name__ __module__ __qualname____doc__r"rr(r*r/r6propertyr9r=rAobjectrCrDr5rSrUr r[rZr^r]r`rcrhrkrnr r r r!r s6   %  r cOs.ddlm}|D] }|}|q dSr0)r#r2allr3reset)senderargsrr2connuir r r!reload_indexess   r{)django.core.exceptionsrdjango.utils.encodingrdjango.utils.textrhaystack.constantsrhaystack.exceptionsrrhaystack.utilsrr$haystack.utils.app_loadingr geopyr rJ ImportErrorr r{r r r r!s        h