function to recursively migrate comments added.

svn path=/plone.app.discussion/trunk/; revision=28016

plone/app/discussion/browser/migration.py

@@ -13,6 +13,8 @@ from Products.CMFCore.interfaces import IContentish
 
 from zope.component import createObject
 
+from plone.app.discussion.comment import CommentFactory
+
 from plone.app.discussion.interfaces import IConversation
 
 
@@ -24,10 +26,42 @@ class View(BrowserView):
 
         context = aq_inner(self.context)
         out = []
+        self.total_comments_migrated = 0
+
         def log(msg):
             context.plone_log(msg)
             out.append(msg)
 
+        def migrate_replies(context, in_reply_to, replies, depth=0):
+            # recursive function to migrate all direct replies
+            # of a comment
+
+            for reply in replies:
+
+                # log
+                indent = "  "
+                for i in range(depth):
+                    indent += "  "
+                log("%smigrate_reply: '%s'." % (indent, reply.title))
+
+                # create a reply object
+                comment = CommentFactory()
+                comment.title = reply.Title()
+                comment.text = reply.text
+                comment.creator = reply.Creator()
+                #comment.author_name = author_username
+                #comment.author_email = ""
+                #comment.creation_date
+
+                comment.reply_to = in_reply_to
+
+                new_in_reply_to = conversation.addComment(comment)
+                self.total_comments_migrated += 1
+
+                # migrate all talkbacks of the reply
+                talkback = getattr( reply, 'talkback', None )
+                migrate_replies(context, new_in_reply_to, talkback.getReplies(), depth=depth+1)
+
         log("Comment migration started.")
 
         # Find content
@@ -35,29 +69,24 @@ class View(BrowserView):
         dtool = context.portal_discussion
         brains = catalog.searchResults(
                     object_provides='Products.CMFCore.interfaces._content.IContentish')
-        log("Found %s content objects to migrate." % len(brains))
+        comment_brains = catalog.searchResults(Type='Discussion Item')
+        total_comment_brains = len(comment_brains)
+        log("Found %s content objects." % len(brains))
+        log("Found %s Discussion Item objects." % total_comment_brains)
 
         for brain in brains:
             if brain.portal_type != 'Discussion Item':
-                old_comments = []
                 obj = brain.getObject()
                 talkback = getattr( obj, 'talkback', None )
                 if talkback:
-                    replies = talkback.objectValues()
-                    log("%s: Found talkback with %s comments to migrate"\
-                        % (obj.absolute_url(relative=1), len(replies)))
-                    for reply in replies:
-                        old_comments.append(reply)
-
-                # Build up new conversation/comments structure
-                conversation = IConversation(obj)
-
-                for old_comment in old_comments:
-                    comment = createObject('plone.Comment')
-                    comment.title = old_comment.Title()
-                    comment.text = old_comment.text
-                    comment.Creator = old_comment.Creator
-                    conversation.addComment(comment)
+                    replies = talkback.getReplies()
+                    if replies:
+                        conversation = IConversation(obj)
+                        log("Create conversation for: '%s'" % obj.Title())
+                    log("%s: Found talkback" % obj.absolute_url(relative=1))
+                    migrate_replies(context, 0, replies)
 
         log("Comment migration finished.")
-        return out
+        log("%s of %s comments migrated."
+            % (self.total_comments_migrated, total_comment_brains))
+        return '\n'.join(out)