7 dZddlZddlZddlZddlZddlmZddlmZddl m Z dZ de de d e d e d ejf d Zde de d e d e d ee f d Zde de d e d e de d df dZy)zRProvides Logic for Fetching and Storing Discovery Documents from an on-disc cache.N)Optional)logging) stringutilzapi_discovery.json cache_root discovery_urlapi_name api_versionreturnctj|jdj}t j ||||t S)z:Returns the cache file name for the given api and version.utf-8)hashlibsha1encode hexdigestpathlibPath_DISCOVERY_CACHE_FILE)rrrr url_hashs ;platform/bq/discovery_documents/discovery_document_cache.py_get_cache_file_namersF \\-..w7 8 B B D(   cjt||||}tjj|st j d|y t |d5}|j}dddjdS#1swYxYw#t$r!}t jd||Yd}~yd}~wwxYw)aLoads a discovery document from the on-disc cache using key `api` and `version`. Args: cache_root: [str], a directory where all cache files are stored. discovery_url: [str], URL where the discovery document was fetched from. api_name: [str], Name of api `discovery_document` to be saved. api_version: [str], Version of document to get Returns: Discovery document as a dict. None if any errors occur during loading, or parsing the document zDiscovery doc not in cache. %sNrbr z'Error loading discovery document %s: %s) rospathisfilerinfoopenreaddecode Exceptionwarning)rrrr filefcontentses rget_from_cacher'!s j-; O$   LL148  dD Qh  ??7 ##    OO=tQG s0 BA<#B<BB B2B--B2discovery_documentc$t||||}|j}|jry|jddt j |5}t j|dz }|jd5} | jtj|d| jtj| jddd|j!|dddy#1swY#xYw#1swYyxYw)aSaves a discovery document to the on-disc cache with key `api` and `version`. Args: cache_root: [str], a directory where all cache files are stored. discovery_url: [str], URL where the discovery document was fetched from. api_name: [str], Name of api `discovery_document` to be saved. api_version: [str], Version of `discovery_document`. discovery_document: [str]. Discovery document as a json string. Raises: OSError: If an error occurs when the file is written. NT)parentsexist_ok)dirztmp.jsonwbutf8)rparentexistsmkdirtempfileTemporaryDirectoryrrrwriter ensure_binaryflushrfsyncfilenorename) rrrr r(r# directorytmpdirtemp_file_pathr$s r save_to_cacher=As( j-; O$kk) [[]  //$/.""y1V\\&)J6N   T "aggj&&'96BCggihhqxxz #$21 " "21s%*D?AC:D:D ?DD)__doc__r rrr2typingrabslrpyglibrrstrrrr'r=rrrDsX -  $' 36 EH  \\ $'36EH c]@. . . .  .  .  . r