python: create a script for reorgnizing an LDB file.
[ira/wip.git] / source4 / scripting / bin / reorgldb.py
1 #!/usr/bin/python
2 #
3 # Copyright (C) Matthieu Patou <mat@matws.net> 2009
4 # This script realize an offline reorganisation of an LDB
5 # file it helps to reduce (sometime drastically) the
6 # size of LDB files.
7 import sys
8 import optparse
9 import os
10 sys.path.insert(0, "bin/python")
11
12 import samba
13 from samba.credentials import DONT_USE_KERBEROS
14 from samba.auth import system_session
15 from samba import Ldb, substitute_var, valid_netbios_name, check_all_substituted
16 from ldb import SCOPE_SUBTREE, SCOPE_ONELEVEL, SCOPE_BASE, LdbError
17 import ldb
18 import samba.getopt as options
19 from samba.samdb import SamDB
20 from samba import param
21 from samba.provision import ProvisionPaths, ProvisionNames,provision_paths_from_lp, Schema
22
23 parser = optparse.OptionParser("provision [options]")
24 sambaopts = options.SambaOptions(parser)
25 parser.add_option_group(sambaopts)
26 parser.add_option_group(options.VersionOptions(parser))
27 credopts = options.CredentialsOptions(parser)
28 parser.add_option_group(credopts)
29 parser.add_option("--database", type="string", metavar="FILE",
30         help="LDB to reorganize")
31 opts = parser.parse_args()[0]
32 lp = sambaopts.get_loadparm()
33 smbconf = lp.configfile
34
35 if not opts.database:
36         print "Parameter database is mandatory"
37         sys.exit(1)
38 creds = credopts.get_credentials(lp)
39 creds.set_kerberos_state(DONT_USE_KERBEROS)
40 session = system_session()
41 empty = ldb.Message()
42 newname="%s.new"%(opts.database)
43 if os.path.exists(newname):
44         os.remove(newname)
45 old_ldb = Ldb(opts.database, session_info=session, credentials=creds,lp=lp)
46 new_ldb = Ldb(newname,session_info=session, credentials=creds,lp=lp)
47
48 new_ldb.transaction_start()
49 res = old_ldb.search(expression="(dn=*)",base="", scope=SCOPE_SUBTREE)
50 for i in range(0,len(res)):
51         if str(res[i].dn) == "@BASEINFO":
52                 continue
53         if str(res[i].dn).startswith("@INDEX:"):
54                 continue
55         delta = new_ldb.msg_diff(empty,res[i])
56         delta.dn = res[i].dn
57         delta.remove("distinguishedName")
58         new_ldb.add(delta)
59
60 new_ldb.transaction_commit()