9
9
# publish, distribute, sublicense, and/or sell copies of the Software,
10
10
# and to permit persons to whom the Software is furnished to do so,
11
11
# subject to the following conditions:
13
13
# The above copyright notice and this permission notice shall be
14
14
# included in all copies or substantial portions of the Software.
16
# The Software is provided "AS IS", without warranty of any kind,
17
# express or implied, including but not limited to the warranties of
18
# merchantability, fitness for a particular purpose and
19
# noninfringement. In no event shall the authors or copyright holders
20
# be liable for any claim, damages or other liability, whether in an
21
# action of contract, tort or otherwise, arising from, out of or in
22
# connection with the Software or the use or other dealings in the
27
newf = x.index('e', f)
30
except (OverflowError, ValueError):
35
elif x[f] == '0' and newf != f+1:
39
def decode_string(x, f):
40
colon = x.index(':', f)
43
except (OverflowError, ValueError):
45
if x[f] == '0' and colon != f+1:
48
return (x[colon:colon+n], colon+n)
50
def decode_list(x, f):
53
v, f = decode_func[x[f]](x, f)
57
def decode_dict(x, f):
61
k, f = decode_string(x, f)
65
r[k], f = decode_func[x[f]](x, f)
69
decode_func['l'] = decode_list
70
decode_func['d'] = decode_dict
71
decode_func['i'] = decode_int
72
decode_func['0'] = decode_string
73
decode_func['1'] = decode_string
74
decode_func['2'] = decode_string
75
decode_func['3'] = decode_string
76
decode_func['4'] = decode_string
77
decode_func['5'] = decode_string
78
decode_func['6'] = decode_string
79
decode_func['7'] = decode_string
80
decode_func['8'] = decode_string
81
decode_func['9'] = decode_string
85
r, l = decode_func[x[0]](x, 0)
86
except (IndexError, KeyError):
104
bdecode('i341foo382e')
108
assert bdecode('i4e') == 4L
109
assert bdecode('i0e') == 0L
110
assert bdecode('i123456789e') == 123456789L
111
assert bdecode('i-10e') == -10L
133
bdecode('35208734823ljdahflajhdf')
138
bdecode('2:abfdjslhfld')
142
assert bdecode('0:') == ''
143
assert bdecode('3:abc') == 'abc'
144
assert bdecode('10:1234567890') == '1234567890'
155
assert bdecode('le') == []
157
bdecode('leanfdldjfh')
161
assert bdecode('l0:0:0:e') == ['', '', '']
163
bdecode('relwjhrlewjh')
167
assert bdecode('li1ei2ei3ee') == [1, 2, 3]
168
assert bdecode('l3:asd2:xye') == ['asd', 'xy']
169
assert bdecode('ll5:Alice3:Bobeli2ei3eee') == [['Alice', 'Bob'], [2, 3]]
180
assert bdecode('de') == {}
181
assert bdecode('d3:agei25e4:eyes4:bluee') == {'age': 25, 'eyes': 'blue'}
182
assert bdecode('d8:spam.mp3d6:author5:Alice6:lengthi100000eee') == {'spam.mp3': {'author': 'Alice', 'length': 100000}}
194
bdecode('d1:b0:1:a0:e')
199
bdecode('d1:a0:1:a0:e')
16
# Modifications copyright (C) 2008 Canonical Ltd
18
class BDecoder(object):
20
def __init__(self, yield_tuples=False):
23
:param yield_tuples: if true, decode "l" elements as tuples rather than
26
self.yield_tuples = yield_tuples
28
decode_func['l'] = self.decode_list
29
decode_func['d'] = self.decode_dict
30
decode_func['i'] = self.decode_int
31
decode_func['0'] = self.decode_string
32
decode_func['1'] = self.decode_string
33
decode_func['2'] = self.decode_string
34
decode_func['3'] = self.decode_string
35
decode_func['4'] = self.decode_string
36
decode_func['5'] = self.decode_string
37
decode_func['6'] = self.decode_string
38
decode_func['7'] = self.decode_string
39
decode_func['8'] = self.decode_string
40
decode_func['9'] = self.decode_string
41
self.decode_func = decode_func
43
def decode_int(self, x, f):
45
newf = x.index('e', f)
48
except (OverflowError, ValueError):
53
elif x[f] == '0' and newf != f+1:
57
def decode_string(self, x, f):
58
colon = x.index(':', f)
61
except (OverflowError, ValueError):
63
if x[f] == '0' and colon != f+1:
66
return (x[colon:colon+n], colon+n)
68
def decode_list(self, x, f):
71
v, f = self.decode_func[x[f]](x, f)
77
def decode_dict(self, x, f):
81
k, f = self.decode_string(x, f)
85
r[k], f = self.decode_func[x[f]](x, f)
92
r, l = self.decode_func[x[0]](x, 0)
93
except (IndexError, KeyError):
100
_decoder = BDecoder()
101
bdecode = _decoder.bdecode
103
_tuple_decoder = BDecoder(True)
104
bdecode_as_tuple = _tuple_decoder.bdecode
250
107
from types import StringType, IntType, LongType, DictType, ListType, TupleType
292
149
from types import BooleanType
293
encode_func[BooleanType] = encode_int
153
def encode_bool(x,r):
154
encode_int(int(x), r)
155
encode_func[BooleanType] = encode_bool
157
from bzrlib._static_tuple_py import StaticTuple
158
encode_func[StaticTuple] = encode_list
160
from bzrlib._static_tuple_c import StaticTuple
164
encode_func[StaticTuple] = encode_list
299
169
encode_func[type(x)](x, r)
300
170
return ''.join(r)
303
assert bencode(4) == 'i4e'
304
assert bencode(0) == 'i0e'
305
assert bencode(-10) == 'i-10e'
306
assert bencode(12345678901234567890L) == 'i12345678901234567890e'
307
assert bencode('') == '0:'
308
assert bencode('abc') == '3:abc'
309
assert bencode('1234567890') == '10:1234567890'
310
assert bencode([]) == 'le'
311
assert bencode([1, 2, 3]) == 'li1ei2ei3ee'
312
assert bencode([['Alice', 'Bob'], [2, 3]]) == 'll5:Alice3:Bobeli2ei3eee'
313
assert bencode({}) == 'de'
314
assert bencode({'age': 25, 'eyes': 'blue'}) == 'd3:agei25e4:eyes4:bluee'
315
assert bencode({'spam.mp3': {'author': 'Alice', 'length': 100000}}) == 'd8:spam.mp3d6:author5:Alice6:lengthi100000eee'
316
assert bencode(Bencached(bencode(3))) == 'i3e'