1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
# Copyright (C) 2010 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
"""The core state needed to make use of bzr is managed here."""
from __future__ import absolute_import
__all__ = [
'BzrLibraryState',
]
import breezy
from .lazy_import import lazy_import
lazy_import(globals(), """
from breezy import (
cleanup,
config,
osutils,
symbol_versioning,
trace,
ui,
)
""")
class BzrLibraryState(object):
"""The state about how breezy has been configured.
This is the core state needed to make use of bzr. The current instance is
currently always exposed as breezy._global_state, but we desired to move
to a point where no global state is needed at all.
:ivar cleanups: An ObjectWithCleanups which can be used for cleanups that
should occur when the use of breezy is completed. This is initialised
in __enter__ and executed in __exit__.
"""
def __init__(self, ui, trace):
"""Create library start for normal use of breezy.
Most applications that embed breezy, including bzr itself, should just
call breezy.initialize(), but it is possible to use the state class
directly. The initialize() function provides sensible defaults for a
CLI program, such as a text UI factory.
More options may be added in future so callers should use named
arguments.
BzrLibraryState implements the Python 2.5 Context Manager protocol
PEP343, and can be used with the with statement. Upon __enter__ the
global variables in use by bzr are set, and they are cleared on
__exit__.
:param ui: A breezy.ui.ui_factory to use.
:param trace: A breezy.trace.Config context manager to use, perhaps
breezy.trace.DefaultConfig.
"""
self._ui = ui
self._trace = trace
# There is no overrides by default, they are set later when the command
# arguments are parsed.
self.cmdline_overrides = config.CommandLineStore()
# No config stores are cached to start with
self.config_stores = {} # By url
self.started = False
def __enter__(self):
if not self.started:
self._start()
return self # This is bound to the 'as' clause in a with statement.
def _start(self):
"""Do all initialization."""
# NB: This function tweaks so much global state it's hard to test it in
# isolation within the same interpreter. It's not reached on normal
# in-process run_bzr calls. If it's broken, we expect that
# TestRunBzrSubprocess may fail.
self.cleanups = cleanup.ObjectWithCleanups()
if breezy.version_info[3] == 'final':
self.cleanups.add_cleanup(
symbol_versioning.suppress_deprecation_warnings(override=True))
self._trace.__enter__()
self._orig_ui = breezy.ui.ui_factory
breezy.ui.ui_factory = self._ui
self._ui.__enter__()
if breezy._global_state is not None:
raise RuntimeError("Breezy already initialized")
breezy._global_state = self
self.started = True
def __exit__(self, exc_type, exc_val, exc_tb):
if exc_type is None:
# Save config changes
for k, store in self.config_stores.items():
store.save_changes()
self.cleanups.cleanup_now()
trace._flush_stdout_stderr()
trace._flush_trace()
osutils.report_extension_load_failures()
self._ui.__exit__(None, None, None)
self._trace.__exit__(None, None, None)
ui.ui_factory = self._orig_ui
breezy._global_state = None
return False # propogate exceptions.
|