~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/__init__.py

  • Committer: Jelmer Vernooij
  • Date: 2012-01-06 22:53:14 UTC
  • mto: (6437.3.11 2.5)
  • mto: This revision was merged to the branch mainline in revision 6444.
  • Revision ID: jelmer@samba.org-20120106225314-0lb0xapr27sctadz
Add tests.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# (C) 2005 Canonical Development Ltd
2
 
 
 
1
# Copyright (C) 2005-2011 Canonical Ltd
 
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
 
 
7
#
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
12
 
 
 
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
 
 
17
 
"""bzr library"""
18
 
 
19
 
from inventory import Inventory, InventoryEntry
20
 
from branch import Branch, ScratchBranch, find_branch
21
 
from osutils import format_date
22
 
from tree import Tree
23
 
from diff import compare_trees
24
 
from trace import mutter, warning, open_tracefile
25
 
from log import show_log
26
 
import add
27
 
 
28
 
BZRDIR = ".bzr"
29
 
 
30
 
DEFAULT_IGNORE = ['.bzr.log',
31
 
                  '*~', '#*#', '*$', '.#*',
32
 
                  '.*.sw[nop]', '.*.tmp',
33
 
                  '*.tmp', '*.bak', '*.BAK', '*.orig',
34
 
                  '*.o', '*.obj', '*.a', '*.py[oc]', '*.so', '*.exe', '*.elc', 
35
 
                  '{arch}', 'CVS', 'CVS.adm', '.svn', '_darcs', 'SCCS', 'RCS',
36
 
                  '*,v',
37
 
                  'BitKeeper',
38
 
                  '.git',
39
 
                  'TAGS', '.make.state', '.sconsign', '.tmp*',
40
 
                  '.del-*']
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
"""All of bzr.
 
18
 
 
19
Developer documentation is available at
 
20
http://doc.bazaar.canonical.com/bzr.dev/developers/
 
21
 
 
22
The project website is at http://bazaar.canonical.com/
 
23
 
 
24
Some particularly interesting things in bzrlib are:
 
25
 
 
26
 * bzrlib.initialize -- setup the library for use
 
27
 * bzrlib.plugin.load_plugins -- load all installed plugins
 
28
 * bzrlib.branch.Branch.open -- open a branch
 
29
 * bzrlib.workingtree.WorkingTree.open -- open a working tree
 
30
 
 
31
We hope you enjoy this library.
 
32
"""
 
33
 
 
34
from __future__ import absolute_import
 
35
 
 
36
import time
 
37
 
 
38
# Keep track of when bzrlib was first imported, so that we can give rough
 
39
# timestamps relative to program start in the log file kept by bzrlib.trace.
 
40
_start_time = time.time()
 
41
 
 
42
import codecs
 
43
import sys
 
44
 
41
45
 
42
46
IGNORE_FILENAME = ".bzrignore"
43
47
 
44
 
import locale
45
 
user_encoding = locale.getpreferredencoding()
46
 
del locale
47
 
 
48
 
__copyright__ = "Copyright 2005 Canonical Development Ltd."
49
 
__author__ = "Martin Pool <mbp@canonical.com>"
50
 
__version__ = '0.0.5'
51
 
 
52
 
 
53
 
def get_bzr_revision():
54
 
    """If bzr is run from a branch, return (revno,revid) or None"""
55
 
    from errors import BzrError
 
48
 
 
49
__copyright__ = "Copyright 2005-2011 Canonical Ltd."
 
50
 
 
51
# same format as sys.version_info: "A tuple containing the five components of
 
52
# the version number: major, minor, micro, releaselevel, and serial. All
 
53
# values except releaselevel are integers; the release level is 'alpha',
 
54
# 'beta', 'candidate', or 'final'. The version_info value corresponding to the
 
55
# Python version 2.0 is (2, 0, 0, 'final', 0)."  Additionally we use a
 
56
# releaselevel of 'dev' for unreleased under-development code.
 
57
 
 
58
version_info = (2, 5, 0, 'dev', 5)
 
59
 
 
60
# API compatibility version
 
61
api_minimum_version = (2, 4, 0)
 
62
 
 
63
 
 
64
def _format_version_tuple(version_info):
 
