o i@sUddlmZddlZddlZddlZddlmZmZddl m Z ddl Z ddl m Z dZ edZd Zdad ed <ddddZddddZeGdddZdS)) annotationsN) dataclassfield)time) data_snapshot)DEFAULT_UPDATE_URL UpdatePriceswait_prices_updated_syncwait_prices_updated_asyncz genai-priceszXhttps://raw.githubusercontent.com/pydantic/genai-prices/refs/heads/main/prices/data.jsonzUpdatePrices | None_global_update_pricestimeout float | NonereturnboolcCstrt|SdS)zSynchronously wait for prices to be updated. Args: timeout: The maximum time to wait for prices to be updated. Defaults to None which waits indefinitely. Returns: True if prices were updated, False otherwise. F)r waitr r`/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/genai_prices/update_prices.pyr s r cstt|IdHS)zAsynchronously wait for prices to be updated. Args: timeout: The maximum time to wait for prices to be updated. Defaults to None which waits indefinitely. Returns: True if prices were updated, False otherwise. N)asyncio to_threadr rrrrr 's r c@seZdZUdZdZded< eZded< eddd Z d ed < ee j d Z d ed <ee j d Z d ed<edddZded<edddZded<ddd1ddZd2d3dd Zd!d"Zd#d$Zd4d'd(Zd5d*d+Zd,d-Zd6d/d0ZdS)7r zUpdate prices in the background using a daemon thread. Can be used either as a context manager or as a simple class, where you'll need to call start() and stop() manually. ifloatupdate_intervalstrurlcCstjdddS)N )r connect)httpxTimeoutrrrr>szUpdatePrices.)default_factoryz httpx.Timeoutrequest_timeoutzthreading.Event _stop_event_prices_updatedNF)defaultinitzthreading.Thread | None_threadzException | None_background_exc)rr bool | floatcCs|jdur tdtdurtd|a|j|jd|_tj|j ddd|_|j |r?|j |dur9dn|ddSdS) zStart the background task. Args: wait: Whether to wait for the prices to be updated before returning, if an int is passed wait for that many seconds, if `True` wait for 30 seconds. Nz,UpdatePrices background task already startedzWUpdatePrices global task already started, only one UpdatePrices can be active at a timeTzgenai_prices:update)targetdaemonnamer) r' RuntimeErrorr r$clearr#r( threadingThread_background_taskstartr)selfrrrrr3Es    zUpdatePrices.startr rrrcCs(|jj|d}|jr|j}d|_||S)zWait for the prices to be updated in the background task. Args: timeout: The maximum time to wait for the prices to be updated in seconds. rN)r$rr()r4r prices_updatedexcrrrr_s zUpdatePrices.waitcCsLdatd|jdur|j|jd|_|jr$|j}d|_|dS)zStop the background task.N)r rset_custom_snapshotr'r#setjoinr()r4r6rrrstopls    zUpdatePrices.stopcCs ||SN)r3)r4rrr __enter__{szUpdatePrices.__enter___argsobjectcGs |dSr;)r:)r4r=rrr__exit__s zUpdatePrices.__exit__Nonec Cstdz ||j|j|jrnqWn%ty?}z||_ |jt dt |j |WYd}~dSd}~wwtddS)Nz%Starting genai-prices background taskTz6Error updating genai-prices in the background (%s): %sz$genai-prices background task stopped) loggerinfo_update_pricesr$r8r#rr Exceptionr(errortype__name__)r4errrr2s   "zUpdatePrices._background_taskcCsLt}|}t|}|rtdt|j|ntd|t|dS)Nz1Successfully fetched %d providers in %.2f secondsz2Successfully fetched null snapshot in %.2f seconds)rfetchrArBlen providersrr7)r4r3snapshotintervalrrrrCs  zUpdatePrices._update_prices!data_snapshot.DataSnapshot | NonecCs>ddlm}tj|j|jd}|tj|j |j ddS)z9Fetches the latest provider data from the configured URL.r)datarT)from_auto_update) rOrgetrr"raise_for_statusr DataSnapshotproviders_schema validate_jsoncontent)r4rOrrrrrIs zUpdatePrices.fetch)rr)r;r rrr)r=r>)rr@)rrN)rG __module__ __qualname____doc__r__annotations__rrrr"r0Eventr#r$r'r(r3rr:r<r?r2rCrIrrrrr 3s(        r r;rY) __future__r _annotationsrloggingr0 dataclassesrrrrrQr__all__ getLoggerrArr r]r r r rrrrs