/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

  • Committer: Aaron Bentley
  • Date: 2008-10-13 16:46:00 UTC
  • mto: (0.15.25 unshelve)
  • mto: This revision was merged to the branch mainline in revision 3820.
  • Revision ID: aaron@aaronbentley.com-20081013164600-zln9mqci0o02orpi
Allow shelving kind change

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)