/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to serialize_transform.py

Merge with serialize-transform

Show diffs side-by-side

added added

removed removed

Lines of Context:
serialize_transform.py
1
 
import os
2
 
 
3
 
from bzrlib import errors, multiparent, osutils
4
 
from bzrlib.util import bencode
5
 
 
6
 
 
7
 
def get_parents_texts(tt, trans_id):
8
 
    file_id = tt.tree_file_id(trans_id)
9
 
    try:
10
 
        if file_id is None or tt._tree.kind(file_id) != 'file':
11
 
            return ()
12
 
    except errors.NoSuchFile:
13
 
        return ()
14
 
    return (tt._tree.get_file_text(file_id),)
15
 
 
16
 
 
17
 
def get_parents_lines(tt, trans_id):
18
 
    return tuple(osutils.split_lines(p) for p
19
 
                 in get_parents_texts(tt, trans_id))
20
 
 
21
 
 
22
 
def serialize(tt, serializer):
23
 
    new_name = dict((k, v.encode('utf-8')) for k, v in tt._new_name.items())
24
 
    new_executability = dict((k, int(v)) for k, v in
25
 
                             tt._new_executability.items())
26
 
    tree_path_ids = dict((k.encode('utf-8'), v)
27
 
                         for k, v in tt._tree_path_ids.items())
28
 
    attribs = {
29
 
        '_id_number': tt._id_number,
30
 
        '_new_name': new_name,
31
 
        '_new_parent': tt._new_parent,
32
 
        '_new_executability': new_executability,
33
 
        '_new_id': tt._new_id,
34
 
        '_tree_path_ids': tree_path_ids,
35
 
        '_removed_id': list(tt._removed_id),
36
 
        '_removed_contents': list(tt._removed_contents),
37
 
        '_non_present_ids': tt._non_present_ids,
38
 
        }
39
 
    yield serializer.bytes_record(bencode.bencode(attribs), (('attribs',),))
40
 
    for trans_id, kind in tt._new_contents.items():
41
 
        if kind == 'file':
42
 
            cur_file = open(tt._limbo_name(trans_id), 'rb')
43
 
            try:
44
 
                lines = osutils.split_lines(cur_file.read())
45
 
            finally:
46
 
                cur_file.close()
47
 
            parents = get_parents_lines(tt, trans_id)
48
 
            mpdiff = multiparent.MultiParent.from_lines(lines, parents)
49
 
            content = ''.join(mpdiff.to_patch())
50
 
        if kind == 'directory':
51
 
            content = ''
52
 
        if kind == 'symlink':
53
 
            content = os.readlink(tt._limbo_name(trans_id))
54
 
        yield serializer.bytes_record(content, ((trans_id, kind),))
55
 
 
56
 
 
57
 
def deserialize(tt, records):
58
 
    names, content = records.next()
59
 
    attribs = bencode.bdecode(content)
60
 
    tt._id_number = attribs['_id_number']
61
 
    tt._new_name = dict((k, v.decode('utf-8'))
62
 
                        for k, v in attribs['_new_name'].items())
63
 
    tt._new_parent = attribs['_new_parent']
64
 
    tt._new_executability = dict((k, bool(v)) for k, v in
65
 
        attribs['_new_executability'].items())
66
 
    tt._new_id = attribs['_new_id']
67
 
    tt._r_new_id = dict((v, k) for k, v in tt._new_id.items())
68
 
    tt._tree_path_ids = {}
69
 
    tt._tree_id_paths = {}
70
 
    for bytepath, trans_id in attribs['_tree_path_ids'].items():
71
 
        path = bytepath.decode('utf-8')
72
 
        tt._tree_path_ids[path] = trans_id
73
 
        tt._tree_id_paths[trans_id] = path
74
 
    tt._removed_id = set(attribs['_removed_id'])
75
 
    tt._removed_contents = set(attribs['_removed_contents'])
76
 
    tt._non_present_ids = attribs['_non_present_ids']
77
 
    for ((trans_id, kind),), content in records:
78
 
        if kind == 'file':
79
 
            mpdiff = multiparent.MultiParent.from_patch(content)
80
 
            lines = mpdiff.to_lines(get_parents_texts(tt, trans_id))
81
 
            tt.create_file(lines, trans_id)
82
 
        if kind == 'directory':
83
 
            tt.create_directory(trans_id)
84
 
        if kind == 'symlink':
85
 
            tt.create_symlink(content, trans_id)