Mercurial > public > mercurial-scm > hg
annotate mercurial/util.py @ 698:df78d8ccac4c
Use python function instead of external 'cp' command when cloning repos.
-----BEGIN PGP SIGNED MESSAGE-----
Hash: SHA1
Use python function instead of external 'cp' command when cloning repos.
Inspired by a patch from Stephen Darnell.
manifest hash: b525b0bf40f349b362db7c46d62be41572ef65cf
-----BEGIN PGP SIGNATURE-----
Version: GnuPG v1.4.1 (GNU/Linux)
iD8DBQFC1ncZW7P1GVgWeRoRAiJmAJ9GaCPaG6yOKb72I+SpDDdcdXTISACePD0H
GR/F+qqzi2imdgIV77ziLcQ=
=YVd2
-----END PGP SIGNATURE-----
| author | Thomas Arendsen Hein <thomas@intevation.de> |
|---|---|
| date | Thu, 14 Jul 2005 15:30:49 +0100 |
| parents | 31a9aa890016 |
| children | 5ca319a641e1 |
| rev | line source |
|---|---|
|
419
28511fc21073
[PATCH] file seperator handling for the other 'OS'
mpm@selenic.com
parents:
diff
changeset
|
1 # util.py - utility functions and platform specfic implementations |
|
28511fc21073
[PATCH] file seperator handling for the other 'OS'
mpm@selenic.com
parents:
diff
changeset
|
2 # |
|
28511fc21073
[PATCH] file seperator handling for the other 'OS'
mpm@selenic.com
parents:
diff
changeset
|
3 # Copyright 2005 K. Thananchayan <thananck@yahoo.com> |
|
28511fc21073
[PATCH] file seperator handling for the other 'OS'
mpm@selenic.com
parents:
diff
changeset
|
4 # |
|
28511fc21073
[PATCH] file seperator handling for the other 'OS'
mpm@selenic.com
parents:
diff
changeset
|
5 # This software may be used and distributed according to the terms |
|
28511fc21073
[PATCH] file seperator handling for the other 'OS'
mpm@selenic.com
parents:
diff
changeset
|
6 # of the GNU General Public License, incorporated herein by reference. |
|
28511fc21073
[PATCH] file seperator handling for the other 'OS'
mpm@selenic.com
parents:
diff
changeset
|
7 |
|
28511fc21073
[PATCH] file seperator handling for the other 'OS'
mpm@selenic.com
parents:
diff
changeset
|
8 import os |
|
28511fc21073
[PATCH] file seperator handling for the other 'OS'
mpm@selenic.com
parents:
diff
changeset
|
9 |
| 556 | 10 def unique(g): |
| 11 seen = {} | |
| 12 for f in g: | |
| 13 if f not in seen: | |
| 14 seen[f] = 1 | |
| 15 yield f | |
| 16 | |
| 508 | 17 class CommandError(Exception): pass |
| 18 | |
| 19 def explain_exit(code): | |
| 20 """return a 2-tuple (desc, code) describing a process's status""" | |
| 21 if os.WIFEXITED(code): | |
| 22 val = os.WEXITSTATUS(code) | |
| 23 return "exited with status %d" % val, val | |
| 24 elif os.WIFSIGNALED(code): | |
| 25 val = os.WTERMSIG(code) | |
| 26 return "killed by signal %d" % val, val | |
| 27 elif os.WIFSTOPPED(code): | |
|
667
31a9aa890016
A number of minor fixes to problems that pychecker found.
mark.williamson@cl.cam.ac.uk
parents:
556
diff
changeset
|
28 val = os.WSTOPSIG(code) |
| 508 | 29 return "stopped by signal %d" % val, val |
| 30 raise ValueError("invalid exit code") | |
| 515 | 31 |
| 521 | 32 def system(cmd, errprefix=None): |
| 508 | 33 """execute a shell command that must succeed""" |
| 34 rc = os.system(cmd) | |
| 35 if rc: | |
| 521 | 36 errmsg = "%s %s" % (os.path.basename(cmd.split(None, 1)[0]), |
| 37 explain_exit(rc)[0]) | |
| 38 if errprefix: | |
| 39 errmsg = "%s: %s" % (errprefix, errmsg) | |
| 508 | 40 raise CommandError(errmsg) |
| 41 | |
| 421 | 42 def rename(src, dst): |
| 43 try: | |
| 44 os.rename(src, dst) | |
| 45 except: | |
| 46 os.unlink(dst) | |
| 47 os.rename(src, dst) | |
| 48 | |
|
698
df78d8ccac4c
Use python function instead of external 'cp' command when cloning repos.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
667
diff
changeset
|
49 def copytree(src, dst, copyfile): |
|
df78d8ccac4c
Use python function instead of external 'cp' command when cloning repos.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
667
diff
changeset
|
50 """Copy a directory tree, files are copied using 'copyfile'.""" |
|
df78d8ccac4c
Use python function instead of external 'cp' command when cloning repos.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
667
diff
changeset
|
51 names = os.listdir(src) |
|
df78d8ccac4c
Use python function instead of external 'cp' command when cloning repos.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
667
diff
changeset
|
52 os.mkdir(dst) |
|
df78d8ccac4c
Use python function instead of external 'cp' command when cloning repos.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
667
diff
changeset
|
53 |
|
df78d8ccac4c
Use python function instead of external 'cp' command when cloning repos.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
667
diff
changeset
|
54 for name in names: |
|
df78d8ccac4c
Use python function instead of external 'cp' command when cloning repos.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
667
diff
changeset
|
55 srcname = os.path.join(src, name) |
|
df78d8ccac4c
Use python function instead of external 'cp' command when cloning repos.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
667
diff
changeset
|
56 dstname = os.path.join(dst, name) |
|
df78d8ccac4c
Use python function instead of external 'cp' command when cloning repos.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
667
diff
changeset
|
57 if os.path.isdir(srcname): |
|
df78d8ccac4c
Use python function instead of external 'cp' command when cloning repos.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
667
diff
changeset
|
58 copytree(srcname, dstname, copyfile) |
|
df78d8ccac4c
Use python function instead of external 'cp' command when cloning repos.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
667
diff
changeset
|
59 elif os.path.isfile(srcname): |
|
df78d8ccac4c
Use python function instead of external 'cp' command when cloning repos.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
667
diff
changeset
|
60 copyfile(srcname, dstname) |
|
df78d8ccac4c
Use python function instead of external 'cp' command when cloning repos.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
667
diff
changeset
|
61 else: |
|
df78d8ccac4c
Use python function instead of external 'cp' command when cloning repos.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
667
diff
changeset
|
62 raise IOError("Not a regular file: %r" % srcname) |
|
df78d8ccac4c
Use python function instead of external 'cp' command when cloning repos.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
667
diff
changeset
|
63 |
| 421 | 64 # Platfor specific varients |
|
419
28511fc21073
[PATCH] file seperator handling for the other 'OS'
mpm@selenic.com
parents:
diff
changeset
|
65 if os.name == 'nt': |
| 461 | 66 nulldev = 'NUL:' |
| 67 | |
| 441 | 68 def is_exec(f, last): |
| 69 return last | |
| 70 | |
| 71 def set_exec(f, mode): | |
| 72 pass | |
| 515 | 73 |
|
419
28511fc21073
[PATCH] file seperator handling for the other 'OS'
mpm@selenic.com
parents:
diff
changeset
|
74 def pconvert(path): |
|
28511fc21073
[PATCH] file seperator handling for the other 'OS'
mpm@selenic.com
parents:
diff
changeset
|
75 return path.replace("\\", "/") |
|
422
10c43444a38e
[PATCH] Enables lock work under the other 'OS'
mpm@selenic.com
parents:
421
diff
changeset
|
76 |
|
10c43444a38e
[PATCH] Enables lock work under the other 'OS'
mpm@selenic.com
parents:
421
diff
changeset
|
77 def makelock(info, pathname): |
|
10c43444a38e
[PATCH] Enables lock work under the other 'OS'
mpm@selenic.com
parents:
421
diff
changeset
|
78 ld = os.open(pathname, os.O_CREAT | os.O_WRONLY | os.O_EXCL) |
|
10c43444a38e
[PATCH] Enables lock work under the other 'OS'
mpm@selenic.com
parents:
421
diff
changeset
|
79 os.write(ld, info) |
|
10c43444a38e
[PATCH] Enables lock work under the other 'OS'
mpm@selenic.com
parents:
421
diff
changeset
|
80 os.close(ld) |
|
10c43444a38e
[PATCH] Enables lock work under the other 'OS'
mpm@selenic.com
parents:
421
diff
changeset
|
81 |
|
10c43444a38e
[PATCH] Enables lock work under the other 'OS'
mpm@selenic.com
parents:
421
diff
changeset
|
82 def readlock(pathname): |
|
10c43444a38e
[PATCH] Enables lock work under the other 'OS'
mpm@selenic.com
parents:
421
diff
changeset
|
83 return file(pathname).read() |
| 461 | 84 |
|
419
28511fc21073
[PATCH] file seperator handling for the other 'OS'
mpm@selenic.com
parents:
diff
changeset
|
85 else: |
| 461 | 86 nulldev = '/dev/null' |
| 87 | |
| 441 | 88 def is_exec(f, last): |
| 89 return (os.stat(f).st_mode & 0100 != 0) | |
| 90 | |
| 91 def set_exec(f, mode): | |
| 92 s = os.stat(f).st_mode | |
| 93 if (s & 0100 != 0) == mode: | |
| 94 return | |
| 95 if mode: | |
| 96 # Turn on +x for every +r bit when making a file executable | |
| 97 # and obey umask. | |
| 98 umask = os.umask(0) | |
| 99 os.umask(umask) | |
| 100 os.chmod(f, s | (s & 0444) >> 2 & ~umask) | |
| 101 else: | |
| 102 os.chmod(f, s & 0666) | |
| 103 | |
|
419
28511fc21073
[PATCH] file seperator handling for the other 'OS'
mpm@selenic.com
parents:
diff
changeset
|
104 def pconvert(path): |
|
28511fc21073
[PATCH] file seperator handling for the other 'OS'
mpm@selenic.com
parents:
diff
changeset
|
105 return path |
|
28511fc21073
[PATCH] file seperator handling for the other 'OS'
mpm@selenic.com
parents:
diff
changeset
|
106 |
|
422
10c43444a38e
[PATCH] Enables lock work under the other 'OS'
mpm@selenic.com
parents:
421
diff
changeset
|
107 def makelock(info, pathname): |
|
10c43444a38e
[PATCH] Enables lock work under the other 'OS'
mpm@selenic.com
parents:
421
diff
changeset
|
108 os.symlink(info, pathname) |
|
10c43444a38e
[PATCH] Enables lock work under the other 'OS'
mpm@selenic.com
parents:
421
diff
changeset
|
109 |
|
10c43444a38e
[PATCH] Enables lock work under the other 'OS'
mpm@selenic.com
parents:
421
diff
changeset
|
110 def readlock(pathname): |
|
10c43444a38e
[PATCH] Enables lock work under the other 'OS'
mpm@selenic.com
parents:
421
diff
changeset
|
111 return os.readlink(pathname) |