ui.py
218 lines
| 7.7 KiB
| text/x-python
|
PythonLexer
/ mercurial / ui.py
|
r207 | # ui.py - user interface bits for mercurial | ||
# | ||||
# Copyright 2005 Matt Mackall <mpm@selenic.com> | ||||
# | ||||
# This software may be used and distributed according to the terms | ||||
# of the GNU General Public License, incorporated herein by reference. | ||||
|
r1866 | import ConfigParser | ||
|
r1400 | from i18n import gettext as _ | ||
|
r613 | from demandload import * | ||
|
r1866 | demandload(globals(), "os re socket sys util") | ||
|
r207 | |||
|
r1559 | class ui(object): | ||
|
r207 | def __init__(self, verbose=False, debug=False, quiet=False, | ||
|
r1839 | interactive=True, parentui=None): | ||
|
r960 | self.overlay = {} | ||
|
r1839 | if parentui is None: | ||
|
r1874 | # this is the parent of all ui children | ||
self.parentui = None | ||||
self.cdata = ConfigParser.SafeConfigParser() | ||||
|
r1951 | self.readconfig(util.rcpath()) | ||
|
r285 | |||
|
r1839 | self.quiet = self.configbool("ui", "quiet") | ||
self.verbose = self.configbool("ui", "verbose") | ||||
self.debugflag = self.configbool("ui", "debug") | ||||
self.interactive = self.configbool("ui", "interactive", True) | ||||
|
r285 | |||
|
r1839 | self.updateopts(verbose, debug, quiet, interactive) | ||
self.diffcache = None | ||||
|
r1866 | else: | ||
|
r1874 | # parentui may point to an ui object which is already a child | ||
self.parentui = parentui.parentui or parentui | ||||
parent_cdata = self.parentui.cdata | ||||
self.cdata = ConfigParser.SafeConfigParser(parent_cdata.defaults()) | ||||
# make interpolation work | ||||
for section in parent_cdata.sections(): | ||||
self.cdata.add_section(section) | ||||
for name, value in parent_cdata.items(section, raw=True): | ||||
self.cdata.set(section, name, value) | ||||
|
r1839 | |||
def __getattr__(self, key): | ||||
return getattr(self.parentui, key) | ||||
|
r1071 | |||
def updateopts(self, verbose=False, debug=False, quiet=False, | ||||
interactive=True): | ||||
|
r285 | self.quiet = (self.quiet or quiet) and not verbose and not debug | ||
self.verbose = (self.verbose or verbose) or debug | ||||
self.debugflag = (self.debugflag or debug) | ||||
self.interactive = (self.interactive and interactive) | ||||
|
r1893 | def readconfig(self, fn, root=None): | ||
|
r1483 | if isinstance(fn, basestring): | ||
fn = [fn] | ||||
for f in fn: | ||||
try: | ||||
self.cdata.read(f) | ||||
except ConfigParser.ParsingError, inst: | ||||
raise util.Abort(_("Failed to parse %s\n%s") % (f, inst)) | ||||
|
r1893 | # translate paths relative to root (or home) into absolute paths | ||
if root is None: | ||||
root = os.path.expanduser('~') | ||||
for name, path in self.configitems("paths"): | ||||
|
r1921 | if path and path.find("://") == -1 and not os.path.isabs(path): | ||
|
r1893 | self.cdata.set("paths", name, os.path.join(root, path)) | ||
|
r337 | |||
|
r960 | def setconfig(self, section, name, val): | ||
self.overlay[(section, name)] = val | ||||
def config(self, section, name, default=None): | ||||
if self.overlay.has_key((section, name)): | ||||
return self.overlay[(section, name)] | ||||
if self.cdata.has_option(section, name): | ||||
|
r1876 | try: | ||
return self.cdata.get(section, name) | ||||
except ConfigParser.InterpolationError, inst: | ||||
raise util.Abort(_("Error in configuration:\n%s") % inst) | ||||
|
r1839 | if self.parentui is None: | ||
return default | ||||
else: | ||||
return self.parentui.config(section, name, default) | ||||
|
r285 | |||
|
r960 | def configbool(self, section, name, default=False): | ||
if self.overlay.has_key((section, name)): | ||||
return self.overlay[(section, name)] | ||||
if self.cdata.has_option(section, name): | ||||
|
r1876 | try: | ||
return self.cdata.getboolean(section, name) | ||||
except ConfigParser.InterpolationError, inst: | ||||
raise util.Abort(_("Error in configuration:\n%s") % inst) | ||||
|
r1839 | if self.parentui is None: | ||
return default | ||||
else: | ||||
return self.parentui.configbool(section, name, default) | ||||
|
r285 | |||
def configitems(self, section): | ||||
|
r1839 | items = {} | ||
if self.parentui is not None: | ||||
items = dict(self.parentui.configitems(section)) | ||||
|
r285 | if self.cdata.has_section(section): | ||
|
r1876 | try: | ||
items.update(dict(self.cdata.items(section))) | ||||
except ConfigParser.InterpolationError, inst: | ||||
raise util.Abort(_("Error in configuration:\n%s") % inst) | ||||
|
r1839 | x = items.items() | ||
x.sort() | ||||
return x | ||||
|
r285 | |||
|
r1839 | def walkconfig(self, seen=None): | ||
if seen is None: | ||||
seen = {} | ||||
|
r1028 | for (section, name), value in self.overlay.iteritems(): | ||
yield section, name, value | ||||
seen[section, name] = 1 | ||||
for section in self.cdata.sections(): | ||||
for name, value in self.cdata.items(section): | ||||
if (section, name) in seen: continue | ||||
yield section, name, value.replace('\n', '\\n') | ||||
seen[section, name] = 1 | ||||
|
r1839 | if self.parentui is not None: | ||
for parent in self.parentui.walkconfig(seen): | ||||
yield parent | ||||
|
r1028 | |||
|
r1071 | def extensions(self): | ||
return self.configitems("extensions") | ||||
|
r1637 | def diffopts(self): | ||
if self.diffcache: | ||||
return self.diffcache | ||||
ret = { 'showfunc' : True, 'ignorews' : False} | ||||
for x in self.configitems("diff"): | ||||
k = x[0].lower() | ||||
v = x[1] | ||||
if v: | ||||
v = v.lower() | ||||
if v == 'true': | ||||
value = True | ||||
else: | ||||
value = False | ||||
ret[k] = value | ||||
self.diffcache = ret | ||||
return ret | ||||
|
r608 | def username(self): | ||
|
r691 | return (os.environ.get("HGUSER") or | ||
self.config("ui", "username") or | ||||
|
r608 | os.environ.get("EMAIL") or | ||
(os.environ.get("LOGNAME", | ||||
os.environ.get("USERNAME", "unknown")) | ||||
+ '@' + socket.getfqdn())) | ||||
|
r1129 | def shortuser(self, user): | ||
"""Return a short representation of a user name or email address.""" | ||||
|
r1903 | if not self.verbose: user = util.shortuser(user) | ||
|
r1129 | return user | ||
|
r1893 | def expandpath(self, loc): | ||
|
r1892 | """Return repository location relative to cwd or from [paths]""" | ||
|
r1893 | if loc.find("://") != -1 or os.path.exists(loc): | ||
|
r1892 | return loc | ||
|
r1893 | return self.config("paths", loc, loc) | ||
|
r506 | |||
|
r207 | def write(self, *args): | ||
for a in args: | ||||
sys.stdout.write(str(a)) | ||||
|
r565 | |||
def write_err(self, *args): | ||||
|
r1609 | if not sys.stdout.closed: sys.stdout.flush() | ||
|
r565 | for a in args: | ||
sys.stderr.write(str(a)) | ||||
|
r1837 | def flush(self): | ||
try: | ||||
sys.stdout.flush() | ||||
finally: | ||||
sys.stderr.flush() | ||||
|
r207 | def readline(self): | ||
return sys.stdin.readline()[:-1] | ||||
|
r1062 | def prompt(self, msg, pat, default="y"): | ||
|
r207 | if not self.interactive: return default | ||
while 1: | ||||
self.write(msg, " ") | ||||
r = self.readline() | ||||
if re.match(pat, r): | ||||
return r | ||||
else: | ||||
|
r1402 | self.write(_("unrecognized response\n")) | ||
|
r207 | def status(self, *msg): | ||
if not self.quiet: self.write(*msg) | ||||
|
r234 | def warn(self, *msg): | ||
|
r565 | self.write_err(*msg) | ||
|
r207 | def note(self, *msg): | ||
if self.verbose: self.write(*msg) | ||||
def debug(self, *msg): | ||||
if self.debugflag: self.write(*msg) | ||||
def edit(self, text): | ||||
|
r249 | import tempfile | ||
|
r207 | (fd, name) = tempfile.mkstemp("hg") | ||
f = os.fdopen(fd, "w") | ||||
f.write(text) | ||||
f.close() | ||||
|
r691 | editor = (os.environ.get("HGEDITOR") or | ||
self.config("ui", "editor") or | ||||
|
r608 | os.environ.get("EDITOR", "vi")) | ||
|
r662 | os.environ["HGUSER"] = self.username() | ||
|
r1882 | util.system("%s \"%s\"" % (editor, name), | ||
environ={'HGUSER': self.username()}, | ||||
onerr=util.Abort, errprefix=_("edit failed")) | ||||
|
r207 | |||
t = open(name).read() | ||||
t = re.sub("(?m)^HG:.*\n", "", t) | ||||
|
r662 | os.unlink(name) | ||
|
r207 | return t | ||