U 1iX@sdZddlmZddlZddlmZddlmZmZm Z ddl m Z e dZ e ddd d d d Zdddd dddddZdddd dddddZdS)z)URL Inspection API with quota management.) annotationsN) HttpError) QuotaTracker RateLimiterretry_on_api_error) IndexStatus seo_optimizerstrdict)site_urlinspection_urlreturncCs.||d}|j|d}|diS)z(Execute a single URL inspection request.)Z inspectionUrlsiteUrl)bodyZinspectionResult)Z urlInspectionindexinspectexecuteget)servicer r request_bodyresponser2/opt/web_system/seo_auto/src/api/url_inspection.py _inspect_urls rzQuotaTracker | NonezRateLimiter | NonezIndexStatus | None)r url quota_tracker rate_limiterr c Cs|r|stddS|r(||zt|||}WnFtk r~}z(td||t|dt|dWYSd}~XYnX|di}|dd}|d d} |d d} |d d} |d d} |d d} |dko| dk}t||| | | | | |dS)aAInspect a single URL's index status. Args: service: Search Console v1 service. site_url: GSC property URL. url: URL to inspect. quota_tracker: Optional daily quota tracker. rate_limiter: Optional rate limiter. Returns: IndexStatus or None if quota exhausted. z$URL Inspection daily quota exhaustedNz URL Inspection failed for %s: %sERROR)rverdictcoverage_stateZindexStatusResultrZ coverageStateZ indexingStateZpageFetchStateZrobotsTxtStateZ lastCrawlTimeZPASSZINDEXING_ALLOWED)rrrindexing_statepage_fetch_staterobots_txt_statelast_crawl_time is_indexed) acquireloggerwarningrrerrorrr r)rr rrrresulteZ index_resultrrr!r"r#r$r%rrr inspect_urls8   (       r,z list[str]zlist[IndexStatus])r urlsrrr c Csg}t|}t|dD]p\}}t|||||d} | dkrJtd||q|| |ddksh||kr|rr|jnd} td||| qtd d |D} td || t|t||S) auInspect multiple URLs with progress logging. Args: service: Search Console v1 service. site_url: GSC property URL. urls: List of URLs to inspect. quota_tracker: Optional daily quota tracker. rate_limiter: Optional rate limiter. Returns: List of IndexStatus results (may be shorter than urls if quota runs out). )rrNu6Stopping batch inspection at %d/%d — quota exhausted rzN/Az4URL Inspection progress: %d/%d (quota remaining: %s)css|]}|jrdVqdS)r.N)r%).0rrrr }sz batch_inspect..z)NN)NN)__doc__ __future__rlogginggoogleapiclient.errorsrsrc.api.rate_limiterrrrsrc.models.data_modelsr getLoggerr'rr,r>rrrrs    ;