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 | |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 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""" |
Tim Peters | 0c94724 | 2001-01-21 20:00:00 +0000 | [diff] [blame] | 37 | if hasattr(os, 'chmod'): |
| 38 | st = os.stat(src) |
| 39 | mode = stat.S_IMODE(st[stat.ST_MODE]) |
| 40 | os.chmod(dst, mode) |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 41 | |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 42 | def copystat(src, dst): |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 43 | """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] | 44 | st = os.stat(src) |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 45 | mode = stat.S_IMODE(st[stat.ST_MODE]) |
Tim Peters | 0c94724 | 2001-01-21 20:00:00 +0000 | [diff] [blame] | 46 | if hasattr(os, 'utime'): |
| 47 | os.utime(dst, (st[stat.ST_ATIME], st[stat.ST_MTIME])) |
| 48 | if hasattr(os, 'chmod'): |
| 49 | os.chmod(dst, mode) |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 50 | |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 51 | |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 52 | def copy(src, dst): |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 53 | """Copy data and mode bits ("cp src dst"). |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 54 | |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 55 | The destination may be a directory. |
| 56 | |
| 57 | """ |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 58 | if os.path.isdir(dst): |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 59 | dst = os.path.join(dst, os.path.basename(src)) |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 60 | copyfile(src, dst) |
| 61 | copymode(src, dst) |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 62 | |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 63 | def copy2(src, dst): |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 64 | """Copy data and all stat info ("cp -p src dst"). |
| 65 | |
| 66 | The destination may be a directory. |
| 67 | |
| 68 | """ |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 69 | if os.path.isdir(dst): |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 70 | dst = os.path.join(dst, os.path.basename(src)) |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 71 | copyfile(src, dst) |
| 72 | copystat(src, dst) |
Guido van Rossum | c636014 | 1990-10-13 19:23:40 +0000 | [diff] [blame] | 73 | |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 74 | |
| 75 | def copytree(src, dst, symlinks=0): |
| 76 | """Recursively copy a directory tree using copy2(). |
| 77 | |
| 78 | The destination directory must not already exist. |
| 79 | Error are reported to standard output. |
| 80 | |
| 81 | If the optional symlinks flag is true, symbolic links in the |
| 82 | source tree result in symbolic links in the destination tree; if |
| 83 | it is false, the contents of the files pointed to by symbolic |
| 84 | links are copied. |
| 85 | |
| 86 | XXX Consider this example code rather than the ultimate tool. |
| 87 | |
| 88 | """ |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 89 | names = os.listdir(src) |
Guido van Rossum | 9d0a3df | 1997-04-29 14:45:19 +0000 | [diff] [blame] | 90 | os.mkdir(dst) |
Guido van Rossum | a2baf46 | 1997-04-29 14:06:46 +0000 | [diff] [blame] | 91 | for name in names: |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 92 | srcname = os.path.join(src, name) |
| 93 | dstname = os.path.join(dst, name) |
| 94 | try: |
| 95 | if symlinks and os.path.islink(srcname): |
| 96 | linkto = os.readlink(srcname) |
| 97 | os.symlink(linkto, dstname) |
| 98 | elif os.path.isdir(srcname): |
Fred Drake | 5fa3886 | 2000-04-07 14:34:50 +0000 | [diff] [blame] | 99 | copytree(srcname, dstname, symlinks) |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 100 | else: |
| 101 | copy2(srcname, dstname) |
| 102 | # XXX What about devices, sockets etc.? |
| 103 | except (IOError, os.error), why: |
| 104 | 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] | 105 | |
| 106 | def rmtree(path, ignore_errors=0, onerror=None): |
| 107 | """Recursively delete a directory tree. |
| 108 | |
| 109 | If ignore_errors is set, errors are ignored; otherwise, if |
| 110 | onerror is set, it is called to handle the error; otherwise, an |
| 111 | exception is raised. |
| 112 | |
| 113 | """ |
| 114 | cmdtuples = [] |
| 115 | _build_cmdtuple(path, cmdtuples) |
| 116 | for cmd in cmdtuples: |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 117 | try: |
| 118 | apply(cmd[0], (cmd[1],)) |
| 119 | except: |
| 120 | exc = sys.exc_info() |
| 121 | if ignore_errors: |
| 122 | pass |
| 123 | elif onerror: |
| 124 | onerror(cmd[0], cmd[1], exc) |
| 125 | else: |
| 126 | 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] | 127 | |
| 128 | # Helper for rmtree() |
| 129 | def _build_cmdtuple(path, cmdtuples): |
| 130 | for f in os.listdir(path): |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 131 | real_f = os.path.join(path,f) |
| 132 | if os.path.isdir(real_f) and not os.path.islink(real_f): |
| 133 | _build_cmdtuple(real_f, cmdtuples) |
| 134 | else: |
Guido van Rossum | d832f9e | 1998-10-07 13:18:17 +0000 | [diff] [blame] | 135 | cmdtuples.append((os.remove, real_f)) |
| 136 | cmdtuples.append((os.rmdir, path)) |