Mercurial > public > mercurial-scm > hg-stable
diff hgext/commitextras.py @ 33546:77c0c36654c8
commitextras: move fb extension to core which add extras to a commit
This patch moves the Facebook extension to add extra fields to a commit to a
in-core extension.
author | Pulkit Goyal <7895pulkit@gmail.com> |
---|---|
date | Wed, 12 Jul 2017 00:23:45 +0530 |
parents | |
children | a6af8560494e |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/hgext/commitextras.py Wed Jul 12 00:23:45 2017 +0530 @@ -0,0 +1,45 @@ +# commitextras.py +# +# Copyright 2013 Facebook, Inc. +# +# This software may be used and distributed according to the terms of the +# GNU General Public License version 2 or any later version. + +'''adds a new flag extras to commit''' + +from __future__ import absolute_import + +from mercurial.i18n import _ +from mercurial import ( + commands, + extensions, + registrar, +) + +cmdtable = {} +command = registrar.command(cmdtable) +testedwith = 'ships-with-hg-core' + +def extsetup(ui): + entry = extensions.wrapcommand(commands.table, 'commit', _commit) + options = entry[1] + options.append(('', 'extra', [], + _('set a changeset\'s extra values'), _("KEY=VALUE"))) + +def _commit(orig, ui, repo, *pats, **opts): + origcommit = repo.commit + try: + def _wrappedcommit(*innerpats, **inneropts): + extras = opts.get('extra') + if extras: + for raw in extras: + k, v = raw.split('=', 1) + inneropts['extra'][k] = v + return origcommit(*innerpats, **inneropts) + + # This __dict__ logic is needed because the normal + # extension.wrapfunction doesn't seem to work. + repo.__dict__['commit'] = _wrappedcommit + return orig(ui, repo, *pats, **opts) + finally: + del repo.__dict__['commit']