def setUp(self):
super(TestStatus, self).setUp()
# The test suite resets after each run, so manually register
# the loom status hook.
try:
from breezy.hooks import install_lazy_named_hook
except ImportError:
pass
else:
from breezy.plugins.loom import show_loom_summary
install_lazy_named_hook('breezy.status', 'hooks', 'post_status',
show_loom_summary, 'loom status')
raise InvalidRevisionId(rev.revision_id, None)
def update_stanza(rev, stanza):
try:
(uuid, branch_path, revno) = extract_svn_foreign_revid(rev)
except InvalidRevisionId:
pass
else:
stanza.add("svn-revno", str(revno))
stanza.add("svn-uuid", uuid)
from breezy.hooks import install_lazy_named_hook
install_lazy_named_hook("breezy.version_info_formats.format_rio",
"RioVersionInfoBuilder.hooks", "revision",
update_stanza, "svn metadata")
install_lazy_named_hook("breezy.info", "hooks", 'repository',
info_svn_repository, "svn repository info")
try:
from breezy.registry import register_lazy
except ImportError:
from breezy.diff import format_registry as diff_format_registry
diff_format_registry.register_lazy('svn', __name__ + '.send',
'SvnDiffTree', 'Subversion diff format')
from breezy.revisionspec import (
revspec_registry, )
revspec_registry.register_lazy("svn:", "breezy.plugins.svn.revspec",
"RevisionSpec_svn")
# hidden power of qbrz ;-)
('breezy.plugins.qbrz.lib.commands', 'cmd_qsubprocess', []),
)
for module, name, aliases in lazy_commands:
plugin_cmds.register_lazy(name, aliases, module)
def post_uncommit_hook(local, master, old_revno, old_tip, new_revno, hook_new_tip):
from breezy.plugins.qbrz.lib.commit_data import QBzrCommitData
branch = local or master
ci_data = QBzrCommitData(branch=branch)
ci_data.set_data_on_uncommit(old_tip, hook_new_tip)
ci_data.save()
try:
from breezy.hooks import install_lazy_named_hook
except ImportError:
from breezy.branch import Branch
Branch.hooks.install_named_hook('post_uncommit', post_uncommit_hook,
'Remember uncomitted revision data for qcommit')
else:
install_lazy_named_hook("breezy.branch", "Branch.hooks", 'post_uncommit',
post_uncommit_hook, 'Remember uncomitted revision data for qcommit')
def load_tests(basic_tests, module, loader):
from breezy.plugins.qbrz.lib.tests import load_tests
return load_tests(basic_tests, module, loader)
from breezy.errors import BzrError
import os
def etckeeper_startcommit_hook(tree):
abspath = getattr(tree, "abspath", None)
if abspath is None or not os.path.exists(abspath(".etckeeper")):
# Only run the commit hook when this is an etckeeper branch
return
import subprocess
ret = subprocess.call(["etckeeper", "pre-commit", abspath(".")])
if ret != 0:
raise BzrError("etckeeper pre-commit failed")
try:
from breezy.hooks import install_lazy_named_hook
except ImportError:
from breezy.mutabletree import MutableTree
MutableTree.hooks.install_named_hook('start_commit',
etckeeper_startcommit_hook, 'etckeeper')
else:
install_lazy_named_hook(
"breezy.mutabletree", "MutableTree.hooks",
'start_commit', etckeeper_startcommit_hook, 'etckeeper')
if __name__ == "__main__":
from distutils.core import setup
setup(name="brz-etckeeper",
packages=["breezy.plugins.etckeeper"],
package_dir={"breezy.plugins.etckeeper":"etckeeper-brz"})
commands.cmd_switch._original_command = breezy.commands.register_command(
getattr(commands, 'cmd_switch'), True)
from breezy.hooks import install_lazy_named_hook
def show_loom_summary(params):
branch = getattr(params.new_tree, "branch", None)
if branch is None:
# Not a working tree, ignore
return
try:
require_loom_branch(branch)
except NotALoom:
return
params.to_file.write('Current thread: %s\n' % branch.nick)
install_lazy_named_hook('breezy.status', 'hooks', 'post_status',
show_loom_summary, 'loom status')
try:
from breezy.registry import register_lazy
except ImportError: # bzr < 2.6
from breezy.revisionspec import revspec_registry
revspec_registry.register_lazy('thread:', 'breezy.plugins.loom.revspec',
'RevisionSpecThread')
revspec_registry.register_lazy('below:', 'breezy.plugins.loom.revspec',
'RevisionSpecBelow')
else:
register_lazy("breezy.revisionspec", "revspec_registry", 'thread:',
'breezy.plugins.loom.revspec', 'RevisionSpecThread')
register_lazy("breezy.revisionspec", "revspec_registry", 'below:',
'breezy.plugins.loom.revspec', 'RevisionSpecBelow')