dZddlmZddlZddlZddlZddlZddlmZdZ eje Z Gdde ZGdd e Zd ZGd d eZGd deZGddeZy)aLocked file interface that should work on Unix and Windows pythons. This module first tries to use fcntl locking to ensure serialized access to a file, then falls back on a lock file if that is unavialable. Usage:: f = LockedFile('filename', 'r+b', 'rb') f.open_and_lock() if f.is_locked(): print('Acquired filename with r+b mode') f.file_handle().write('locked data') else: print('Acquired filename with rb mode') f.unlock_and_close() )print_functionN)utilz$cache@google.com (David T McWherter)ceZdZdZy) CredentialsFileSymbolicLinkErrorz-Credentials files must not be symbolic links.N__name__ __module__ __qualname____doc__3lib/third_party/oauth2client/contrib/locked_file.pyrr0s7r rceZdZdZy)AlreadyLockedExceptionzETrying to lock a file that has already been locked by the LockedFile.Nrr r rrr4sOr rcvtjj|rtdj |y)NzFile: {0} is a symbolic link.)ospathislinkrformat)filenames r validate_filer9s2 ww~~h. + 2 28 <> > r c4eZdZdZdZdZdZdZdZdZ y) _Openerz,Base class for different locking primitives.cXd|_||_||_||_d|_d|_y)zCreate an Opener. Args: filename: string, The pathname of the file. mode: string, The preferred mode to access the file with. fallback_mode: string, The mode to use if locking fails. FN)_locked _filename_mode_fallback_mode_fh_lock_fd)selfrmode fallback_modes r__init__z_Opener.__init__Bs/ ! + r c|jS)zWas the file locked.)rr!s r is_lockedz_Opener.is_lockedQs ||r c|jS)z5The file handle to the file. Valid only after opened.)rr&s r file_handlez_Opener.file_handleUs xxr c|jS)z"The filename that is being locked.)rr&s rrz_Opener.filenameYs ~~r cy)zOpen the file and lock it. Args: timeout: float, How long to try to lock for. delay: float, How long to wait between retries. Nr r!timeoutdelays r open_and_lockz_Opener.open_and_lock]s r cy)zUnlock and close the file.Nr r&s runlock_and_closez_Opener.unlock_and_closefs r N) rr r r r$r'r)rr/r1r r rrr?s#6   r rc"eZdZdZdZdZdZy) _PosixOpenerz+Lock files using Posix advisory lock files.c@|jr$tdj|jd|_t |j t |j|j |_|j|j}tj} tj |tjtj ztj"z|_d|_y#t$rQ}|jtjk(r*t |j|j|_Yd}~yYd}~d}~wwxYw#t&$r}|jtj(k7rtj|z |k\rgt*j-d|||jr|jj/t |j|j|_Yd}~ytj0|Yd}~nd}~wwxYw})aOpen the file and lock it. Tries to create a .lock file next to the file we're trying to open. Args: timeout: float, How long to try to lock for. delay: float, How long to wait between retries. Raises: AlreadyLockedException: if the lock is already acquired. IOError: if the open fails. CredentialsFileSymbolicLinkError if the file is a symbolic link. zFile {0} is already lockedFNTz'Could not acquire lock %s in %s seconds)rrrrropenrrIOErrorerrnoEACCESr_posix_lockfiletimerO_CREATO_EXCLO_RDWRr OSErrorEEXISTloggerwarnclosesleep)r!r-r.e lock_filename start_times rr/z_PosixOpener.open_and_lockns <<(,33DNNCE E dnn% DNNDJJ7DH,,T^^< YY[  " " (* RYY(>(J!L #  ww%,,&0C0CD'  "77ell*IIK*,8KK I -w8xx(#DNND4G4GHDH 5!! "s9%C7$AE7 EAE  E HBH<HHc$|jr]|j|j}tj|j tj |d|_d|_|jr|jj yy)z?Unlock a file by removing the .lock file, and close the handle.FN)rr9rrrBr unlinkr)r!rEs rr1z_PosixOpener.unlock_and_closesa << 00@M HHT]] # IIm $ DL DM 88 HHNN  r c$dj|S)z3The name of the lock file to use for posix locking.z{0}.lock)r)r!rs rr9z_PosixOpener._posix_lockfiles  **r N)rr r r r/r1r9r r rr3r3ks50"d +r r3cbeZdZdZej dd dZdZdZdZ d dZ dZ y ) LockedFilez+Represent a file that has exclusive access.cd}|s|r ddlm}||||}|s t |||}||_y#t$r$ ddlm}||||}n#t$rYnwxYwYCwxYw)acConstruct a LockedFile. Args: filename: string, The path of the file to open. mode: string, The mode to try to open the file with. fallback_mode: string, The mode to use if locking fails. use_native_locking: bool, Whether or not fcntl/win32 locking is used. Nr) _Win32Opener) _FcntlOpener)"oauth2client.contrib._win32_openerrN ImportError"oauth2client.contrib._fcntl_openerrOr3_opener)r!rr"r#use_native_lockingopenerrNrOs rr$zLockedFile.__init__sx, K%hmD!(D-@F  O)(D-HF"  s2/ AA  A AAAAAc.|jjS)z-Return the filename we were constructed with.)rSrr&s rrzLockedFile.filenames||%%%r c6|jjS)z*Return the file_handle to the opened file.)rSr)r&s rr)zLockedFile.file_handles||''))r c6|jjS)z/Return whether we successfully locked the file.)rSr'r&s rr'zLockedFile.is_lockeds||%%''r c<|jj||y)aUOpen the file, trying to lock it. Args: timeout: float, The number of seconds to try to acquire the lock. delay: float, The number of seconds to wait between retry attempts. Raises: AlreadyLockedException: if the lock is already acquired. IOError: if the open fails. N)rSr/r,s rr/zLockedFile.open_and_locks ""7E2r c8|jjy)zUnlock and close a file.N)rSr1r&s rr1zLockedFile.unlock_and_closes %%'r N)T)rg?) rr r r r positionalr$rr)r'r/r1r r rrKrKs;5T__Q6&*( 3(r rK)r __future__rr7loggingrr: oauth2clientr __author__ getLoggerrr@ Exceptionrrrobjectrr3rKr r rrcs|$&  4   8 $8y8 Y > ) f) XB+7B+J:(:(r