Guido van Rossum | e7b146f | 2000-02-04 15:28:42 +0000 | [diff] [blame] | 1 | """Utility functions for copying files and directory trees. |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 2 | |
Guido van Rossum | 959fa01 | 1999-08-18 20:03:17 +0000 | [diff] [blame] | 3 | XXX The functions here don't copy the resource fork or other metadata on Mac. |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 4 | |
| 5 | """ |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 6 | |
Guido van Rossum | c96207a | 1992-03-31 18:55:40 +0000 | [diff] [blame] | 7 | import os |
Guido van Rossum | 83c03e2 | 1999-02-23 23:07:51 +0000 | [diff] [blame] | 8 | import sys |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 9 | import stat |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 10 | |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 11 | |
Greg Stein | 42bb8b3 | 2000-07-12 09:55:30 +0000 | [diff] [blame] | 12 | def copyfileobj(fsrc, fdst, length=16*1024): |
| 13 | """copy data from file-like object fsrc to file-like object fdst""" |
| 14 | while 1: |
| 15 | buf = fsrc.read(length) |
| 16 | if not buf: |
| 17 | break |
| 18 | fdst.write(buf) |
| 19 | |
| 20 | |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 21 | def copyfile(src, dst): |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 22 | """Copy data from src to dst""" |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 23 | fsrc = None |
| 24 | fdst = None |
| 25 | try: |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 26 | fsrc = open(src, 'rb') |
| 27 | fdst = open(dst, 'wb') |
Greg Stein | 42bb8b3 | 2000-07-12 09:55:30 +0000 | [diff] [blame] | 28 | copyfileobj(fsrc, fdst) |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 29 | finally: |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 30 | if fdst: |
| 31 | fdst.close() |
| 32 | if fsrc: |
| 33 | fsrc.close() |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 34 | |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 35 | def copymode(src, dst): |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 36 | """Copy mode bits from src to dst""" |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 37 | st = os.stat(src) |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 38 | mode = stat.S_IMODE(st[stat.ST_MODE]) |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 39 | os.chmod(dst, mode) |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 40 | |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 41 | def copystat(src, dst): |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 42 | """Copy all stat info (mode bits, atime and mtime) from src to dst""" |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 43 | st = os.stat(src) |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 44 | mode = stat.S_IMODE(st[stat.ST_MODE]) |
Guido van Rossum | 96372a2 | 1998-10-02 03:16:08 +0000 | [diff] [blame] | 45 | os.utime(dst, (st[stat.ST_ATIME], st[stat.ST_MTIME])) |
Guido van Rossum | e1bf7e8 | 1999-01-14 00:42:00 +0000 | [diff] [blame] | 46 | os.chmod(dst, mode) |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 47 | |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 48 | |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 49 | def copy(src, dst): |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 50 | """Copy data and mode bits ("cp src dst"). |
| 51 | |
| 52 | The destination may be a directory. |
| 53 | |
| 54 | """ |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 55 | if os.path.isdir(dst): |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 56 | dst = os.path.join(dst, os.path.basename(src)) |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 57 | copyfile(src, dst) |
| 58 | copymode(src, dst) |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 59 | |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 60 | def copy2(src, dst): |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 61 | """Copy data and all stat info ("cp -p src dst"). |
| 62 | |
| 63 | The destination may be a directory. |
| 64 | |
| 65 | """ |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 66 | if os.path.isdir(dst): |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 67 | dst = os.path.join(dst, os.path.basename(src)) |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 68 | copyfile(src, dst) |
| 69 | copystat(src, dst) |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 70 | |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 71 | |
| 72 | def copytree(src, dst, symlinks=0): |
| 73 | """Recursively copy a directory tree using copy2(). |
| 74 | |
| 75 | The destination directory must not already exist. |
| 76 | Error are reported to standard output. |
| 77 | |
| 78 | If the optional symlinks flag is true, symbolic links in the |
| 79 | source tree result in symbolic links in the destination tree; if |
| 80 | it is false, the contents of the files pointed to by symbolic |
| 81 | links are copied. |
| 82 | |
| 83 | XXX Consider this example code rather than the ultimate tool. |
| 84 | |
| 85 | """ |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 86 | names = os.listdir(src) |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 87 | os.mkdir(dst) |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 88 | for name in names: |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 89 | srcname = os.path.join(src, name) |
| 90 | dstname = os.path.join(dst, name) |
| 91 | try: |
| 92 | if symlinks and os.path.islink(srcname): |
| 93 | linkto = os.readlink(srcname) |
| 94 | os.symlink(linkto, dstname) |
| 95 | elif os.path.isdir(srcname): |
Fred Drake | 5fa3886 | 2000-04-07 14:34:50 +0000 | [diff] [blame] | 96 | copytree(srcname, dstname, symlinks) |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 97 | else: |
| 98 | copy2(srcname, dstname) |
| 99 | # XXX What about devices, sockets etc.? |
| 100 | except (IOError, os.error), why: |
| 101 | print "Can't copy %s to %s: %s" % (`srcname`, `dstname`, str(why)) |
Guido van Rossum | d767329 | 1998-02-06 21:38:09 +0000 | [diff] [blame] | 102 | |
| 103 | def rmtree(path, ignore_errors=0, onerror=None): |
| 104 | """Recursively delete a directory tree. |
| 105 | |
| 106 | If ignore_errors is set, errors are ignored; otherwise, if |
| 107 | onerror is set, it is called to handle the error; otherwise, an |
| 108 | exception is raised. |
| 109 | |
| 110 | """ |
| 111 | cmdtuples = [] |
| 112 | _build_cmdtuple(path, cmdtuples) |
| 113 | for cmd in cmdtuples: |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 114 | try: |
| 115 | apply(cmd[0], (cmd[1],)) |
| 116 | except: |
| 117 | exc = sys.exc_info() |
| 118 | if ignore_errors: |
| 119 | pass |
| 120 | elif onerror: |
| 121 | onerror(cmd[0], cmd[1], exc) |
| 122 | else: |
| 123 | raise exc[0], (exc[1][0], exc[1][1] + ' removing '+cmd[1]) |
Guido van Rossum | d767329 | 1998-02-06 21:38:09 +0000 | [diff] [blame] | 124 | |
| 125 | # Helper for rmtree() |
| 126 | def _build_cmdtuple(path, cmdtuples): |
| 127 | for f in os.listdir(path): |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 128 | real_f = os.path.join(path,f) |
| 129 | if os.path.isdir(real_f) and not os.path.islink(real_f): |
| 130 | _build_cmdtuple(real_f, cmdtuples) |
| 131 | else: |
Guido van Rossum | d832f9e | 1998-10-07 13:18:17 +0000 | [diff] [blame] | 132 | cmdtuples.append((os.remove, real_f)) |
| 133 | cmdtuples.append((os.rmdir, path)) |