ui.py
284 lines
| 10.1 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 * | ||
|
r2281 | demandload(globals(), "errno getpass os re smtplib socket sys tempfile util") | ||
|
r207 | |||
|
r1559 | class ui(object): | ||
|
r207 | def __init__(self, verbose=False, debug=False, quiet=False, | ||
|
r2166 | interactive=True, traceback=False, 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) | ||||
|
r2166 | self.traceback = traceback | ||
|
r285 | |||
|
r1839 | self.updateopts(verbose, debug, quiet, interactive) | ||
self.diffcache = None | ||||
|
r2033 | self.header = [] | ||
self.prev_header = [] | ||||
|
r2072 | self.revlogopts = self.configrevlog() | ||
|
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, | ||||
|
r2166 | interactive=True, traceback=False): | ||
|
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) | ||||
|
r2166 | self.traceback = self.traceback or traceback | ||
|
r285 | |||
|
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") | ||||
|
r2003 | def hgignorefiles(self): | ||
result = [] | ||||
cfgitems = self.configitems("ui") | ||||
for key, value in cfgitems: | ||||
if key == 'ignore' or key.startswith('ignore.'): | ||||
path = os.path.expanduser(value) | ||||
result.append(path) | ||||
return result | ||||
|
r2072 | def configrevlog(self): | ||
ret = {} | ||||
for x in self.configitems("revlog"): | ||||
k = x[0].lower() | ||||
ret[k] = x[1] | ||||
return ret | ||||
|
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): | ||
|
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. | ||||
If not found, use $LOGNAME or $USERNAME +"@full.hostname". | ||||
""" | ||||
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: | ||||
user = os.environ.get("LOGNAME") or os.environ.get("USERNAME") | ||||
if user: | ||||
user = "%s@%s" % (user, socket.getfqdn()) | ||||
if not user: | ||||
raise util.Abort(_("Please specify a username.")) | ||||
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 | ||
|
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): | ||
|
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 | |||
def sendmail(self): | ||||
s = smtplib.SMTP() | ||||
s.connect(host = self.config('smtp', 'host', 'mail'), | ||||
port = int(self.config('smtp', 'port', 25))) | ||||
if self.configbool('smtp', 'tls'): | ||||
s.ehlo() | ||||
s.starttls() | ||||
s.ehlo() | ||||
username = self.config('smtp', 'username') | ||||
password = self.config('smtp', 'password') | ||||
if username and password: | ||||
s.login(username, password) | ||||
return s | ||||