Mercurial > hg > pyhgsh
annotate mercurial/ui.py @ 1129:ee4f60abad93
Move generating short username to display in hg/hgweb annotate to ui module.
| author | Thomas Arendsen Hein <thomas@intevation.de> |
|---|---|
| date | Sun, 28 Aug 2005 17:29:28 +0200 |
| parents | 8f0ac653f85e |
| children | d32b91ebad5d |
| rev | line source |
|---|---|
| 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.""" |
|
ee4f60abad93
Move generating short username to display in hg/hgweb annotate to ui module.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
1071
diff
changeset
|
83 f = user.find('@') |
|
ee4f60abad93
Move generating short username to display in hg/hgweb annotate to ui module.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
1071
diff
changeset
|
84 if f >= 0: |
|
ee4f60abad93
Move generating short username to display in hg/hgweb annotate to ui module.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
1071
diff
changeset
|
85 user = user[:f] |
|
ee4f60abad93
Move generating short username to display in hg/hgweb annotate to ui module.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
1071
diff
changeset
|
86 f = user.find('<') |
|
ee4f60abad93
Move generating short username to display in hg/hgweb annotate to ui module.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
1071
diff
changeset
|
87 if f >= 0: |
|
ee4f60abad93
Move generating short username to display in hg/hgweb annotate to ui module.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
1071
diff
changeset
|
88 user = user[f+1:] |
|
ee4f60abad93
Move generating short username to display in hg/hgweb annotate to ui module.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
1071
diff
changeset
|
89 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
|
90 |
| 506 | 91 def expandpath(self, loc): |
| 92 paths = {} | |
| 93 for name, path in self.configitems("paths"): | |
| 94 paths[name] = path | |
| 95 | |
| 96 return paths.get(loc, loc) | |
| 97 | |
| 207 | 98 def write(self, *args): |
| 99 for a in args: | |
| 100 sys.stdout.write(str(a)) | |
| 565 | 101 |
| 102 def write_err(self, *args): | |
| 103 sys.stdout.flush() | |
| 104 for a in args: | |
| 105 sys.stderr.write(str(a)) | |
| 106 | |
| 207 | 107 def readline(self): |
| 108 return sys.stdin.readline()[:-1] | |
| 1062 | 109 def prompt(self, msg, pat, default="y"): |
| 207 | 110 if not self.interactive: return default |
| 111 while 1: | |
| 112 self.write(msg, " ") | |
| 113 r = self.readline() | |
| 114 if re.match(pat, r): | |
| 115 return r | |
| 116 else: | |
| 117 self.write("unrecognized response\n") | |
| 118 def status(self, *msg): | |
| 119 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
|
120 def warn(self, *msg): |
| 565 | 121 self.write_err(*msg) |
| 207 | 122 def note(self, *msg): |
| 123 if self.verbose: self.write(*msg) | |
| 124 def debug(self, *msg): | |
| 125 if self.debugflag: self.write(*msg) | |
| 126 def edit(self, text): | |
| 249 | 127 import tempfile |
| 207 | 128 (fd, name) = tempfile.mkstemp("hg") |
| 129 f = os.fdopen(fd, "w") | |
| 130 f.write(text) | |
| 131 f.close() | |
| 132 | |
|
691
61c6b4178b9e
HG environment variables take precedence over hgrc
mpm@selenic.com
parents:
662
diff
changeset
|
133 editor = (os.environ.get("HGEDITOR") or |
|
61c6b4178b9e
HG environment variables take precedence over hgrc
mpm@selenic.com
parents:
662
diff
changeset
|
134 self.config("ui", "editor") or |
|
608
d2994b5298fb
Add username/merge/editor to .hgrc
Matt Mackall <mpm@selenic.com>
parents:
565
diff
changeset
|
135 os.environ.get("EDITOR", "vi")) |
|
d2994b5298fb
Add username/merge/editor to .hgrc
Matt Mackall <mpm@selenic.com>
parents:
565
diff
changeset
|
136 |
|
662
b55a78595ef6
Pass username to hgeditor, remove temporary file
Radoslaw "AstralStorm" Szkodzinski <astralstorm@gorzow.mm.pl>
parents:
613
diff
changeset
|
137 os.environ["HGUSER"] = self.username() |
| 1062 | 138 util.system("%s %s" % (editor, name), errprefix="edit failed") |
| 207 | 139 |
| 140 t = open(name).read() | |
| 141 t = re.sub("(?m)^HG:.*\n", "", t) | |
| 142 | |
|
662
b55a78595ef6
Pass username to hgeditor, remove temporary file
Radoslaw "AstralStorm" Szkodzinski <astralstorm@gorzow.mm.pl>
parents:
613
diff
changeset
|
143 os.unlink(name) |
|
b55a78595ef6
Pass username to hgeditor, remove temporary file
Radoslaw "AstralStorm" Szkodzinski <astralstorm@gorzow.mm.pl>
parents:
613
diff
changeset
|
144 |
| 207 | 145 return t |