author | Bryan O'Sullivan <bos@serpentine.com> |
Sun, 04 Sep 2005 15:47:59 -0700 | |
changeset 1205 | 4003ea658693 |
parent 1147 | d32b91ebad5d |
child 1292 | 141951276ba1 |
permissions | -rw-r--r-- |
207 | 1 |
# ui.py - user interface bits for mercurial |
2 |
# |
|
3 |
# Copyright 2005 Matt Mackall <mpm@selenic.com> |
|
4 |
# |
|
5 |
# This software may be used and distributed according to the terms |
|
6 |
# of the GNU General Public License, incorporated herein by reference. |
|
7 |
||
613
5374955ec5b1
Demand-load most modules in the commands and ui modules.
Bryan O'Sullivan <bos@serpentine.com>
parents:
608
diff
changeset
|
8 |
import os, ConfigParser |
5374955ec5b1
Demand-load most modules in the commands and ui modules.
Bryan O'Sullivan <bos@serpentine.com>
parents:
608
diff
changeset
|
9 |
from demandload import * |
5374955ec5b1
Demand-load most modules in the commands and ui modules.
Bryan O'Sullivan <bos@serpentine.com>
parents:
608
diff
changeset
|
10 |
demandload(globals(), "re socket sys util") |
207 | 11 |
|
12 |
class ui: |
|
13 |
def __init__(self, verbose=False, debug=False, quiet=False, |
|
14 |
interactive=True): |
|
960 | 15 |
self.overlay = {} |
285 | 16 |
self.cdata = ConfigParser.SafeConfigParser() |
951
859de3ebc041
Read global config file /etc/mercurial/hgrc and fix reading hgrc on Windows.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
691
diff
changeset
|
17 |
self.cdata.read([os.path.normpath(hgrc) for hgrc in |
859de3ebc041
Read global config file /etc/mercurial/hgrc and fix reading hgrc on Windows.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
691
diff
changeset
|
18 |
"/etc/mercurial/hgrc", os.path.expanduser("~/.hgrc")]) |
285 | 19 |
|
20 |
self.quiet = self.configbool("ui", "quiet") |
|
21 |
self.verbose = self.configbool("ui", "verbose") |
|
22 |
self.debugflag = self.configbool("ui", "debug") |
|
23 |
self.interactive = self.configbool("ui", "interactive", True) |
|
24 |
||
1071 | 25 |
self.updateopts(verbose, debug, quiet, interactive) |
26 |
||
27 |
def updateopts(self, verbose=False, debug=False, quiet=False, |
|
28 |
interactive=True): |
|
285 | 29 |
self.quiet = (self.quiet or quiet) and not verbose and not debug |
30 |
self.verbose = (self.verbose or verbose) or debug |
|
31 |
self.debugflag = (self.debugflag or debug) |
|
32 |
self.interactive = (self.interactive and interactive) |
|
33 |
||
337 | 34 |
def readconfig(self, fp): |
35 |
self.cdata.readfp(fp) |
|
36 |
||
960 | 37 |
def setconfig(self, section, name, val): |
38 |
self.overlay[(section, name)] = val |
|
39 |
||
40 |
def config(self, section, name, default=None): |
|
41 |
if self.overlay.has_key((section, name)): |
|
42 |
return self.overlay[(section, name)] |
|
43 |
if self.cdata.has_option(section, name): |
|
44 |
return self.cdata.get(section, name) |
|
285 | 45 |
return default |
46 |
||
960 | 47 |
def configbool(self, section, name, default=False): |
48 |
if self.overlay.has_key((section, name)): |
|
49 |
return self.overlay[(section, name)] |
|
50 |
if self.cdata.has_option(section, name): |
|
51 |
return self.cdata.getboolean(section, name) |
|
285 | 52 |
return default |
53 |
||
54 |
def configitems(self, section): |
|
55 |
if self.cdata.has_section(section): |
|
56 |
return self.cdata.items(section) |
|
57 |
return [] |
|
58 |
||
1028
25e7ea0f2cff
Add commands.debugconfig.
Bryan O'Sullivan <bos@serpentine.com>
parents:
981
diff
changeset
|
59 |
def walkconfig(self): |
25e7ea0f2cff
Add commands.debugconfig.
Bryan O'Sullivan <bos@serpentine.com>
parents:
981
diff
changeset
|
60 |
seen = {} |
25e7ea0f2cff
Add commands.debugconfig.
Bryan O'Sullivan <bos@serpentine.com>
parents:
981
diff
changeset
|
61 |
for (section, name), value in self.overlay.iteritems(): |
25e7ea0f2cff
Add commands.debugconfig.
Bryan O'Sullivan <bos@serpentine.com>
parents:
981
diff
changeset
|
62 |
yield section, name, value |
25e7ea0f2cff
Add commands.debugconfig.
Bryan O'Sullivan <bos@serpentine.com>
parents:
981
diff
changeset
|
63 |
seen[section, name] = 1 |
25e7ea0f2cff
Add commands.debugconfig.
Bryan O'Sullivan <bos@serpentine.com>
parents:
981
diff
changeset
|
64 |
for section in self.cdata.sections(): |
25e7ea0f2cff
Add commands.debugconfig.
Bryan O'Sullivan <bos@serpentine.com>
parents:
981
diff
changeset
|
65 |
for name, value in self.cdata.items(section): |
25e7ea0f2cff
Add commands.debugconfig.
Bryan O'Sullivan <bos@serpentine.com>
parents:
981
diff
changeset
|
66 |
if (section, name) in seen: continue |
25e7ea0f2cff
Add commands.debugconfig.
Bryan O'Sullivan <bos@serpentine.com>
parents:
981
diff
changeset
|
67 |
yield section, name, value.replace('\n', '\\n') |
25e7ea0f2cff
Add commands.debugconfig.
Bryan O'Sullivan <bos@serpentine.com>
parents:
981
diff
changeset
|
68 |
seen[section, name] = 1 |
25e7ea0f2cff
Add commands.debugconfig.
Bryan O'Sullivan <bos@serpentine.com>
parents:
981
diff
changeset
|
69 |
|
1071 | 70 |
def extensions(self): |
71 |
return self.configitems("extensions") |
|
72 |
||
608
d2994b5298fb
Add username/merge/editor to .hgrc
Matt Mackall <mpm@selenic.com>
parents:
565
diff
changeset
|
73 |
def username(self): |
691
61c6b4178b9e
HG environment variables take precedence over hgrc
mpm@selenic.com
parents:
662
diff
changeset
|
74 |
return (os.environ.get("HGUSER") or |
61c6b4178b9e
HG environment variables take precedence over hgrc
mpm@selenic.com
parents:
662
diff
changeset
|
75 |
self.config("ui", "username") or |
608
d2994b5298fb
Add username/merge/editor to .hgrc
Matt Mackall <mpm@selenic.com>
parents:
565
diff
changeset
|
76 |
os.environ.get("EMAIL") or |
d2994b5298fb
Add username/merge/editor to .hgrc
Matt Mackall <mpm@selenic.com>
parents:
565
diff
changeset
|
77 |
(os.environ.get("LOGNAME", |
d2994b5298fb
Add username/merge/editor to .hgrc
Matt Mackall <mpm@selenic.com>
parents:
565
diff
changeset
|
78 |
os.environ.get("USERNAME", "unknown")) |
d2994b5298fb
Add username/merge/editor to .hgrc
Matt Mackall <mpm@selenic.com>
parents:
565
diff
changeset
|
79 |
+ '@' + socket.getfqdn())) |
d2994b5298fb
Add username/merge/editor to .hgrc
Matt Mackall <mpm@selenic.com>
parents:
565
diff
changeset
|
80 |
|
1129
ee4f60abad93
Move generating short username to display in hg/hgweb annotate to ui module.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
1071
diff
changeset
|
81 |
def shortuser(self, user): |
ee4f60abad93
Move generating short username to display in hg/hgweb annotate to ui module.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
1071
diff
changeset
|
82 |
"""Return a short representation of a user name or email address.""" |
1147 | 83 |
if not self.verbose: |
84 |
f = user.find('@') |
|
85 |
if f >= 0: |
|
86 |
user = user[:f] |
|
87 |
f = user.find('<') |
|
88 |
if f >= 0: |
|
89 |
user = user[f+1:] |
|
1129
ee4f60abad93
Move generating short username to display in hg/hgweb annotate to ui module.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
1071
diff
changeset
|
90 |
return user |
ee4f60abad93
Move generating short username to display in hg/hgweb annotate to ui module.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
1071
diff
changeset
|
91 |
|
506 | 92 |
def expandpath(self, loc): |
93 |
paths = {} |
|
94 |
for name, path in self.configitems("paths"): |
|
95 |
paths[name] = path |
|
96 |
||
97 |
return paths.get(loc, loc) |
|
98 |
||
207 | 99 |
def write(self, *args): |
100 |
for a in args: |
|
101 |
sys.stdout.write(str(a)) |
|
565 | 102 |
|
103 |
def write_err(self, *args): |
|
104 |
sys.stdout.flush() |
|
105 |
for a in args: |
|
106 |
sys.stderr.write(str(a)) |
|
107 |
||
207 | 108 |
def readline(self): |
109 |
return sys.stdin.readline()[:-1] |
|
1062 | 110 |
def prompt(self, msg, pat, default="y"): |
207 | 111 |
if not self.interactive: return default |
112 |
while 1: |
|
113 |
self.write(msg, " ") |
|
114 |
r = self.readline() |
|
115 |
if re.match(pat, r): |
|
116 |
return r |
|
117 |
else: |
|
118 |
self.write("unrecognized response\n") |
|
119 |
def status(self, *msg): |
|
120 |
if not self.quiet: self.write(*msg) |
|
234
3427806d5ab9
ui.warn can use more than one argument like the other ui methods.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
207
diff
changeset
|
121 |
def warn(self, *msg): |
565 | 122 |
self.write_err(*msg) |
207 | 123 |
def note(self, *msg): |
124 |
if self.verbose: self.write(*msg) |
|
125 |
def debug(self, *msg): |
|
126 |
if self.debugflag: self.write(*msg) |
|
127 |
def edit(self, text): |
|
249 | 128 |
import tempfile |
207 | 129 |
(fd, name) = tempfile.mkstemp("hg") |
130 |
f = os.fdopen(fd, "w") |
|
131 |
f.write(text) |
|
132 |
f.close() |
|
133 |
||
691
61c6b4178b9e
HG environment variables take precedence over hgrc
mpm@selenic.com
parents:
662
diff
changeset
|
134 |
editor = (os.environ.get("HGEDITOR") or |
61c6b4178b9e
HG environment variables take precedence over hgrc
mpm@selenic.com
parents:
662
diff
changeset
|
135 |
self.config("ui", "editor") or |
608
d2994b5298fb
Add username/merge/editor to .hgrc
Matt Mackall <mpm@selenic.com>
parents:
565
diff
changeset
|
136 |
os.environ.get("EDITOR", "vi")) |
d2994b5298fb
Add username/merge/editor to .hgrc
Matt Mackall <mpm@selenic.com>
parents:
565
diff
changeset
|
137 |
|
662
b55a78595ef6
Pass username to hgeditor, remove temporary file
Radoslaw "AstralStorm" Szkodzinski <astralstorm@gorzow.mm.pl>
parents:
613
diff
changeset
|
138 |
os.environ["HGUSER"] = self.username() |
1062 | 139 |
util.system("%s %s" % (editor, name), errprefix="edit failed") |
207 | 140 |
|
141 |
t = open(name).read() |
|
142 |
t = re.sub("(?m)^HG:.*\n", "", t) |
|
143 |
||
662
b55a78595ef6
Pass username to hgeditor, remove temporary file
Radoslaw "AstralStorm" Szkodzinski <astralstorm@gorzow.mm.pl>
parents:
613
diff
changeset
|
144 |
os.unlink(name) |
b55a78595ef6
Pass username to hgeditor, remove temporary file
Radoslaw "AstralStorm" Szkodzinski <astralstorm@gorzow.mm.pl>
parents:
613
diff
changeset
|
145 |
|
207 | 146 |
return t |