cQj NdZddlmZddlmZddlmZmZddlm Z m Z ddl m Z m Z mZmZddlmZmZmZed ZGd d Ze Gd d eZe GddeZe GddeZe GddeZe GddeZGddeeeZy)zResult classes.)ABC)Counter)CallableMapping) dataclassfield)AnyClassVarGenericTypeVar)_normalize_date_normalize_text_normalize_urlTc feZdZUdZeeeeeeeeeed Zee e e e ge ffe d<de de ddfdZy) BaseResultz=Base class for all results. Contains normalization functions.) titlebodyhrefurl thumbnailimagedateauthor publisherinfo _normalizersnamevaluereturnNc|r%|jj|x}r||}tj|||y)zHOverride setattr to apply normalization functions to certain attributes.N)rgetobject __setattr__)selfrr normalizers Q/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/ddgs/results.pyr%zBaseResult.__setattr__s< D$5$5$9$9$$??j?u%E4u-)__name__ __module__ __qualname____doc__rrrrr rstrrr __annotations__r%r)r(rrshG!#!$ BL(73#(<#<=> ..C.D.r)rc<eZdZUdZdZeed<dZeed<dZeed<y) TextResultzText search result.rrrN) r*r+r,r-rr.r/rrr0r)r(r2r2%s E3OD#ND#Nr)r2cteZdZUdZdZeed<dZeed<dZeed<dZ eed<dZ eed<dZ eed<dZ eed <y ) ImagesResultzImage search result.r3rrrrheightwidthsourceN) r*r+r,r-rr.r/rrrr6r7r8r0r)r(r5r5.sGE3OE3OIsCMFCE3OFCr)r5cfeZdZUdZdZeed<dZeed<dZeed<dZ eed<dZ eed<dZ eed<y ) NewsResultzNews search result.r3rrrrrr8N) r*r+r,r-rr.r/rrrrr8r0r)r(r:r:;s<D#NE3OD#NCME3OFCr)r:ceZdZUdZdZeed<dZeed<dZeed<dZ eed<dZ eed<dZ eed<dZ eed <e e Zeeefed <dZeed <dZeed <dZeed<e e Zeeefed<dZeed<y) VideosResultzVideo search result.r3rcontent descriptionduration embed_html embed_url image_token)default_factoryimagesprovider publishedr statisticsuploaderN)r*r+r,r-rr.r/r=r>r?r@rArBrdictrDrErFrrGrHr0r)r(r<r<GsE3OGSKHcJIsK"48FDcN8HcIsIs!&t!E3OFCIsD#NCMIsr)rKceZdZdZdeeddfdZdedefdZde fdZ deddfd Z d e eddfd Z de eeeffd Zy) ResultsAggregatorzAggregates incoming results. Items are deduplicated by `cache_field`. Append just increments a counter; `extract_results` returns items sorted by descending frequency. cache_fieldsr!Ncn|s d}t|t||_t|_i|_y)Nz)At least one cache_field must be provided) ValueErrorsetrNr_counter_cache)r&rNmsgs r(__init__zResultsAggregator.__init__ms1=CS/ ! -&-i $& r)itemc|jD]*}||jvst|j|cSd|d|j}t|)NzItem z has none of the cache fields )__dict__rNr.AttributeError)r&rVkeyrTs r(_get_keyzResultsAggregator._get_keyus^== /Cd'''4==-.. /dX;D|j|}||jvsXt|jj ddt|j|jj ddkDr||j|<|j |xxdz cc<y)zAdd an item to the cache. Register an occurrence of `item`. First time we see its key, we store the item; every time we bump the counter. rr3r N)r[rSr]rXr#rR)r&rVrZs r(appendzResultsAggregator.appends mmD! dkk !S):):62)F%G# KK  % % ) )&" 5K &  $DKK  car)itemsc4|D]}|j|y)z!Add a list of items to the cache.N)r`)r&rarVs r(extendzResultsAggregator.extends D KK  r)c|jjDcgc]\}}|j|j c}}Scc}}w)zXReturn a list of items, sorted by descending frequency. Each item is returned as a dict.)rR most_commonrSrX)r&rZ_s r( extract_dictszResultsAggregator.extract_dictss68< 8Q8Q8STfc1 C ))TTTs#A)r*r+r,r-rQr.rUrr[intr^r`listrcrIr rgr0r)r(rMrMfs 'SX'$'"Q"3"    1    DG UtDcN3Ur)rMN)r-abcr collectionsrcollections.abcrr dataclassesrrtypingr r r r utilsrrrrrr2r5r:r<rKrMr0r)r(rps-(22CC CL...    :      : $ * .UWQZ.Ur)