7j% dZddlmZddlmZddlmZmZddlm Z ddl m Z ddl m Z e dZd d Zdd Z d dd Zy )aDiscovery upload via the session flow (Phase 2). Drives open-session -> per-category items -> complete/failed against the Central API. Upload is best-effort: any transport failure is logged and the local last.json snapshot remains the fallback. A category whose upload fails is reported in failed_categories so the backend records the session as partial (never silently dropped, never reconciled away). ) annotations)Any) ApiClientApiError)Config)load_agent_token) get_loggerzdiscovery.uploadct|tr$|Dcgc]}t|tr|nd|ic}St|tr|gSd|igScc}w)zENormalize a category value into a list of dict items for raw storage.value) isinstancelistdict)r items J/var/www/downloads.wolfpanel.net/public_html/agent/src/discovery/upload.py _coerce_itemsrsT%PUV 4.WdOCVV%w e  WsA ch|jDcic]\}}|t|c}}Scc}}w)zAMap a full-discovery inventory dict to category -> list-of-items.)itemsr) inventorycategoryr s rcategories_from_inventoryrs+BK//BS TxHmE* * TT Ts.Nc|xs t|}t|j} |j||d|}|d}g} g} |jD]-\} } |j|| | d|| j| / | s|j|dd|} n|j|| | d |} tjd ||| jd | S#tt t f$r }tjd|Yd}~yd}~wwxYw#t$r2}tjd| || j| Yd}~d}~wwxYw#t$r }tjd |Yd}~yd}~wwxYw) zRun the full session flow for one scan. Returns the finalize response. Never raises: returns None if the session could not even be opened. ) scan_typeexpected_categories session_idz!discovery session open failed: %sN)rrrz(discovery items upload failed for %s: %szall category uploads failed)rerror)rsucceeded_categoriesfailed_categoriesz1discovery %s scan finalized: session=%s status=%sstatuszdiscovery finalize failed: %s)rsortedkeysopen_discovery_sessionrKeyError TypeErrorlogwarningrupload_discovery_itemsappendfail_discoverycomplete_discoveryinfoget)config api_clientr categoriestokenexpectedopenedrexc succeededfailedrrresults rupload_inventoryr6$s  -%f-Ejoo'(H22#H Eu L) IF%++- $% $  - -)x%P    X & $..)4QRF  22",,5)/  F  ?   JJx   O h * 7= $ KKBHc R MM( # # $6  3S9sHC 'D AE D4DD E (E  E E>E99E>)r rreturnzlist[dict[str, Any]])rzdict[str, Any]r7dict[str, list[dict[str, Any]]])N) r,rr-rrstrr.r8r/z str | Noner7zdict[str, Any] | None)__doc__ __future__rtypingrr-rrr,ridentityrloggerr r$rrr6rrAsv#*%#$U = ===0 =  =  =r@