ui.py
83 lines
| 2.6 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. | ||||
|
r285 | import os, sys, re, ConfigParser | ||
|
r207 | |||
class ui: | ||||
def __init__(self, verbose=False, debug=False, quiet=False, | ||||
interactive=True): | ||||
|
r285 | self.cdata = ConfigParser.SafeConfigParser() | ||
self.cdata.read(os.path.expanduser("~/.hgrc")) | ||||
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) | ||||
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) | ||||
|
r337 | def readconfig(self, fp): | ||
self.cdata.readfp(fp) | ||||
|
r285 | def config(self, section, val, default=None): | ||
if self.cdata.has_option(section, val): | ||||
return self.cdata.get(section, val) | ||||
return default | ||||
def configbool(self, section, val, default=False): | ||||
if self.cdata.has_option(section, val): | ||||
return self.cdata.getboolean(section, val) | ||||
return default | ||||
def configitems(self, section): | ||||
if self.cdata.has_section(section): | ||||
return self.cdata.items(section) | ||||
return [] | ||||
|
r207 | def write(self, *args): | ||
for a in args: | ||||
sys.stdout.write(str(a)) | ||||
def readline(self): | ||||
return sys.stdin.readline()[:-1] | ||||
def prompt(self, msg, pat, default = "y"): | ||||
if not self.interactive: return default | ||||
while 1: | ||||
self.write(msg, " ") | ||||
r = self.readline() | ||||
if re.match(pat, r): | ||||
return r | ||||
else: | ||||
self.write("unrecognized response\n") | ||||
def status(self, *msg): | ||||
if not self.quiet: self.write(*msg) | ||||
|
r234 | def warn(self, *msg): | ||
|
r207 | self.write(*msg) | ||
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() | ||||
editor = os.environ.get("HGEDITOR") or os.environ.get("EDITOR", "vi") | ||||
r = os.system("%s %s" % (editor, name)) | ||||
if r: | ||||
raise "Edit failed!" | ||||
t = open(name).read() | ||||
t = re.sub("(?m)^HG:.*\n", "", t) | ||||
return t | ||||