setup demo graph

cosent · Oct 1, 2012 · 63d6084 · 63d6084
1 parent 043dd1d
commit 63d6084
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/plonesocial/suite/setuphandlers.py b/plonesocial/suite/setuphandlers.py
@@ -11,6 +11,7 @@
 from plone import api
 
 
+from plonesocial.network.interfaces import INetworkGraph
 from plonesocial.microblog.interfaces import IMicroblogTool
 from plonesocial.microblog.statusupdate import StatusUpdate
 
@@ -60,6 +61,13 @@ def demo(context):
         memberdata._setPortrait(portrait, userid)
 
     # setup social network
+    graph = queryUtility(INetworkGraph)
+    graph.clear()
+    for i in xrange(25):
+        followee = random.choice(users)
+        follower = random.choice(users)
+        if followee != follower:
+            graph.set_follow(follower, followee)
 
     # microblog
     microblog = queryUtility(IMicroblogTool)