diff options
Diffstat (limited to 'libbe/command/html.py')
| -rw-r--r-- | libbe/command/html.py | 39 |
1 files changed, 34 insertions, 5 deletions
diff --git a/libbe/command/html.py b/libbe/command/html.py index 0b4cf89..bac310f 100644 --- a/libbe/command/html.py +++ b/libbe/command/html.py @@ -28,6 +28,7 @@ import libbe import libbe.command import libbe.command.util import libbe.util.encoding +import libbe.util.id class HTML (libbe.command.Command): @@ -222,16 +223,19 @@ class HTMLGen (object): if depth == 0: comment_entries.append('<div class="comment root">') else: - comment_entries.append('<div class="comment">') + comment_entries.append( + '<div class="comment" id="%s">' % comment.uuid) template_info = {'shortname': comment.id.user()} for attr in ['uuid', 'author', 'date', 'body']: value = getattr(comment, attr) if attr == 'body': + link_long_ids = False save_body = False if comment.content_type == 'text/html': - pass # no need to escape html... + link_long_ids = True elif comment.content_type.startswith('text/'): value = '<pre>\n'+self._escape(value)+'\n</pre>' + link_long_ids = True elif comment.content_type.startswith('image/'): save_body = True value = '<img src="./%s/%s" />' \ @@ -240,6 +244,8 @@ class HTMLGen (object): save_body = True value = '<a href="./%s/%s">Link to %s file</a>.' \ % (bug.uuid, comment.uuid, comment.content_type) + if link_long_ids == True: + value = self._long_to_linked_user(value) if save_body == True: per_bug_dir = os.path.join(self.out_dir_bugs, bug.uuid) if not os.path.exists(per_bug_dir): @@ -249,9 +255,7 @@ class HTMLGen (object): '<Files %s>\n ForceType %s\n</Files>' \ % (comment.uuid, comment.content_type), [per_bug_dir, '.htaccess'], mode='a') - self._write_file( # TODO: long_to_linked_user() - libbe.util.id.long_to_short_text( - [self.bd], comment.body), + self._write_file(comment.body, [per_bug_dir, comment.uuid], mode='wb') else: value = self._escape(value) @@ -262,6 +266,31 @@ class HTMLGen (object): comment_entries.append('</div>\n') # close every remaining <div class='comment... return '\n'.join(comment_entries) + def _long_to_linked_user(self, text): + replacer = libbe.util.id.IDreplacer( + [self.bd], self._long_to_linked_user_replacer, wrap=False) + return re.sub( + libbe.util.id.REGEXP, replacer, text) + + def _long_to_linked_user_replacer(self, bugdirs, long_id): + try: + p = libbe.util.id.parse_user(bugdirs[0], long_id) + short_id = libbe.util.id.long_to_short_user(bugdirs, long_id) + except (libbe.util.id.MultipleIDMatches, + libbe.util.id.NoIDMatches, + libbe.util.id.InvalidIDStructure), e: + return long_id + if p['type'] == 'bugdir': + return long_id + elif p['type'] == 'bug': + return '<a href="./%s.html">%s</a>' \ + % (p['bug'], short_id) + elif p['type'] == 'comment': + return '<a href="./%s.html#%s">%s</a>' \ + % (p['bug'], p['comment'], short_id) + raise Exception('Invalid id type %s for "%s"' + % (p['type'], long_id)) + def _write_index_file(self, bugs, title, index_header, bug_type='active'): if self.verbose: print >> self.stdout, 'Writing %s index file for %d bugs' % (bug_type, len(bugs)) |