コミット: 1c05dfadcc8744a02c9f05d8db8e4ff2a5d951ff - allura (git)

allura

コミットメタ情報

• リビジョン: 1c05dfadcc8744a02c9f05d8db8e4ff2a5d951ff (tree)
• 日時: 2012-05-17 05:24:10
• 作者: Cory Johns <johnsca@geek...>
• コミッター: Dave Brondsema

ログメッセージ

[#4118] Added chunking, intermediate flushing, and filtering to user projects

Signed-off-by: Cory Johns <[e-mail]>

変更サマリ

• modified: scripts/migrations/024-migrate-custom-profile-text.py

差分

--- a/scripts/migrations/024-migrate-custom-profile-text.py
+++ b/scripts/migrations/024-migrate-custom-profile-text.py
@@ -6,6 +6,7 @@ from pylons import c
 from ming.orm import ThreadLocalORMSession
 
 from allura import model as M
+from allura.lib import utils
 from forgewiki import model as WM
 from forgewiki.wiki_main import ForgeWikiApp
 
@@ -19,38 +20,40 @@ default_personal_project_tmpl = ("This is the personal project of %s."
             " project such as cloned repositories.\n\n%s")
 
 def main():
-    for p in M.Project.query.find().all():
-        user = p.user_project_of
-        if not user:
-            continue
-
-        description = p.description
-        if description is None or re.match(default_description, description):
-            continue
-
-        app = p.app_instance('wiki')
-        if app is None:
-            p.install_app('wiki')
-
-        page = WM.Page.query.get(app_config_id=app.config._id, title='Home')
-        if page is None:
-            continue
-
-        c.app = app
-        c.project = p
-        c.user = user
-
-        if "This is the personal project of" in page.text:
-            if description not in page.text:
-                page.text = "%s\n\n%s" % (page.text, description)
+    users = M.Neighborhood.query.get(name='Users')
+    for chunk in utils.chunked_find(M.Project, {'neighborhood_id': users._id}):
+        for p in chunk:
+            user = p.user_project_of
+            if not user:
+                continue
+
+            description = p.description
+            if description is None or re.match(default_description, description):
+                continue
+
+            app = p.app_instance('wiki')
+            if app is None:
+                p.install_app('wiki')
+
+            page = WM.Page.query.get(app_config_id=app.config._id, title='Home')
+            if page is None:
+                continue
+
+            c.app = app
+            c.project = p
+            c.user = user
+
+            if "This is the personal project of" in page.text:
+                if description not in page.text:
+                    page.text = "%s\n\n%s" % (page.text, description)
+                    log.info("Update wiki home page text for %s" % user.username)
+            elif "This is the default page" in page.text:
+                page.text = default_personal_project_tmpl % (user.display_name, description)
                 log.info("Update wiki home page text for %s" % user.username)
-        elif "This is the default page" in page.text:
-            page.text = default_personal_project_tmpl % (user.display_name, description)
-            log.info("Update wiki home page text for %s" % user.username)
-        else:
-            pass
+            else:
+                pass
 
-    ThreadLocalORMSession.flush_all()
+        ThreadLocalORMSession.flush_all()
 
 if __name__ == '__main__':
     main()