65
    """Turn a version number 2, 3 or 5-tuple into a short string.
 
66
 
 
67
    This format matches <http://docs.python.org/dist/meta-data.html>
 
68
    and the typical presentation used in Python output.
 
69
 
 
70
    This also checks that the version is reasonable: the sub-release must be
 
71
    zero for final releases.
 
72
 
 
73
    >>> print _format_version_tuple((1, 0, 0, 'final', 0))
 
74
    1.0.0
 
75
    >>> print _format_version_tuple((1, 2, 0, 'dev', 0))
 
76
    1.2.0dev
 
77
    >>> print _format_version_tuple((1, 2, 0, 'dev', 1))
 
78
    1.2.0dev1
 
79
    >>> print _format_version_tuple((1, 1, 1, 'candidate', 2))
 
80
    1.1.1rc2
 
81
    >>> print _format_version_tuple((2, 1, 0, 'beta', 1))
 
82
    2.1b1
 
83
    >>> print _format_version_tuple((1, 4, 0))
 
84
    1.4.0
 
85
    >>> print _format_version_tuple((1, 4))
 
86
    1.4
 
87
    >>> print _format_version_tuple((2, 1, 0, 'final', 42))
 
88
    2.1.0.42
 
89
    >>> print _format_version_tuple((1, 4, 0, 'wibble', 0))
 
90
    1.4.0.wibble.0
 
91
    """
 
92
    if len(version_info) == 2:
 
93
        main_version = '%d.%d' % version_info[:2]
 
94
    else:
 
95
        main_version = '%d.%d.%d' % version_info[:3]
 
96
    if len(version_info) <= 3:
 
97
        return main_version
 
98
 
 
99
    release_type = version_info[3]
 
100
    sub = version_info[4]
 
101
 
 
102
    if release_type == 'final' and sub == 0:
 
103
        sub_string = ''
 
104
    elif release_type == 'final':
 
105
        sub_string = '.' + str(sub)
 
106
    elif release_type == 'dev' and sub == 0:
 
107
        sub_string = 'dev'
 
108
    elif release_type == 'dev':
 
109
        sub_string = 'dev' + str(sub)
 
110
    elif release_type in ('alpha', 'beta'):
 
111
        if version_info[2] == 0:
 
112
            main_version = '%d.%d' % version_info[:2]
 
113
        sub_string = release_type[0] + str(sub)
 
114
    elif release_type == 'candidate':
 
115
        sub_string = 'rc' + str(sub)
 
116
    else:
 
117
        return '.'.join(map(str, version_info))
 
118
 
 
119
    return main_version + sub_string
 
120
 
 
121
 
 
122
# lazy_regex import must be done after _format_version_tuple definition
 
123
# to avoid "no attribute '_format_version_tuple'" error when using
 
124
# deprecated_function in the lazy_regex module.
 
125
if getattr(sys, '_bzr_lazy_regex', False):
 
126
    # The 'bzr' executable sets _bzr_lazy_regex.  We install the lazy regex
 
127
    # hack as soon as possible so that as much of the standard library can
 
128
    # benefit, including the 'string' module.
 
129
    del sys._bzr_lazy_regex
 
130
    import bzrlib.lazy_regex
 
131
    bzrlib.lazy_regex.install_lazy_compile()
 
132
 
 
133
 
 
134
__version__ = _format_version_tuple(version_info)
 
135
version_string = __version__
 
136
 
 
137
 
 
138
def _patch_filesystem_default_encoding(new_enc):
 
139
    """Change the Python process global encoding for filesystem names
 
140
    
 
141
    The effect is to change how open() and other builtin functions handle
 
142
    unicode filenames on posix systems. This should only be done near startup.
 
143
 
 
144
    The new encoding string passed to this function must survive until process
 
145
    termination, otherwise the interpreter may access uninitialized memory.
 
146
    The use of intern() may defer breakage is but is not enough, the string
 
147
    object should be secure against module reloading and during teardown.
 
148
    """
56
149
    try:
57
 
        branch = Branch(__path__[0])
58
 
        rh = branch.revision_history()
59
 
        if rh:
60
 
            return len(rh), rh[-1]
61
 
        else:
62
 
            return None
63
 
    except BzrError:
64
 
        return None
65
 
    
 
150
        import ctypes
 
151
    except ImportError:
 
152
        return
 
153
    pythonapi = getattr(ctypes, "pythonapi", None)
 
