U 9if@sdZddlmZddlZddlZddlmZddlmZm Z ddl m Z e dZ e dd d d d Ze ddd dddZd ddddddZd!ddddddddZd"ddddddddZdS)#z)Sitemap status checking and resubmission.) annotationsN) HttpError) RateLimiterretry_on_api_error) SitemapStatus seo_optimizerstrz list[dict])site_urlreturncCs |j|d}|dgS)z'List all submitted sitemaps for a site.)siteUrlZsitemap)sitemapslistexecuteget)servicer responser,/opt/web_system/seo_auto/src/api/sitemaps.py_list_sitemapssrNone)r sitemap_urlr cCs|j||ddS)zSubmit (or resubmit) a sitemap.)r ZfeedpathN)r submitr)rr rrrr_submit_sitemapsrzRateLimiter | Nonezlist[SitemapStatus])r rate_limiterr c Cs|r||t||}g}|D]n}t|dd|dd|dd|dd|ddt|dd t|d d |d gd }||q td t|||S)zjGet status of all submitted sitemaps for a site. Returns: List of SitemapStatus objects. pathZ lastSubmittedZlastDownloadedZ isPendingFZisSitemapsIndexwarningsrerrorscontents)rlast_submittedlast_downloaded is_pendingis_sitemaps_indexrrrzFound %d sitemaps for %s) acquirerrrintappendloggerinfolen)rr rZ raw_sitemapsstatusesZsmstatusrrrget_sitemap_statusess,          r+r$bool)r r max_retriesrr c Cstd|dD]}z0|r"||t|||td||WdStk r}z,td||||||kr|td|W5d}~XYqXqdS)zlResubmit a sitemap with retries. Returns: True if resubmission succeeded, False otherwise. z1Sitemap resubmitted successfully: %s (attempt %d)Tz3Sitemap resubmission failed: %s (attempt %d/%d): %sNF) ranger#rr&r'rwarningtimesleep)rr rr.rattempterrrresubmit_sitemap@s,   "r7z list[str])r known_sitemap_urlsr.rr c Cst|||}dd|D}|D](}||krtd|t|||||q|D]2}|jdkrLtd|j|jt|||j||qLt|||} | S)aCheck sitemap statuses and resubmit those with errors. Args: service: Webmasters v3 service. site_url: GSC property URL. known_sitemap_urls: Expected sitemap URLs from config. max_retries: Max resubmission attempts per sitemap. rate_limiter: Optional rate limiter. Returns: Updated list of SitemapStatus after resubmission attempts. cSsh|] }|jqSr)r).0srrr wsz%check_and_resubmit..zSubmitting new sitemap: %sru,Sitemap has errors (%d): %s — resubmitting)r+r&r'r7rr2r) rr r8r.rr)Zsubmitted_pathsurlr*Zupdated_statusesrrrcheck_and_resubmitbs,    r=)N)r,N)r,N)__doc__ __future__rloggingr3googleapiclient.errorsrsrc.api.rate_limiterrrsrc.models.data_modelsr getLoggerr&rrr+r7r=rrrrs&    '&