TKj(dZddlmZddlmZddlmZddlmZm Z ddl m Z m Z m Z ddZddZ dd Zdd Zdd Z dd Z dd Z ddZy)z0Layered matching engine with confidence scoring.) annotations)fuzz) CatalogIndex)deduplicate_offersexpand_category_synonyms) MatchResultOffer PartQueryc|jr0t|j|}|rtddt|dS|jr1t |j||}|rtddt|dS|j s|js |jr)t||\}}|rt|dt|dSt||\}}|rt|dt|dStd d gd S) uMatch a PartQuery against the catalog index using layered strategy. Priority: OE (0.95) → engine_code (0.80) → attributes (≤0.90) → fuzzy (≤0.50). First layer that produces results wins. gffffff?oeT) confidencemethodoffersmatchedg? engine_codeatrybutyfuzzybrakF) oe_numbers _match_by_oerrr_match_by_engine_codebrandmodelcategory_match_by_attributes _match_fuzzy)queryindexrconfs G/var/www/html/agent.samochodowy/src/agent_samochodowy/matcher/engine.pymatchr" s  e..6 )&1   &u'8'8%G $)&1   {{ekkU^^+E59  !)&1   u-LFD %f-    #fR OOcg}t}|D]S}|j|D]=}|j|vs|j|j|j |?U|S)z7Search for offers matching any of the given OE numbers.)set search_by_oeoffer_idaddappend)rrresultsseen_idsr offers r!rrAsfGH&''+ &E~~X- U^^,u% && Nr#c|j|}|sgS|jrV|Dcgc]G}|js|jj|jjk(sF|I}}|r|S|Scc}w)z;Search by engine code, optionally filtering by brand/model.)search_by_engine_coderlower)rrrrofiltereds r!rrMso ( ( 5F   {{%\!QWW]]_ HYHYH[5[A\\ O M]sA:5A:.A:ct|}|jr/|jjtfd|Dry|jjtfd|DS)zDCheck if an offer plausibly matches the query category via synonyms.c3&K|]}|v ywN).0syn cat_lowers r! z*_offer_matches_category..gs4Csi4Tc3&K|]}|v ywr4r5)r6r7 title_lowers r!r9z*_offer_matches_category..ks6csk!6r:)rrr/anytitle)r,query_categorysynonymsr8r<s @@r!_offer_matches_categoryrAasZ'7H ~~NN((* 484 4++##%K 6X6 66r#c|jDcgc]!}t|dk\s|j#}}|sgSg}|D]}|jxsdj}|dz|jzj}|D]@}||vr|j |]|s|j |s/|j ||Scc}w)zReturn offers where the query model appears in offer model/title. Supports prefix matching: query "R410" matches offer model "R", and query "LF" matches offer model "LF45".  )splitlenr/rr>r) startswith) query_modelrttokensrr0 offer_modelhaystacktoks r!_model_in_resultsrOns "-!2!2!4 DAA! aggi DF D  G  ww}"++- #%/668 Chq!s~~k:q!   N!Es C C ci}|r||d<|r||d<|jdd|i|}|r|S|jd}|j|jk7r|jdd|i|}|r|Sddl}|j d|}|ru|j d} | j|jk7rC| j|jk7r"t | dk\r|jdd| i|}|S) u'Try exact model, then base token, then series prefix. Examples: - "Sprinter w904" → try "Sprinter w904", then "Sprinter" - "R410" → try "R410", then "R410" (prefix matching in index handles R→R410) - "LF" → try "LF" (prefix matching in index handles LF→LF45/LF55) rrrrNz ^([A-Za-z]+)r5)search_by_attributesrFr/rer"grouprG) rrrrkwargsrbaserS prefix_matchprefixs r!_try_model_variantsrYs F w%z(U ' ' >e >v >F   ;;= D zz|u{{}$+++A$A&A M88OU3L##A& <<>U[[] *v||~/MRUV\R]abRb/U//GfGGF Mr#c|jrJ|jr>|jr2t||j|j|j}|r|dfS|jre|jrY|j |j|j}|r0|jr t |j|}|r|dfS|dfS|dfS|jr4|jr(t|d|j|j}|r|dfS|jrp|jrdt||j|jd}|r@|jr0|Dcgc]}t ||js|}}|r|dfSgdfS|d fSgdfScc}w) uAttribute-based matching with category validation. Category validation: offers whose category/title doesn't match the query category are filtered out. This prevents "rozrusznik Iveco" from returning "zbiornik paliwa Iveco". Confidence tiers: 0.90 — brand + model + category all confirmed 0.85 — brand + category confirmed, model found in results 0.65 — brand + model confirmed, category confirmed via post-filter 0.55 — brand + category (no model to check) or model + category 0.45 — brand + model only (no category in query) 0.35 — brand + category but model absent from results g?)rrg333333?gffffff?g?Ng?r?)rrrrYrRrOrA)rrr model_matchedr0r1s r!rrs]& {{u{{u~~$UEKKennU 4<  {{u~~++%+++W {{ 1%++v F  ($..t|#4<  {{u~~$UD%++u~~N 4<  {{u{{$UEKKdK ~~'-\!1HENN1[A\\#T>)3w4<  s7N]s F0Fc|j}|j|}|sgdfSt}|jr0t |jDchc]}|j }}|j xsdj }g}|D]}tj|j |jj } | dkrE|xr+|j xr|j j |k(} |jj |rtfd|Dnd} | xr| } |j|| | f|sgdfS|jdd|d d } |d d }|rd | dz d z dzz}t|d}nd| dz d z dzz}t|d }|ddDcgc]\}}}| c}}t|d fScc}wcc}}w)zFTS5 search + rapidfuzz re-scoring. If the query has a known brand and/or category, offers that match those attributes are scored higher and the confidence ceiling is raised. rrD(c3&K|]}|v ywr4r5)r6r7offer_title_lowers r!r9z_match_fuzzy..sO3c%66Or:Fc|d|dfS)NrCrQr5)xs r!z_match_fuzzy..sqtQqTlr#T)keyreverserrQrCr[<g333333?g333333?g333333?N)raw_text search_fuzzyr%rrr/rrtoken_set_ratior>r=r)sortminround)rr search_text fts_results cat_synonymssq_brandscoredr,ratiooffer_brand_matchoffer_cat_matchbrand_cat_confirmed best_ratiobest_confirmedr r0_r`s @r!rrs..K$$[1K 3w!UL ~~+CENN+STa T T{{ b'')G/1F ;$$[%6%6%8%++:K:K:MN 2: $V V 8I8I8Kw8V!KK--/S_#O,OOej/CO ue%89: ; 3w KK*DK91JAYq\NZ"_2T99 T* Z"_2T99 T* #BQZ ('!QA (% A*> >>KUJ )s G8 GN)rr rrreturnr)rz list[str]rrr{ list[Offer])rstrrr rrr{r|)r,r r?r}r{bool)rIr}rr|r{r|) rrr str | Nonerr}rrr{r|)rr rrr{ztuple[list[Offer], float])__doc__ __future__r rapidfuzzragent_samochodowy.matcher.indexr#agent_samochodowy.matcher.normalizerragent_samochodowy.modelsrr r r"rrrArOrYrrr5r#r!rs6"8\BB2Pj   & /;  ( 72% % *%36%BL%%P6 6)66r5? 5?)5?5?r#