s4: Simple test script to create lots of contacts to stress the LDB

Autobuild-User: Matthieu Patou <mat@samba.org>
Autobuild-Date: Fri Nov 11 22:02:53 CET 2011 on sn-devel-104

1 files changed, 96 insertions, 0 deletions

diff --git a/source4/scripting/devel/addlotscontacts b/source4/scripting/devel/addlotscontacts
new file mode 100644
index 0000000000..cc073a3500
--- /dev/null
+++ b/source4/scripting/devel/addlotscontacts
@@ -0,0 +1,96 @@
+#!/usr/bin/env python
+#
+# Copyright (C) Matthieu Patou <mat@matws.net>  2010
+#
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+
+__docformat__ = "restructuredText"
+
+
+import optparse
+import sys
+# Allow to run from s4 source directory (without installing samba)
+sys.path.insert(0, "bin/python")
+
+import samba.getopt as options
+from samba.credentials import DONT_USE_KERBEROS
+from samba.auth import system_session
+from samba import param
+from samba.provision import find_provision_key_parameters
+from samba.upgradehelpers import (get_paths, get_ldbs)
+from ldb import SCOPE_BASE, Message, MessageElement, Dn, FLAG_MOD_ADD
+
+parser = optparse.OptionParser("chgtdcpass [options]")
+sambaopts = options.SambaOptions(parser)
+parser.add_option_group(sambaopts)
+parser.add_option_group(options.VersionOptions(parser))
+credopts = options.CredentialsOptions(parser)
+parser.add_option_group(credopts)
+
+(opts, args) = parser.parse_args()
+
+lp = sambaopts.get_loadparm()
+smbconf = lp.configfile
+creds = credopts.get_credentials(lp)
+creds.set_kerberos_state(DONT_USE_KERBEROS)
+
+if len(args) > 0:
+    num_contacts = int(args[0])
+else:
+    num_contacts = 10000
+
+if __name__ == '__main__':
+    paths = get_paths(param, smbconf=smbconf)
+    session = system_session()
+
+    ldbs = get_ldbs(paths, creds, session, lp)
+    ldbs.startTransactions()
+
+    names = find_provision_key_parameters(ldbs.sam, ldbs.secrets, ldbs.idmap,
+                                            paths, smbconf, lp)
+
+    contactdn = "OU=Contacts,%s" % str(names.domaindn)
+    res = ldbs.sam.search(expression="(dn=%s)" % contactdn,
+                    base=str(names.domaindn),
+                    scope=SCOPE_BASE)
+
+    if (len(res) == 0):
+        msg = Message()
+        msg.dn =  Dn(ldbs.sam, contactdn)
+        msg["objectClass"] = MessageElement("organizationalUnit", FLAG_MOD_ADD,
+                                            "objectClass")
+
+        ldbs.sam.add(msg)
+
+    print "Creating %d contacts" % num_contacts
+    count = 0
+    increment = num_contacts / 10
+    if increment > 5000:
+        increment = 5000
+
+    while (count < num_contacts):
+        msg = Message()
+        msg.dn =  Dn(ldbs.sam, "CN=contact%d,%s" % (count + 1, contactdn))
+        msg["objectClass"] = MessageElement("contact", FLAG_MOD_ADD,
+                                            "objectClass")
+
+        if count !=0 and (count % increment) == 0:
+            print "Added contacts: %d" % count
+
+        ldbs.sam.add(msg)
+        count += 1
+
+    ldbs.groupedCommit()