j(dZddlmZddlZddlmZmZmZGddeZ GddZ ejZ dd Z dd Zdd Zdd Zdd ZddZddZ ddZddZddZddZgdZddZy) usW20E5 — runtime image call budget primitive. A thread-safe counter installed for the duration of a background run that acts as the authoritative cap at every gpt-image-2 / Gemini image / fal provider call site. ``reserve()`` raises ``ImageCallBudgetExceeded`` BEFORE the network request is allowed, so cap=0 / exhausted runs never reach the provider. The primitive is stdlib-only; no ``app.*`` imports and no provider SDK imports (verified by an AST test). Call sites import ``reserve_current_call`` which is a no-op when no budget is installed on the current thread, keeping read-only paths and existing tests unaffected. ) annotationsN)AnyDictOptionalc$eZdZdZdfd ZxZS)ImageCallBudgetExceededzRaised by ``ImageCallBudget.reserve`` when the cap is reached. The exception is raised *before* the provider/network call so the caller can abort without touching the SDK. ct||_t||_t||_t |d|jd|jd|jy)Nu#image_call_budget exceeded — cap=z, used=z, denied source=)intcapusedstrsourcesuper__init__)selfr r r __class__s P/Users/manta/Documents/Projects/TheRoad-I1/backend/app/core/image_call_budget.pyrz ImageCallBudgetExceeded.__init__sZs8I &k  1$((7499+N![[O - )r r r r rr returnNone)__name__ __module__ __qualname____doc__r __classcell__)rs@rrrs   rrc,eZdZdZdZddZddZd dZy) ImageCallBudgetzCounter with a hard cap on image-provider calls. Counts are managed under a re-entrant lock; ``reserve`` and ``snapshot`` are safe to call from multiple threads against the same instance. )_cap_used_denied_lockct|}|dkrtd|d||_d|_d|_t j |_y)Nrzcap must be >= 0 (got ))r ValueErrorrrr threadingRLockr!)rr cap_ints rrzImageCallBudget.__init__.sJc( Q;5gYa@A A   __& rc|j5|j|jk\r7|xjdz c_t |j|j||xjdz c_dddy#1swYyxYw)a9Reserve one unit of the budget for a provider call. Must be called *before* the actual provider/network call. Raises :class:`ImageCallBudgetExceeded` and increments the denied counter when the cap has already been reached; the used counter is left unchanged in that case. )r r rN)r!rrr r)rrs rreservezImageCallBudget.reserve7s^ZZzzTYY& ! -  6 JJ!OJ ZZs A&A<<Bc|j5|j|j|j|j|jz dcdddS#1swYyxYw)z6Return a JSON-serialisable copy of the current counts.)r r denied remainingN)r!rrr )rs rsnapshotzImageCallBudget.snapshotGs= ZZyy ,,!YY3 ZZs ._wrappeds"8RA$A&AArrLrrMrrr)rG)rKrUrTs` @rbind_current_stop_checkrWs&'HB OrcDttdfd }|S)u<Return a callable that re-installs *this thread's* current budget **and stop check** when invoked on another thread. ★정지 확인을 예산과 **같이** 실어 나른다. 둘 다 스레드 지역이고 둘 다 `reserve_current_call` 에서 쓰이므로, 하나만 나르면 팬아웃 안에서 정지가 통째로 안 들린다. 기존 호출부를 안 고쳐도 되도록 이 함수가 둘을 다 잡는다. Intended use:: from concurrent.futures import ThreadPoolExecutor with ThreadPoolExecutor(max_workers=4) as pool: pool.submit(bind_current_budget(_worker), arg1, arg2) The budget is captured at ``bind_current_budget`` call time, not at callable invocation time — so the parent may safely clear its own install before the worker actually runs. c0ttg|i|SrD)rQrO)rLrMrT captured_stoprKs rrUz%bind_current_budget.._wrappeds)" ?Hb ;? CI  rrV)r=rG)rKrUrTrZs` @@rbind_current_budgetr[s!$"#H*,M Or) rrr7r9r=rIrOr[rBrErGrQrWc<t}|y|jSrD)r=r.r6s rcurrent_snapshotr] s  !F ~ ?? r)r5rrr)rr)rOptional[ImageCallBudget])rA Optional[Any]rr)rr_r/) r5r^rKrrLrrMrrr) r@r_rKrrLrrMrrr)rKrrr)rzOptional[Dict[str, Any]])r __future__rr%typingrrr RuntimeErrorrrlocalr4r7r9r=rBrErGrIrOrQrWr[__all__r]r1rrres #&& l "**j   + /"H% %% % % %  %2 % : &r