~bzr-pqm/bzr/bzr.dev

# \subsection{Example usage}

# 

# \textbf{XXX:} Move these to object serialization code. 

def write_revision(writer, revision):

    s = Stanza(revision=revision.revision_id,

               committer=revision.committer, 

               timezone=long(revision.timezone),

               timestamp=long(revision.timestamp),

               inventory_sha1=revision.inventory_sha1,

               message=revision.message)

    for parent_id in revision.parent_ids:

        s.add('parent', parent_id)

    for prop_name, prop_value in revision.properties.items():

        s.add(prop_name, prop_value)

    writer.write_stanza(s)

def write_inventory(writer, inventory):

    s = Stanza(inventory_version=7)

    writer.write_stanza(s)

    for path, ie in inventory.iter_entries():

        s = Stanza()

        s.add(ie.kind, ie.file_id)

        for attr in ['name', 'parent_id', 'revision', \

                     'text_sha1', 'text_size', 'executable', \

                     'symlink_target', \

                     ]:

            attr_val = getattr(ie, attr, None)

            if attr == 'executable' and attr_val == 0:

                continue

            if attr == 'parent_id' and attr_val == 'TREE_ROOT':

                continue

            if attr_val is not None:

                s.add(attr, attr_val)

        writer.write_stanza(s)

def read_inventory(inv_file):

    """Read inventory object from rio formatted inventory file"""

    from bzrlib.inventory import Inventory, InventoryFile

    s = read_stanza(inv_file)

    assert s['inventory_version'] == 7

    inv = Inventory()

    for s in read_stanzas(inv_file):

        kind, file_id = s.items[0]

        parent_id = None

        if 'parent_id' in s:

            parent_id = s['parent_id']

        if kind == 'file':

            ie = InventoryFile(file_id, s['name'], parent_id)

            ie.text_sha1 = s['text_sha1']

            ie.text_size = s['text_size']

        else:

            raise NotImplementedError()

        inv.add(ie)

    return inv