o &@sdZddlmZddlmZddlmZddlZddlZddlZddlm Z ddlm Z ddlm Z dd l m Z ddlZdd lmZddlZd d Zdd dZGddde jZGddde jZdS)znA persistent cache implementation using sqlite3. See the persistent_cache module for a detailed description. )absolute_import)division)unicode_literalsN) exceptions)metadata_table)persistent_cache_base)files)rangecCs dj|dS)zReturns a field reference name. Args: column: The field column number counting from 0. Returns: A field reference name. z f{column}column)formatr r C/tmp/google-cloud-sdk/lib/googlecloudsdk/core/cache/sqlite_cache.py _FieldRef's rcCsg}|rCtt|D]8}||}|durq t|tjr6|dddddd}|djt||d q |d jt||d q |sGd Sd d |S)aZReturns a WHERE clause for the row template. Column string matching supports * and ? match ops. Args: row_template: A template row tuple. A column value None means match all values for this column. A None value for row means all rows. Returns: A WHERE clause for the row template or the empty string if there is no none. N*%._"z""z{field} LIKE "{pattern}")fieldpatternz{field} = {term})rtermz WHERE z AND ) r len isinstancesix string_typesreplaceappendr rjoin) row_templatetermsindexrrr r r_Where3s"    r#csReZdZdZ  dfdd ZddZd d Zd d ZdddZdddZ Z S)_TableaXA persistent cache table. Attributes: name: The table name. deleted: Table was deleted if True. modified: Table modify timestamp. timeout: Tables older than timeout are invalid. _cache: The parent cache object. _fields: The f1,... fields name string. _values: The ?,... parameter replacement string for INSERT. rFc sd|_tt|j|||||||d|r|jj|dddt|D|_ ddg||_ d|_ |jj r@||jj |<dSdS)N)columnskeystimeoutmodified restricted, cSg|]}t|qSr r.0ir r r fz#_Table.__init__..?F)_rowssuperr$__init___cache _restrictedaddrr _fields_valuesdeleted _metadata_tables)selfcachenamer&r'r(r)r* __class__r rr6^sz_Table.__init__cCsJ||jjdj|jd|jj|jj |jfgd|_ dS)zDeletes the table.zDROP TABLE "{table}"tableTN) Invalidater7cursorexecuter rA_dbcommitr= DeleteRowsr<r?r r rDeletems   z _Table.Deletec Csz|jr;d|_|jrd|_|jj|jfg|jj|j=dS|jjtj j |j|j |j |j |j|j|jjdgdSdS)z#Commits changed/deleted table data.F)rAr&r'r(r)r*versionN)changedr<r7r=rKrAr>AddRowsrMetadataRowr&r'r(r)r*rNrLr r r_Commitws$z_Table._CommitcCs<|||jjdj|j|j|jd||jj dS)z#Adds each row in rows to the table.z=INSERT OR REPLACE INTO "{table}" ({fields}) VALUES ({values}))rEfieldsvaluesN) _CheckRowsr7rG executemanyr rAr:r;rIrJ)r?rowsr r rrPs  z_Table.AddRowsNcCs`|r|||D]}|jjdj|jt|dq n |jjdj|jd|jjdS)z@Deletes each row in the table matching any of the row_templates.zDELETE FROM "{table}"{where})rEwherezDELETE FROM "{table}" WHERE 1rDN) _CheckRowTemplatesr7rGrHr rAr#rIrJ)r? row_templatestemplater r rrKs   z_Table.DeleteRowscCsj|dur ||g|s|js|jstd|jj|j|jj dj|j |jt |d|jj S)z?Returns the list of rows that match row_template, None for all.Nz"[{}] cache table [{}] has expired.z%SELECT {fields} FROM "{table}"{where})rTrErY) rZr*r)rCacheTableExpiredr r7rArGrHr:r#fetchall)r?r ignore_expirationr r rSelects   z _Table.Select)r%r%rrFN)NF) __name__ __module__ __qualname____doc__r6rMrSrPrKr` __classcell__r r rBrr$Qs    r$csPeZdZdZdZdfdd ZddZd d Zd d Zdd dZ ddZ Z S)CacheaA persistent cache object. Attributes: cursor: The _db operations cursor. name: The db path name. Created/removed by this object. May be a file or directory. In this implementation its a file. timeout: The default table timeout. version: A caller defined version string that must match the version string stored when the persistent object was created. _db: The db connection. _metadata: The metadata restricted _Table. _persistent: True if the persistent object has been committed at least once. _restricted: The set of restricted table names. _start: The cache instance start time. _tables: The map of open table objects. sSQLite format 3TNcs4tt|jt||||dd|_z0t|}|t|j }||j kr-t d |j Wdn1s7wYd|_Wn%tjyU|sSt d |j Yntjyet d |j wt||_|j|_tdg|_i|_d|_t|_z|WdSt jy|jddw)N)creater(rNFz[{}] is not a persistent cache.Tz Persistent cache [{}] not found.__lock__rJ)r5rgr6r$ _persistentrBinaryFileReaderreadr_EXPECTED_MAGICr CacheInvalidr rAMissingFileError CacheNotFoundErrorsqlite3connectrIrGsetr8r>r=rNow_startInitializeMetadataClose)r?rArhr(rNf actual_magicrBr rr6sJ             zCache.__init__c CsNz t|jWdSty&}z|jtjtjfvrWYd}~dSd}~ww)z.Permanently deletes the persistent cache file.N)osremoverAOSErrorerrnoENOENTEISDIR)r?er r r_DeleteCacheFileszCache._DeleteCacheFilecCs|jdd|dS)z4Closes and permanently deletes the persistent cache.FrjN)ryrrLr r rrMs  z Cache.DeletecCsHdd|jDD]}|q |jjr|j|jd|_dS)z(Commits all operations up to this point.cSsg|]}|jr|qSr )rO)r/xr r rr1sz Cache.Commit..TN)r>rUrSr=rOrIrJrk)r?rEr r rCommits     z Cache.CommitcCsX|jr&|r ||`|jd|_tdd|_|s(|js*|dSdSdSdS)z~Closes the cache, optionally committing any changes. Args: commit: Commits any changes before closing if True. N) rIrrGclosegccollectr>rkr)r?rJr r rrys     z Cache.ClosecCslddt|D}ddt|pdD}|djd|ddjd|d }|jd j||d d S) z-sqlite3 implementation specific _CreateTable.cSr,r r-r.r r rr1r2z4Cache._ImplementationCreateTable..cSr,r r-r.r r rr1r2r%zPRIMARY KEY ({keys})r+)r'z ({fields}))rTz,CREATE TABLE IF NOT EXISTS "{name}" {fields})rArTN)r rr rrGrH)r?rAr&r' field_listkey_listrTr r r_ImplementationCreateTablesz Cache._ImplementationCreateTable)TNN)T) rbrcrdrernr6rrMrryrrfr r rBrrgs" rgra)re __future__rrrrrr|googlecloudsdk.core.cacherrrgooglecloudsdk.core.utilrr six.movesr rsrr#Tabler$CacheUsingMetadataTablergr r r rs$         _