← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/_static_tuple_py.py

Turn completion assertions into separate methods.

Many common assertions used to be expressed as arguments to the complete
method.  This makes the checks more explicit, and the code easier to read.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/_static_tuple_py.py
1
 
# Copyright (C) 2009, 2010 Canonical Ltd
2
 
#
3
 
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
7
 
#
8
 
# This program is distributed in the hope that it will be useful,
9
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
# GNU General Public License for more details.
12
 
#
13
 
# You should have received a copy of the GNU General Public License
14
 
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
17
 
"""The pure-python implementation of the StaticTuple type.
18
 
 
19
 
Note that it is generally just implemented as using tuples of tuples of
20
 
strings.
21
 
"""
22
 
 
23
 
 
24
 
class StaticTuple(tuple):
25
 
    """A static type, similar to a tuple of strings."""
26
 
 
27
 
    __slots__ = ()
28
 
 
29
 
    def __new__(cls, *args):
30
 
        # Make the empty StaticTuple a singleton
31
 
        if not args and _empty_tuple is not None:
32
 
            return _empty_tuple
33
 
        return tuple.__new__(cls, args)
34
 
 
35
 
    def __init__(self, *args):
36
 
        """Create a new 'StaticTuple'"""
37
 
        num_keys = len(args)
38
 
        if num_keys < 0 or num_keys > 255:
39
 
            raise TypeError('StaticTuple(...) takes from 0 to 255 items')
40
 
        for bit in args:
41
 
            if type(bit) not in (str, StaticTuple, unicode, int, long, float,
42
 
                                 None.__class__, bool):
43
 
                raise TypeError('StaticTuple can only point to'
44
 
                    ' StaticTuple, str, unicode, int, long, float, bool, or'
45
 
                    ' None not %s' % (type(bit),))
46
 
        # We don't need to pass args to tuple.__init__, because that was
47
 
        # already handled in __new__.
48
 
        tuple.__init__(self)
49
 
 
50
 
    def __repr__(self):
51
 
        return '%s%s' % (self.__class__.__name__, tuple.__repr__(self))
52
 
 
53
 
    def __reduce__(self):
54
 
        return (StaticTuple, tuple(self))
55
 
 
56
 
    def __add__(self, other):
57
 
        """Concatenate self with other"""
58
 
        return StaticTuple.from_sequence(tuple.__add__(self,other))
59
 
 
60
 
    def as_tuple(self):
61
 
        return tuple(self)
62
 
 
63
 
    def intern(self):
64
 
        return _interned_tuples.setdefault(self, self)
65
 
 
66
 
    @staticmethod
67
 
    def from_sequence(seq):
68
 
        """Convert a sequence object into a StaticTuple instance."""
69
 
        if isinstance(seq, StaticTuple):
70
 
            # it already is
71
 
            return seq
72
 
        return StaticTuple(*seq)
73
 
 
74
 
 
75
 
 
76
 
# Have to set it to None first, so that __new__ can determine whether
77
 
# the _empty_tuple singleton has been created yet or not.
78
 
_empty_tuple = None
79
 
_empty_tuple = StaticTuple()
80
 
_interned_tuples = {}