util.py
96 lines
| 2.4 KiB
| text/x-python
|
PythonLexer
/ mercurial / util.py
|
r419 | # util.py - utility functions and platform specfic implementations | ||
# | ||||
# Copyright 2005 K. Thananchayan <thananck@yahoo.com> | ||||
# | ||||
# This software may be used and distributed according to the terms | ||||
# of the GNU General Public License, incorporated herein by reference. | ||||
import os | ||||
|
r556 | def unique(g): | ||
seen = {} | ||||
for f in g: | ||||
if f not in seen: | ||||
seen[f] = 1 | ||||
yield f | ||||
|
r508 | class CommandError(Exception): pass | ||
def explain_exit(code): | ||||
"""return a 2-tuple (desc, code) describing a process's status""" | ||||
if os.WIFEXITED(code): | ||||
val = os.WEXITSTATUS(code) | ||||
return "exited with status %d" % val, val | ||||
elif os.WIFSIGNALED(code): | ||||
val = os.WTERMSIG(code) | ||||
return "killed by signal %d" % val, val | ||||
elif os.WIFSTOPPED(code): | ||||
|
r667 | val = os.WSTOPSIG(code) | ||
|
r508 | return "stopped by signal %d" % val, val | ||
raise ValueError("invalid exit code") | ||||
|
r515 | |||
|
r521 | def system(cmd, errprefix=None): | ||
|
r508 | """execute a shell command that must succeed""" | ||
rc = os.system(cmd) | ||||
if rc: | ||||
|
r521 | errmsg = "%s %s" % (os.path.basename(cmd.split(None, 1)[0]), | ||
explain_exit(rc)[0]) | ||||
if errprefix: | ||||
errmsg = "%s: %s" % (errprefix, errmsg) | ||||
|
r508 | raise CommandError(errmsg) | ||
|
r421 | def rename(src, dst): | ||
try: | ||||
os.rename(src, dst) | ||||
except: | ||||
os.unlink(dst) | ||||
os.rename(src, dst) | ||||
# Platfor specific varients | ||||
|
r419 | if os.name == 'nt': | ||
|
r461 | nulldev = 'NUL:' | ||
|
r441 | def is_exec(f, last): | ||
return last | ||||
def set_exec(f, mode): | ||||
pass | ||||
|
r515 | |||
|
r419 | def pconvert(path): | ||
return path.replace("\\", "/") | ||||
|
r422 | |||
def makelock(info, pathname): | ||||
ld = os.open(pathname, os.O_CREAT | os.O_WRONLY | os.O_EXCL) | ||||
os.write(ld, info) | ||||
os.close(ld) | ||||
def readlock(pathname): | ||||
return file(pathname).read() | ||||
|
r461 | |||
|
r419 | else: | ||
|
r461 | nulldev = '/dev/null' | ||
|
r441 | def is_exec(f, last): | ||
return (os.stat(f).st_mode & 0100 != 0) | ||||
def set_exec(f, mode): | ||||
s = os.stat(f).st_mode | ||||
if (s & 0100 != 0) == mode: | ||||
return | ||||
if mode: | ||||
# Turn on +x for every +r bit when making a file executable | ||||
# and obey umask. | ||||
umask = os.umask(0) | ||||
os.umask(umask) | ||||
os.chmod(f, s | (s & 0444) >> 2 & ~umask) | ||||
else: | ||||
os.chmod(f, s & 0666) | ||||
|
r419 | def pconvert(path): | ||
return path | ||||
|
r422 | def makelock(info, pathname): | ||
os.symlink(info, pathname) | ||||
def readlock(pathname): | ||||
return os.readlink(pathname) | ||||