pep8-fied distutils.dir_util

This commit is contained in:
Tarek Ziadé 2009-05-17 11:22:36 +00:00
parent 294c9d90df
commit 513a8b7d99

View file

@ -5,7 +5,6 @@ Utility functions for manipulating directories and directory trees."""
__revision__ = "$Id$" __revision__ = "$Id$"
import os, sys import os, sys
from types import *
from distutils.errors import DistutilsFileError, DistutilsInternalError from distutils.errors import DistutilsFileError, DistutilsInternalError
from distutils import log from distutils import log
@ -17,19 +16,20 @@ _path_created = {}
# b) it blows up if the directory already exists (I want to silently # b) it blows up if the directory already exists (I want to silently
# succeed in that case). # succeed in that case).
def mkpath(name, mode=0777, verbose=1, dry_run=0): def mkpath(name, mode=0777, verbose=1, dry_run=0):
"""Create a directory and any missing ancestor directories. If the """Create a directory and any missing ancestor directories.
directory already exists (or if 'name' is the empty string, which
means the current directory, which of course exists), then do If the directory already exists (or if 'name' is the empty string, which
nothing. Raise DistutilsFileError if unable to create some means the current directory, which of course exists), then do nothing.
directory along the way (eg. some sub-path exists, but is a file Raise DistutilsFileError if unable to create some directory along the way
rather than a directory). If 'verbose' is true, print a one-line (eg. some sub-path exists, but is a file rather than a directory).
summary of each mkdir to stdout. Return the list of directories If 'verbose' is true, print a one-line summary of each mkdir to stdout.
actually created.""" Return the list of directories actually created.
"""
global _path_created global _path_created
# Detect a common bug -- name is None # Detect a common bug -- name is None
if not isinstance(name, StringTypes): if not isinstance(name, basestring):
raise DistutilsInternalError, \ raise DistutilsInternalError, \
"mkpath: 'name' must be a string (got %r)" % (name,) "mkpath: 'name' must be a string (got %r)" % (name,)
@ -77,19 +77,16 @@ def mkpath (name, mode=0777, verbose=1, dry_run=0):
_path_created[abs_head] = 1 _path_created[abs_head] = 1
return created_dirs return created_dirs
# mkpath ()
def create_tree(base_dir, files, mode=0777, verbose=1, dry_run=0): def create_tree(base_dir, files, mode=0777, verbose=1, dry_run=0):
"""Create all the empty directories under 'base_dir' needed to put 'files'
there.
"""Create all the empty directories under 'base_dir' needed to 'base_dir' is just the a name of a directory which doesn't necessarily
put 'files' there. 'base_dir' is just the a name of a directory exist yet; 'files' is a list of filenames to be interpreted relative to
which doesn't necessarily exist yet; 'files' is a list of filenames 'base_dir'. 'base_dir' + the directory portion of every file in 'files'
to be interpreted relative to 'base_dir'. 'base_dir' + the will be created if it doesn't already exist. 'mode', 'verbose' and
directory portion of every file in 'files' will be created if it 'dry_run' flags are as for 'mkpath()'.
doesn't already exist. 'mode', 'verbose' and 'dry_run' flags are as """
for 'mkpath()'."""
# First get the list of directories to create # First get the list of directories to create
need_dir = {} need_dir = {}
for file in files: for file in files:
@ -101,19 +98,11 @@ def create_tree (base_dir, files, mode=0777, verbose=1, dry_run=0):
for dir in need_dirs: for dir in need_dirs:
mkpath(dir, mode, verbose=verbose, dry_run=dry_run) mkpath(dir, mode, verbose=verbose, dry_run=dry_run)
# create_tree () def copy_tree(src, dst, preserve_mode=1, preserve_times=1,
preserve_symlinks=0, update=0, verbose=1, dry_run=0):
"""Copy an entire directory tree 'src' to a new location 'dst'.
Both 'src' and 'dst' must be directory names. If 'src' is not a
def copy_tree (src, dst,
preserve_mode=1,
preserve_times=1,
preserve_symlinks=0,
update=0,
verbose=1,
dry_run=0):
"""Copy an entire directory tree 'src' to a new location 'dst'. Both
'src' and 'dst' must be directory names. If 'src' is not a
directory, raise DistutilsFileError. If 'dst' does not exist, it is directory, raise DistutilsFileError. If 'dst' does not exist, it is
created with 'mkpath()'. The end result of the copy is that every created with 'mkpath()'. The end result of the copy is that every
file in 'src' is copied to 'dst', and directories under 'src' are file in 'src' is copied to 'dst', and directories under 'src' are
@ -128,8 +117,8 @@ def copy_tree (src, dst,
directories. If 'preserve_symlinks' is true, symlinks will be directories. If 'preserve_symlinks' is true, symlinks will be
copied as symlinks (on platforms that support them!); otherwise copied as symlinks (on platforms that support them!); otherwise
(the default), the destination of the symlink will be copied. (the default), the destination of the symlink will be copied.
'update' and 'verbose' are the same as for 'copy_file'.""" 'update' and 'verbose' are the same as for 'copy_file'.
"""
from distutils.file_util import copy_file from distutils.file_util import copy_file
if not dry_run and not os.path.isdir(src): if not dry_run and not os.path.isdir(src):
@ -174,10 +163,8 @@ def copy_tree (src, dst,
return outputs return outputs
# copy_tree ()
# Helper for remove_tree()
def _build_cmdtuple(path, cmdtuples): def _build_cmdtuple(path, cmdtuples):
"""Helper for remove_tree()."""
for f in os.listdir(path): for f in os.listdir(path):
real_f = os.path.join(path,f) real_f = os.path.join(path,f)
if os.path.isdir(real_f) and not os.path.islink(real_f): if os.path.isdir(real_f) and not os.path.islink(real_f):
@ -186,10 +173,11 @@ def _build_cmdtuple(path, cmdtuples):
cmdtuples.append((os.remove, real_f)) cmdtuples.append((os.remove, real_f))
cmdtuples.append((os.rmdir, path)) cmdtuples.append((os.rmdir, path))
def remove_tree(directory, verbose=1, dry_run=0): def remove_tree(directory, verbose=1, dry_run=0):
"""Recursively remove an entire directory tree. Any errors are ignored """Recursively remove an entire directory tree.
(apart from being reported to stdout if 'verbose' is true).
Any errors are ignored (apart from being reported to stdout if 'verbose'
is true).
""" """
from distutils.util import grok_environment_error from distutils.util import grok_environment_error
global _path_created global _path_created
@ -211,10 +199,10 @@ def remove_tree (directory, verbose=1, dry_run=0):
log.warn(grok_environment_error( log.warn(grok_environment_error(
exc, "error removing %s: " % directory)) exc, "error removing %s: " % directory))
def ensure_relative(path): def ensure_relative(path):
"""Take the full path 'path', and make it a relative path so """Take the full path 'path', and make it a relative path.
it can be the second argument to os.path.join().
This is useful to make 'path' the second argument to os.path.join().
""" """
drive, path = os.path.splitdrive(path) drive, path = os.path.splitdrive(path)
if path[0:1] == os.sep: if path[0:1] == os.sep: