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()
|
