Provide hook to get PAU in a different way

zopefoundation · Sep 15, 2006 · 8592d61 · 8592d61
1 parent c3a140a
commit 8592d61
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/z3c/sampledata/generator/principals.py b/src/z3c/sampledata/generator/principals.py
@@ -111,9 +111,7 @@ def generate(self, context, param={}, dataSource=[], seed=None):
         originalSite = hooks.getSite()
         hooks.setSite(context)
         sm = zope.component.getSiteManager(context)
-        self.pau = sm
-        for loc in param['pauLocation'].split('/'):
-            self.pau = self.pau[loc]
+        self.pau = self._getPAU(sm, param)
         self.passwordManagerName = param['passwordManager']
 
         numCreated = 0
@@ -140,6 +138,13 @@ def generate(self, context, param={}, dataSource=[], seed=None):
 
         return self.pau
 
+    def _getPAU(self, sm, param):
+        pau = sm
+        for loc in param['pauLocation'].split('/'):
+            pau = pau[loc]
+        return pau
+
+
     def _createPrincipal(self, info):
         login = unicode(info[0])
         self.logins.append(login)