Your IP : 216.73.216.213
C��]� � @ s d Z d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z Gd d � d e � Z
d d � Z d d
� Z d d d d
d � Z
d d d d � Z e j d d � � Z e j d d d d d � � Z d d d � Z d d � Z d d d � Z d S) z Simple file system utilities.�"elaforge@google.com (Evan LaForge)�%matthewb@google.com (Matthew Blecker)� Nc @ s e Z d Z d Z d S)� PasswdErrorz:Exception class for errors loading a password from a file.N)�__name__�
__module__� __qualname__�__doc__� r r �J/opt/alt/python35/lib/python3.5/site-packages/google/apputils/file_util.pyr s r c s � f d d � t j � � D� S)zHLike os.listdir with prepended dir_name, which is often more convenient.c s% g | ] } t j j � | � � q Sr )�os�path�join)�.0�fn)�dir_namer r
�
<listcomp># s z ListDirPath.<locals>.<listcomp>)r �listdir)r r )r r
� ListDirPath! s r c C s$ t | � � } | j � SWd QRXd S)z2Read entire contents of file with name 'filename'.N)�open�read)�filename�fpr r r
�Read& s r Ti� c C s� t j t j Bt j B} | s* | t j O} t j | | | � } z t j | | � Wd t j | � X| d k r� t j | d | � d S)aO Create a file 'filename' with 'contents', with the mode given in 'mode'.
The 'mode' is modified by the umask, as in open(2). If
'overwrite_existing' is False, the file will be opened in O_EXCL mode.
An optional gid can be specified.
Args:
filename: str; the name of the file
contents: str; the data to write to the file
overwrite_existing: bool; whether or not to allow the write if the file
already exists
mode: int; permissions with which to create the file (default is 0666 octal)
gid: int; group id with which to create the file
N� ���) r �O_WRONLY�O_TRUNC�O_CREAT�O_EXCLr �write�close�chown)r �contentsZoverwrite_existing�mode�gid�flags�fdr r r
�Write, s
r' c $ C s
t j d t j j | � � \ } } z t j | | � Wd t j | � XyC t j | | � | d k r{ t j | d | � t j | | � Wnz t
k
r} zZ y t j | � Wn; t
k
r� } z t
d | | f � } WYd d } ~ Xn X| � WYd d } ~ Xn Xd S)a_ Create a file 'filename' with 'contents' atomically.
As in Write, 'mode' is modified by the umask. This creates and moves
a temporary file, and errors doing the above will be propagated normally,
though it will try to clean up the temporary file in that case.
This is very similar to the prodlib function with the same name.
An optional gid can be specified.
Args:
filename: str; the name of the file
contents: str; the data to write to the file
mode: int; permissions with which to create the file (default is 0666 octal)
gid: int; group id with which to create the file
�dirNr z%%s. Additional errors cleaning up: %sr ) �tempfile�mkstempr r �dirnamer r �chmodr! �rename�OSError�remove)r r" r# r$ r& Z tmp_filename�exc�er r r
� AtomicWriteH s $ )r2 c k s9 t j | � } | j | � | j � | V| j � d S)a[ A contextmanager that writes out a string to a file on disk.
This is useful whenever you need to call a function or command that expects a
file on disk with some contents that you have in memory. The context manager
abstracts the writing, flushing, and deletion of the temporary file. This is a
common idiom that boils down to a single with statement.
Note: if you need a temporary file-like object for calling an internal
function, you should use a StringIO as a file-like object and not this.
Temporary files should be avoided unless you need a file name or contents in a
file on disk to be read by some other function or program.
Args:
contents: a string with the contents to write to the file.
**kw: Optional arguments passed on to tempfile.NamedTemporaryFile.
Yields:
The temporary file object, opened in 'w' mode.
N)r) �NamedTemporaryFiler �flushr )r" �kwZtemporary_filer r r
�TemporaryFileWithContentsk s
r6 � �tmpc c s� t j | | | � } z | VWd y t j | � WnG t k
r{ } z' | j d k rf t j | � n � WYd d } ~ Xn XXd S)a� A context manager to create a temporary directory and clean up on exit.
The parameters are the same ones expected by tempfile.mkdtemp.
The directory will be securely and atomically created.
Everything under it will be removed when exiting the context.
Args:
suffix: optional suffix.
prefix: options prefix.
base_path: the base path under which to create the temporary directory.
Yields:
The absolute path of the new temporary directory.
Nz%Cannot call rmtree on a symbolic link)r) �mkdtemp�shutil�rmtreer. �messager �unlink)�suffix�prefix� base_pathZ
temp_dir_pathr1 r r r
�TemporaryDirectory� s rA c C s� t j j | � } | j t j j � } x� t d t | � � D]� } t j j j | d | d � � } y3 | r� t j | � | d k r� t j | | � Wq= t
k
r� } z, | j t j k o� t j j
| � s� � WYd d } ~ Xq= Xq= Wd S)aZ Makes a directory including its parent directories.
This function is equivalent to os.makedirs() but it avoids a race
condition that os.makedirs() has. The race is between os.mkdir() and
os.path.exists() which fail with errors when run in parallel.
Args:
directory: str; the directory to make
force_mode: optional octal, chmod dir to get rid of umask interaction
Raises:
Whatever os.mkdir() raises when it fails for any reason EXCLUDING
"dir already exists". If a directory already exists, it does not
raise anything. This behaviour is different than os.makedirs()
r Nr )r r �normpath�split�sep�range�lenr
�mkdirr, r. �errno�EEXIST�isdir)� directoryZ
force_mode�name�dirs�ir r0 r r r
�MkDirs� s #
$rO c $ C s! y t j | � Wn: t k
rM } z | j t j k r; � WYd d } ~ Xn Xy� t j j | � } xj | r� y t j | � Wn: t k
r� } z | j t j k r� � WYd d } ~ Xn Xt j j | � } qf WWnI t k
r } z) | j t j t j
t j f k r
� WYd d } ~ Xn Xd S)z�Removes dir_name and every subsequently empty directory above it.
Unlike os.removedirs and shutil.rmtree, this function doesn't raise an error
if the directory does not exist.
Args:
dir_name: Directory to be removed.
N) r: r; r. rH �ENOENTr r r+ �rmdirZEACCESZ ENOTEMPTY�EPERM)r �errZparent_directoryr r r
�RmDirs� s" !rT c C sy | d k r$ t j t j � � } nN t | t � rE t j | � } n- t | t � rf t j | � } n t d � � | j S)a4 Find the home directory of a user.
Args:
user: int, str, or None - the uid or login of the user to query for,
or None (the default) to query for the current process' effective user
Returns:
str - the user's home directory
Raises:
TypeError: if user is not int, str, or None.
Nz.user must be None or an instance of int or str)
�pwd�getpwuidr �geteuid�
isinstance�int�str�getpwnam� TypeError�pw_dir)�userZ pw_structr r r
�HomeDir� s
r_ )r r )r �
__author__�
contextlibrH r rU r: �statr) � Exceptionr r r r' r2 �contextmanagerr6 rA rO rT r_ r r r r
�<module> s( #