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 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98
|
#!/usr/bin/env python
# -*- encoding: utf-8 -*-
# Copyright (C) 2007-2009 Guillermo Gonzalez
#
# The code taken from bzrlib is under: Copyright (C) 2005-2007 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
#
# Contributors:
# Martin Albisetti
"""
This code is a modified copy from bzrlib.annotate
(see there for copyrights and licensing)
"""
import sys
_annotate_file = None
try:
from bzrlib.annotate import _expand_annotations
except ImportError:
# to support bzr < 1.8
from bzrlib.annotate import _annotate_file
from bzrlib.annotate import _annotations
from bzrlib import osutils
from writer import _escape_cdata
empty_annotation = 'revno="" author="" date=""'
def annotate_file_xml(branch, rev_id, file_id, to_file=None,
show_ids=False, wt_root_path=None, file_path=None, full=False):
"""custom annotate_file that spits xml """
if to_file is None:
to_file = sys.stdout
encoding = getattr(to_file, 'encoding', None) or \
osutils.get_terminal_encoding()
prevanno = ''
last_rev_id = None
to_file.write('<?xml version="1.0"?>')
to_file.write(('<annotation workingtree-root="%s" %s>' % \
(wt_root_path,
'file="%s"' % file_path)).encode(encoding, 'replace'))
if _annotate_file: # bzr < 1.8
annotations = _annotations(branch.repository, file_id, rev_id)
annotation = list(_annotate_file(branch, rev_id, file_id))
else:
tree = branch.repository.revision_tree(rev_id)
annotations = tree.annotate_iter(file_id)
annotation = list(_expand_annotations(annotations, branch))
for (revno_str, author, date_str, line_rev_id,
text, origin) in _annotation_iter(annotation, annotations):
if not show_ids:
origin = None
prevanno = _show_entry(to_file, prevanno, revno_str, author,
date_str, line_rev_id, text, origin)
to_file.write('</annotation>')
def _annotation_iter(annotation, annotations):
"""a annotaion iterator """
for ((revno_str, author, date_str, line_rev_id, text), \
(origin, text_dup)) in zip(annotation, annotations):
yield (revno_str, author, date_str, line_rev_id, text, origin)
def _show_entry(to_file, prevanno, revno_str, author,
date_str, line_rev_id, text, fid):
"""output one entry of the annotation"""
anno = 'revno="%s" author="%s" date="%s"' % \
(_escape_cdata(revno_str), _escape_cdata(author), date_str)
if anno.lstrip() == empty_annotation:
anno = prevanno
if fid:
to_file.write('<entry %s fid="%s">%s</entry>' % \
(anno, fid, _escape_cdata(text)))
else:
to_file.write('<entry %s>' % anno)
to_file.write('%s</entry>' % _escape_cdata(text))
return anno
|