diff -r e1c539e1282b -r ba000e29ecf3 hgkw/pretxnkw.py --- a/hgkw/pretxnkw.py Wed Dec 20 14:54:38 2006 +0100 +++ b/hgkw/pretxnkw.py Wed Dec 20 17:41:19 2006 +0100 @@ -2,6 +2,8 @@ from mercurial import cmdutil, commands, util import os.path, re, sys +hgkeywords = 'Id|Header|Author|Date|Revision|RCSFile|Source' + def pretxnkw(ui, repo, hooktype, **args): '''Collects candidates for keyword expansion on commit and expands keywords in working dir. @@ -37,20 +39,31 @@ if not files: # nothing to do return False - cid = args['node'][:12] user, date = repo.changelog.read(repo.changelog.tip())[1:3] - user = util.shortuser(user) - date = util.datestr(date=date, format=util.defaultdateformats[0]) + strdate = util.datestr(date=date) + shortuser = util.shortuser(user) + shortdate = util.datestr(date=date, format=util.defaultdateformats[0]) # %Y-%m-%d %H:%M:%S - re_kw = re.compile(r'\$Hg(: [^$]+?,v [a-z0-9]{12} [^$]+? )?\$') + re_kw = re.compile(r'\$(%s)(: [^$]+? )?\$' % hgkeywords) for f in files: data = repo.wfile(f).read() if not util.binary(data): - kw = '$Hg: %s,v %s %s %s $' % ( - os.path.basename(f), cid, date, user) - data, kwct = re_kw.subn(kw, data) + + def kwexpand(matchobj): + RCSFile = os.path.basename(f)+',v' + Source = os.path.join(repo.root, f)+',v' + Revision = args['node'][:12] + Date = strdate + Author = user + revdateauth = '%s %s %s' % (Revision, shortdate, shortuser) + Header = '%s %s' % (Source, revdateauth) + Id = '%s %s' % (RCSFile, revdateauth) + return '$%s: %s $' % ( + matchobj.group(1), eval(matchobj.group(1))) + + data, kwct = re_kw.subn(kwexpand, data) if kwct: ui.note(_('expanding keywords in %s\n' % f)) # backup file?