U hu @sXdZddlmZddlZddlmZddddd Zdddd d Zddd d dZdS)z Utilities for working with URLs.) annotationsN)parsestrNone)partreturncCstd|std|dS)Nz[a-zA-Z0-9_-]*z,"{}" is outside the restricted character set)re fullmatch ValueErrorformat)rr T/tmp/pip-unpacked-wheel-doshhd5e/google/generativeai/notebook/sheets_sanitize_url.py_validate_url_parts rcCs6t|D]"\}}t||D] }t|q"qdS)N)rparse_qsitemsr)rkeyvaluesvaluer r r _validate_url_query_or_fragmentsr)urlrc Cs6t|}|jdkr$td|j|jdkr>td|jz|jdD] }t|qLWn4tk r}ztd|j|W5d}~XYnX|j rtd|j zt |j Wn4tk r}ztd |j |W5d}~XYnXzt |j Wn6tk r0}ztd |j |W5d}~XYnX|S) 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 "{}") rurlparseschemer r netlocpathsplitrparamsrqueryfragment)rZ parse_resultrexcr r r sanitize_sheets_url$sZ      r!) __doc__ __future__rrurllibrrrr!r r r r s