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
|
import os
from bzrlib import multiparent
from bzrlib.util import bencode
def get_parents_texts(tt, trans_id):
return tuple(''.join(p) for p in get_parents_lines(tt, trans_id))
def get_parents_lines(tt, trans_id):
file_id = tt.tree_file_id(trans_id)
if file_id is None:
return ()
else:
return (tt._tree.get_file(file_id).readlines(),)
def serialize(tt, serializer):
new_name = dict((k, v.encode('utf-8')) for k, v in tt._new_name.items())
new_executability = dict((k, int(v)) for k, v in
tt._new_executability.items())
tree_path_ids = dict((k.encode('utf-8'), v)
for k, v in tt._tree_path_ids.items())
attribs = {
'_id_number': tt._id_number,
'_new_name': new_name,
'_new_parent': tt._new_parent,
'_new_executability': new_executability,
'_new_id': tt._new_id,
'_tree_path_ids': tree_path_ids,
'_removed_id': list(tt._removed_id),
'_removed_contents': list(tt._removed_contents),
'_non_present_ids': tt._non_present_ids,
}
yield serializer.bytes_record(bencode.bencode(attribs), (('attribs',),))
for trans_id, kind in tt._new_contents.items():
if kind == 'file':
cur_file = open(tt._limbo_name(trans_id), 'rb')
try:
lines = cur_file.readlines()
finally:
cur_file.close()
parents = get_parents_lines(tt, trans_id)
mpdiff = multiparent.MultiParent.from_lines(lines, parents)
content = ''.join(mpdiff.to_patch())
if kind == 'directory':
content = ''
if kind == 'symlink':
content = os.readlink(tt._limbo_name(trans_id))
yield serializer.bytes_record(content, ((trans_id, kind),))
def deserialize(tt, records):
names, content = records.next()
attribs = bencode.bdecode(content)
tt._id_number = attribs['_id_number']
tt._new_name = dict((k, v.decode('utf-8'))
for k, v in attribs['_new_name'].items())
tt._new_parent = attribs['_new_parent']
tt._new_executability = dict((k, bool(v)) for k, v in
attribs['_new_executability'].items())
tt._new_id = attribs['_new_id']
tt._r_new_id = dict((v, k) for k, v in tt._new_id.items())
tt._tree_path_ids = {}
tt._tree_id_paths = {}
for bytepath, trans_id in attribs['_tree_path_ids'].items():
path = bytepath.decode('utf-8')
tt._tree_path_ids[path] = trans_id
tt._tree_id_paths[trans_id] = path
tt._removed_id = set(attribs['_removed_id'])
tt._removed_contents = set(attribs['_removed_contents'])
tt._non_present_ids = attribs['_non_present_ids']
for ((trans_id, kind),), content in records:
if kind == 'file':
mpdiff = multiparent.MultiParent.from_patch(content)
lines = mpdiff.to_lines(get_parents_texts(tt, trans_id))
tt.create_file(lines, trans_id)
if kind == 'directory':
tt.create_directory(trans_id)
if kind == 'symlink':
tt.create_symlink(content, trans_id)
|