# HG changeset patch # User Pierre-Yves David # Date 1459400749 25200 # Node ID 54810b543bf4d8bfa24b0867ccfb8689a81e7b95 # Parent 3458aa1027e6744bfc651f3e9beaee9cb07caec7 patch: move setup of import/export logic into a function The function is now called during uisetup. diff -r 3458aa1027e6 -r 54810b543bf4 hgext3rd/topic/__init__.py --- a/hgext3rd/topic/__init__.py Wed Mar 30 22:04:25 2016 -0700 +++ b/hgext3rd/topic/__init__.py Wed Mar 30 22:05:49 2016 -0700 @@ -90,6 +90,7 @@ destination.modsetup(ui) topicrevset.modsetup(ui) discovery.modsetup(ui) + setupimportexport(ui) @contextlib.contextmanager def usetopicmap(repo): @@ -347,11 +348,13 @@ if 'topic' in patchdata: extra['topic'] = patchdata['topic'] -cmdutil.extraexport.append('topic') -cmdutil.extraexportmap['topic'] = _exporttopic -cmdutil.extrapreimport.append('topic') -cmdutil.extrapreimportmap['topic'] = _importtopic -patch.patchheadermap.append(('EXP-Topic', 'topic')) +def setupimportexport(ui): + """run at ui setup time to install import/export logic""" + cmdutil.extraexport.append('topic') + cmdutil.extraexportmap['topic'] = _exporttopic + cmdutil.extrapreimport.append('topic') + cmdutil.extrapreimportmap['topic'] = _importtopic + patch.patchheadermap.append(('EXP-Topic', 'topic')) extensions.afterloaded('rebase', _fixrebase)