/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 roundtrip.py

  • Committer: Jelmer Vernooij
  • Date: 2010-04-30 23:03:01 UTC
  • mto: (0.200.912 trunk)
  • mto: This revision was merged to the branch mainline in revision 6960.
  • Revision ID: jelmer@samba.org-20100430230301-w4xncjn5svh2ghl4
Support storing revision id data.

Show diffs side-by-side

added added

removed removed

Lines of Context:
roundtrip.py
22
22
 
23
23
class BzrGitRevisionMetadata(object):
24
24
    """Metadata for a Bazaar revision roundtripped into Git.
25
 
 
 
25
    
26
26
    :ivar revision_id: Revision id, as string
27
27
    :ivar properties: Revision properties, as dictionary
28
 
    :ivar explicit_parent_ids: Parent ids (needed if there are ghosts)
29
 
    :ivar verifiers: Verifier information
 
28
    :ivar file_ids: File ids, as map from path -> file id
30
29
    """
31
30
 
32
31
    revision_id = None
33
32
 
34
 
    explicit_parent_ids = None
35
 
 
36
 
    verifiers = {}
37
 
 
38
 
    def __init__(self):
39
 
        self.properties = {}
40
 
 
41
 
    def __nonzero__(self):
42
 
        return bool(self.revision_id or self.properties)
 
33
    file_ids = {}
 
34
 
 
35
    properties = {}
43
36
 
44
37
 
45
38
def parse_roundtripping_metadata(text):
50
43
        (key, value) = l.split(":", 1)
51
44
        if key == "revision-id":
52
45
            ret.revision_id = value.strip()
53
 
        elif key == "parent-ids":
54
 
            ret.explicit_parent_ids = tuple(value.strip().split(" "))
55
 
        elif key == "testament3-sha1":
56
 
            ret.verifiers["testament3-sha1"] = value.strip()
57
 
        elif key.startswith("property-"):
58
 
            ret.properties[key[len("property-"):]] = value[1:].rstrip("\n")
59
46
        else:
60
47
            raise ValueError
61
48
    return ret
62
 
 
63
 
 
64
 
def generate_roundtripping_metadata(metadata, encoding):
65
 
    """Serialize the roundtripping metadata.
66
 
 
67
 
    :param metadata: A `BzrGitRevisionMetadata` instance
68
 
    :return: String with revision metadata
69
 
    """
70
 
    lines = []
71
 
    if metadata.revision_id:
72
 
        lines.append("revision-id: %s\n" % metadata.revision_id)
73
 
    if metadata.explicit_parent_ids:
74
 
        lines.append("parent-ids: %s\n" % " ".join(metadata.explicit_parent_ids))
75
 
    for key in sorted(metadata.properties.keys()):
76
 
        lines.append("property-%s: %s\n" % (key.encode(encoding), metadata.properties[key].encode(encoding)))
77
 
    if "testament3-sha1" in metadata.verifiers:
78
 
        lines.append("testament3-sha1: %s\n" %
79
 
                     metadata.verifiers["testament3-sha1"])
80
 
    return "".join(lines)
81
 
 
82
 
 
83
 
def extract_bzr_metadata(message):
84
 
    """Extract Bazaar metadata from a commit message.
85
 
 
86
 
    :param message: Commit message to extract from
87
 
    :return: Tuple with original commit message and metadata object
88
 
    """
89
 
    split = message.split("\n--BZR--\n", 1)
90
 
    if len(split) != 2:
91
 
        return message, None
92
 
    return split[0], parse_roundtripping_metadata(split[1])
93
 
 
94
 
 
95
 
def inject_bzr_metadata(message, metadata, encoding):
96
 
    if not metadata:
97
 
        return message
98
 
    rt_data = generate_roundtripping_metadata(metadata, encoding)
99
 
    if not rt_data:
100
 
        return message
101
 
    assert type(rt_data) == str
102
 
    return message + "\n--BZR--\n" + rt_data
103
 
 
104
 
 
105
 
def serialize_fileid_map(file_ids):
106
 
    """Serialize a file id map."""
107
 
    lines = []
108
 
    for path in sorted(file_ids.keys()):
109
 
        lines.append("%s\0%s\n" % (path, file_ids[path]))
110
 
    return lines
111
 
 
112
 
 
113
 
def deserialize_fileid_map(filetext):
114
 
    """Deserialize a file id map."""
115
 
    ret = {}
116
 
    f = StringIO(filetext)
117
 
    lines = f.readlines()
118
 
    for l in lines:
119
 
        (path, file_id) = l.rstrip("\n").split("\0")
120
 
        ret[path] = file_id
121
 
    return ret