h-i @dZddlmZddlZddlmZddZddZd dZy) z Utilities for working with URLs.) annotationsN)parsecdtjd|stdj|y)Nz[a-zA-Z0-9_-]*z,"{}" is outside the restricted character set)re fullmatch ValueErrorformat)parts oD:\DhanshreeandTeamAI\Github\ai_code\venv\Lib\site-packages\google/generativeai/notebook/sheets_sanitize_url.py_validate_url_partr s- <<($ /GNNtTUU 0ctj|jD]"\}}t||D] }t|$y)N)rparse_qsitemsr )r keyvaluesvalues r _validate_url_query_or_fragmentrs<~~d+113 V3E u %4r cPtj|}|jdk7r$tdj |j|j dvr$tdj |j  |j jdD] }t| |jr$tdj |j t|j t|j|S#t$r*}tdj |j |d}~wwxYw#t$r*}td j |j|d}~wwxYw#t$r*}td j |j|d}~wwxYw) aSanitize a Sheets URL. Run some saftey checks to check whether `url` is a Sheets URL. This is not a general-purpose URL sanitizer. Rather, it makes use of the fact that we know the URL has to be for Sheets so we can make a few assumptions about (e.g. the domain). Args: url: The url to sanitize. Returns: The sanitized url. Raises: ValueError: If `url` does not match the expected restrictions for a Sheets URL. httpsz/Scheme for Sheets url must be "https", got "{}")zdocs.google.comzsheets.googleapis.comz9Domain for Sheets url must be "docs.google.com", got "{}"/z%Invalid path for Sheets url, got "{}"Nz(Params component must be empty, got "{}"z&Invalid query for Sheets url, got "{}"z)Invalid fragment for Sheets url, got "{}") rurlparseschemerr netlocpathsplitr paramsrqueryfragment)url parse_resultrexcs r sanitize_sheets_urlr#"s$>>#&Lg% = D D\EXEX Y  "NN G N N|ObOb c  e$))//4H x (5 CJJ J/ e@GG HYHYZ[adde  4 ; ;L >|?T?T U  sH<+DD</E2 D9%D44D9< E/%E**E/2 F%;%F  F%)r strreturnNone)r r$r%r$) __doc__ __future__rrurllibrr rr#r r r+s#'" V &7r