summaryrefslogtreecommitdiffstats
path: root/source4/scripting/devel/addlotscontacts
blob: 9ecd16b94bae0410815012202b762eeabd0a0c58 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
#!/usr/bin/env python3
#
# 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("addlotscontacts [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="(distinguishedName=%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()