ui.py
323 lines
| 11.4 KiB
| text/x-python
|
PythonLexer
/ mercurial / ui.py
|
r207 | # ui.py - user interface bits for mercurial | ||
# | ||||
|
r2859 | # Copyright 2005, 2006 Matt Mackall <mpm@selenic.com> | ||
|
r207 | # | ||
# This software may be used and distributed according to the terms | ||||
# of the GNU General Public License, incorporated herein by reference. | ||||
|
r1400 | from i18n import gettext as _ | ||
|
r613 | from demandload import * | ||
|
r2889 | demandload(globals(), "errno getpass os re socket sys tempfile") | ||
|
r3385 | demandload(globals(), "ConfigParser traceback util") | ||
|
r207 | |||
|
r3344 | def dupconfig(orig): | ||
new = ConfigParser.SafeConfigParser(orig.defaults()) | ||||
updateconfig(orig, new) | ||||
return new | ||||
def updateconfig(source, dest): | ||||
for section in source.sections(): | ||||
if not dest.has_section(section): | ||||
dest.add_section(section) | ||||
for name, value in source.items(section, raw=True): | ||||
dest.set(section, name, value) | ||||
|
r1559 | class ui(object): | ||
|
r207 | def __init__(self, verbose=False, debug=False, quiet=False, | ||
|
r3014 | interactive=True, traceback=False, parentui=None): | ||
|
r3344 | self.overlay = None | ||
|
r3339 | self.header = [] | ||
self.prev_header = [] | ||||
|
r1839 | if parentui is None: | ||
|
r1874 | # this is the parent of all ui children | ||
self.parentui = None | ||||
|
r3014 | self.readhooks = [] | ||
|
r3350 | self.quiet = quiet | ||
self.verbose = verbose | ||||
self.debugflag = debug | ||||
self.interactive = interactive | ||||
self.traceback = traceback | ||||
|
r1874 | self.cdata = ConfigParser.SafeConfigParser() | ||
|
r1951 | self.readconfig(util.rcpath()) | ||
|
r1839 | self.updateopts(verbose, debug, quiet, interactive) | ||
|
r1866 | else: | ||
|
r1874 | # parentui may point to an ui object which is already a child | ||
self.parentui = parentui.parentui or parentui | ||||
|
r3338 | self.readhooks = self.parentui.readhooks[:] | ||
|
r3344 | self.cdata = dupconfig(self.parentui.cdata) | ||
if self.parentui.overlay: | ||||
self.overlay = dupconfig(self.parentui.overlay) | ||||
|
r1839 | |||
def __getattr__(self, key): | ||||
return getattr(self.parentui, key) | ||||
|
r1071 | |||
def updateopts(self, verbose=False, debug=False, quiet=False, | ||||
|
r2293 | interactive=True, traceback=False, config=[]): | ||
|
r3346 | for section, name, value in config: | ||
self.setconfig(section, name, value) | ||||
|
r285 | |||
|
r3350 | if quiet or verbose or debug: | ||
self.setconfig('ui', 'quiet', str(bool(quiet))) | ||||
self.setconfig('ui', 'verbose', str(bool(verbose))) | ||||
self.setconfig('ui', 'debug', str(bool(debug))) | ||||
self.verbosity_constraints() | ||||
if not interactive: | ||||
self.setconfig('ui', 'interactive', 'False') | ||||
self.interactive = False | ||||
self.traceback = self.traceback or traceback | ||||
def verbosity_constraints(self): | ||||
self.quiet = self.configbool('ui', 'quiet') | ||||
self.verbose = self.configbool('ui', 'verbose') | ||||
self.debugflag = self.configbool('ui', 'debug') | ||||
|
r3349 | if self.debugflag: | ||
self.verbose = True | ||||
self.quiet = False | ||||
elif self.verbose and self.quiet: | ||||
|
r3350 | self.quiet = self.verbose = False | ||
|
r3349 | |||
|
r1893 | def readconfig(self, fn, root=None): | ||
|
r1483 | if isinstance(fn, basestring): | ||
fn = [fn] | ||||
for f in fn: | ||||
try: | ||||
|
r3098 | self.cdata.read(f) | ||
|
r1483 | except ConfigParser.ParsingError, inst: | ||
raise util.Abort(_("Failed to parse %s\n%s") % (f, inst)) | ||||
|
r3344 | # override data from config files with data set with ui.setconfig | ||
if self.overlay: | ||||
updateconfig(self.overlay, self.cdata) | ||||
|
r3347 | if root is None: | ||
root = os.path.expanduser('~') | ||||
self.fixconfig(root=root) | ||||
|
r2944 | for hook in self.readhooks: | ||
hook(self) | ||||
|
r337 | |||
|
r3014 | def addreadhook(self, hook): | ||
self.readhooks.append(hook) | ||||
|
r3347 | def fixconfig(self, section=None, name=None, value=None, root=None): | ||
# translate paths relative to root (or home) into absolute paths | ||||
if section is None or section == 'paths': | ||||
if root is None: | ||||
root = os.getcwd() | ||||
items = section and [(name, value)] or [] | ||||
for cdata in self.cdata, self.overlay: | ||||
if not cdata: continue | ||||
if not items and cdata.has_section('paths'): | ||||
pathsitems = cdata.items('paths') | ||||
else: | ||||
pathsitems = items | ||||
for n, path in pathsitems: | ||||
if path and "://" not in path and not os.path.isabs(path): | ||||
cdata.set("paths", n, os.path.join(root, path)) | ||||
|
r3350 | # update quiet/verbose/debug and interactive status | ||
if section is None or section == 'ui': | ||||
if name is None or name in ('quiet', 'verbose', 'debug'): | ||||
self.verbosity_constraints() | ||||
if name is None or name == 'interactive': | ||||
self.interactive = self.configbool("ui", "interactive", True) | ||||
|
r3344 | def setconfig(self, section, name, value): | ||
if not self.overlay: | ||||
self.overlay = ConfigParser.SafeConfigParser() | ||||
for cdata in (self.overlay, self.cdata): | ||||
if not cdata.has_section(section): | ||||
cdata.add_section(section) | ||||
cdata.set(section, name, value) | ||||
|
r3347 | self.fixconfig(section, name, value) | ||
|
r960 | |||
|
r3341 | def _config(self, section, name, default, funcname): | ||
|
r960 | if self.cdata.has_option(section, name): | ||
|
r1876 | try: | ||
|
r3341 | func = getattr(self.cdata, funcname) | ||
return func(section, name) | ||||
|
r1876 | except ConfigParser.InterpolationError, inst: | ||
|
r3073 | raise util.Abort(_("Error in configuration section [%s] " | ||
"parameter '%s':\n%s") | ||||
% (section, name, inst)) | ||||
|
r3343 | return default | ||
|
r3341 | |||
def config(self, section, name, default=None): | ||||
return self._config(section, name, default, 'get') | ||||
def configbool(self, section, name, default=False): | ||||
return self._config(section, name, default, 'getboolean') | ||||
|
r285 | |||
|
r2499 | def configlist(self, section, name, default=None): | ||
"""Return a list of comma/space separated strings""" | ||||
result = self.config(section, name) | ||||
if result is None: | ||||
|
r2502 | result = default or [] | ||
if isinstance(result, basestring): | ||||
result = result.replace(",", " ").split() | ||||
return result | ||||
|
r2499 | |||
|
r2343 | def has_config(self, section): | ||
'''tell whether section exists in config.''' | ||||
return self.cdata.has_section(section) | ||||
|
r285 | def configitems(self, section): | ||
|
r1839 | items = {} | ||
|
r285 | if self.cdata.has_section(section): | ||
|
r1876 | try: | ||
items.update(dict(self.cdata.items(section))) | ||||
except ConfigParser.InterpolationError, inst: | ||||
|
r3073 | raise util.Abort(_("Error in configuration section [%s]:\n%s") | ||
% (section, inst)) | ||||
|
r1839 | x = items.items() | ||
x.sort() | ||||
return x | ||||
|
r285 | |||
|
r3343 | def walkconfig(self): | ||
|
r3342 | sections = self.cdata.sections() | ||
sections.sort() | ||||
for section in sections: | ||||
for name, value in self.configitems(section): | ||||
yield section, name, value.replace('\n', '\\n') | ||||
|
r1028 | |||
|
r1071 | def extensions(self): | ||
|
r2403 | result = self.configitems("extensions") | ||
for i, (key, value) in enumerate(result): | ||||
if value: | ||||
result[i] = (key, os.path.expanduser(value)) | ||||
return result | ||||
|
r1071 | |||
|
r2003 | def hgignorefiles(self): | ||
|
r2403 | result = [] | ||
for key, value in self.configitems("ui"): | ||||
if key == 'ignore' or key.startswith('ignore.'): | ||||
result.append(os.path.expanduser(value)) | ||||
return result | ||||
|
r2003 | |||
|
r2072 | def configrevlog(self): | ||
|
r2403 | result = {} | ||
for key, value in self.configitems("revlog"): | ||||
result[key.lower()] = value | ||||
return result | ||||
|
r2388 | |||
|
r608 | def username(self): | ||
|
r1985 | """Return default username to be used in commits. | ||
Searched in this order: $HGUSER, [ui] section of hgrcs, $EMAIL | ||||
and stop searching if one of these is set. | ||||
Abort if found username is an empty string to force specifying | ||||
the commit user elsewhere, e.g. with line option or repo hgrc. | ||||
|
r2343 | If not found, use ($LOGNAME or $USER or $LNAME or | ||
$USERNAME) +"@full.hostname". | ||||
|
r1985 | """ | ||
user = os.environ.get("HGUSER") | ||||
if user is None: | ||||
user = self.config("ui", "username") | ||||
if user is None: | ||||
user = os.environ.get("EMAIL") | ||||
if user is None: | ||||
|
r2343 | try: | ||
|
r2652 | user = '%s@%s' % (util.getuser(), socket.getfqdn()) | ||
|
r2343 | except KeyError: | ||
raise util.Abort(_("Please specify a username.")) | ||||
|
r1985 | return user | ||
|
r608 | |||
|
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 | ||
|
r2494 | def expandpath(self, loc, default=None): | ||
|
r1892 | """Return repository location relative to cwd or from [paths]""" | ||
|
r2624 | if "://" in loc or os.path.isdir(loc): | ||
|
r1892 | return loc | ||
|
r2495 | path = self.config("paths", loc) | ||
if not path and default is not None: | ||||
path = self.config("paths", default) | ||||
|
r2498 | return path or loc | ||
|
r506 | |||
|
r207 | def write(self, *args): | ||
|
r2033 | if self.header: | ||
if self.header != self.prev_header: | ||||
self.prev_header = self.header | ||||
self.write(*self.header) | ||||
self.header = [] | ||||
|
r207 | for a in args: | ||
sys.stdout.write(str(a)) | ||||
|
r565 | |||
|
r2033 | def write_header(self, *args): | ||
for a in args: | ||||
self.header.append(str(a)) | ||||
|
r565 | def write_err(self, *args): | ||
|
r1989 | try: | ||
if not sys.stdout.closed: sys.stdout.flush() | ||||
for a in args: | ||||
sys.stderr.write(str(a)) | ||||
except IOError, inst: | ||||
if inst.errno != errno.EPIPE: | ||||
raise | ||||
|
r565 | |||
|
r1837 | def flush(self): | ||
|
r2013 | try: sys.stdout.flush() | ||
except: pass | ||||
try: sys.stderr.flush() | ||||
except: pass | ||||
|
r1837 | |||
|
r207 | def readline(self): | ||
return sys.stdin.readline()[:-1] | ||||
|
r2281 | def prompt(self, msg, pat=None, default="y"): | ||
|
r207 | if not self.interactive: return default | ||
while 1: | ||||
self.write(msg, " ") | ||||
r = self.readline() | ||||
|
r2281 | if not pat or re.match(pat, r): | ||
|
r207 | return r | ||
else: | ||||
|
r1402 | self.write(_("unrecognized response\n")) | ||
|
r2281 | def getpass(self, prompt=None, default=None): | ||
if not self.interactive: return default | ||||
return getpass.getpass(prompt or _('password: ')) | ||||
|
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) | ||||
|
r1983 | def edit(self, text, user): | ||
|
r2206 | (fd, name) = tempfile.mkstemp(prefix="hg-editor-", suffix=".txt", | ||
text=True) | ||||
|
r1984 | try: | ||
f = os.fdopen(fd, "w") | ||||
f.write(text) | ||||
f.close() | ||||
editor = (os.environ.get("HGEDITOR") or | ||||
self.config("ui", "editor") or | ||||
os.environ.get("EDITOR", "vi")) | ||||
|
r207 | |||
|
r1984 | util.system("%s \"%s\"" % (editor, name), | ||
environ={'HGUSER': user}, | ||||
onerr=util.Abort, errprefix=_("edit failed")) | ||||
|
r608 | |||
|
r1984 | f = open(name) | ||
t = f.read() | ||||
f.close() | ||||
t = re.sub("(?m)^HG:.*\n", "", t) | ||||
finally: | ||||
os.unlink(name) | ||||
|
r662 | |||
|
r207 | return t | ||
|
r2200 | |||
|
r2335 | def print_exc(self): | ||
'''print exception traceback if traceback printing enabled. | ||||
only to call in exception handler. returns true if traceback | ||||
printed.''' | ||||
if self.traceback: | ||||
traceback.print_exc() | ||||
return self.traceback | ||||