154
    if pythonapi is None:
 
155
        # Not CPython ctypes implementation
 
156
        return
 
157
    old_ptr = ctypes.c_void_p.in_dll(pythonapi, "Py_FileSystemDefaultEncoding")
 
158
    new_ptr = ctypes.cast(ctypes.c_char_p(intern(new_enc)), ctypes.c_void_p)
 
159
    old_ptr.value = new_ptr.value
 
160
    if sys.getfilesystemencoding() != new_enc:
 
161
        raise RuntimeError("Failed to change the filesystem default encoding")
 
162
    return new_enc
 
163
 
 
164
 
 
165
# When running under the bzr script, override bad filesystem default encoding.
 
166
# This is not safe to do for all users of bzrlib, other scripts should instead
 
167
# just ensure a usable locale is set via the $LANG variable on posix systems.
 
168
_fs_enc = sys.getfilesystemencoding()
 
169
if getattr(sys, "_bzr_default_fs_enc", None) is not None:
 
170
    if (_fs_enc is None or codecs.lookup(_fs_enc).name == "ascii"):
 
171
        _fs_enc = _patch_filesystem_default_encoding(sys._bzr_default_fs_enc)
 
172
if _fs_enc is None:
 
173
    _fs_enc = "ascii"
 
174
else:
 
175
    _fs_enc = codecs.lookup(_fs_enc).name
 
176
 
 
177
 
 
178
# bzr has various bits of global state that are slowly being eliminated.
 
179
# This variable is intended to permit any new state-like things to be attached
 
180
# to a library_state.BzrLibraryState object rather than getting new global
 
181
# variables that need to be hunted down. Accessing the current BzrLibraryState
 
182
# through this variable is not encouraged: it is better to pass it around as
 
183
# part of the context of an operation than to look it up directly, but when
 
184
# that is too hard, it is better to use this variable than to make a brand new
 
185
# global variable.
 
186
# If using this variable by looking it up (because it can't be easily obtained)
 
187
# it is important to store the reference you get, rather than looking it up
 
188
# repeatedly; that way your code will behave properly in the bzrlib test suite
 
189
# and from programs that do use multiple library contexts.
 
190
global_state = None
 
191
 
 
192
 
 
193
def initialize(setup_ui=True, stdin=None, stdout=None, stderr=None):
 
194
    """Set up everything needed for normal use of bzrlib.
 
195
 
 
196
    Most applications that embed bzrlib, including bzr itself, should call
 
197
    this function to initialize various subsystems.  
 
198
 
 
199
    More options may be added in future so callers should use named arguments.
 
200
 
 
201
    The object returned by this function can be used as a contex manager
 
202
    through the 'with' statement to automatically shut down when the process
 
203
    is finished with bzrlib.  However (from bzr 2.4) it's not necessary to
 
204
    separately enter the context as well as starting bzr: bzrlib is ready to
 
205
    go when this function returns.
 
206
 
 
207
    :param setup_ui: If true (default) use a terminal UI; otherwise 
 
208
        some other ui_factory must be assigned to `bzrlib.ui.ui_factory` by
 
209
        the caller.
 
210
    :param stdin, stdout, stderr: If provided, use these for terminal IO;
 
211
        otherwise use the files in `sys`.
 
212
    :return: A context manager for the use of bzrlib. The __exit__
 
213
        should be called by the caller before exiting their process or
 
214
        otherwise stopping use of bzrlib. Advanced callers can use
 
215
        BzrLibraryState directly.
 
216
    """
 
217
    from bzrlib import library_state, trace
 
218
    if setup_ui:
 
219
        import bzrlib.ui
 
220
        stdin = stdin or sys.stdin
 
221
        stdout = stdout or sys.stdout
 
222
        stderr = stderr or sys.stderr
 
223
        ui_factory = bzrlib.ui.make_ui_for_terminal(stdin, stdout, stderr)
 
224
    else:
 
225
        ui_factory = None
 
226
    tracer = trace.DefaultConfig()
 
227
    state = library_state.BzrLibraryState(ui=ui_factory, trace=tracer)
 
228
    # Start automatically in case people don't realize this returns a context.
 
229
    state._start()
 
230
    return state
 
231
 
 
232
 
 
233
def test_suite():
 
234
    import tests
 
235
    return tests.test_suite()