s4:python tools - try to fix some test problems
[amitay/samba.git] / source4 / setup / setexpiry
1 #!/usr/bin/python
2 #
3 # Sets the user password expiry on a Samba4 server
4 # Copyright Jelmer Vernooij 2008
5 #
6 # Based on the original in EJS:
7 # Copyright Andrew Tridgell 2005
8 #
9 # This program is free software; you can redistribute it and/or modify
10 # it under the terms of the GNU General Public License as published by
11 # the Free Software Foundation; either version 3 of the License, or
12 # (at your option) any later version.
13 #
14 # This program is distributed in the hope that it will be useful,
15 # but WITHOUT ANY WARRANTY; without even the implied warranty of
16 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17 # GNU General Public License for more details.
18 #
19 # You should have received a copy of the GNU General Public License
20 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
21 #
22
23 import sys
24
25 # Find right directory when running from source tree
26 sys.path.insert(0, "bin/python")
27
28 import samba.getopt as options
29 import optparse
30
31 from samba.auth import system_session
32 from samba.samdb import SamDB
33
34 parser = optparse.OptionParser("setexpiry [username] [options]")
35 sambaopts = options.SambaOptions(parser)
36 parser.add_option_group(sambaopts)
37 parser.add_option_group(options.VersionOptions(parser))
38 credopts = options.CredentialsOptions(parser)
39 parser.add_option_group(credopts)
40 parser.add_option("--filter", help="LDAP Filter to set password on", type=str)
41 parser.add_option("--days", help="Days to expiry", type=int)
42 parser.add_option("--noexpiry", help="Password does never expire", action="store_true")
43
44 opts, args = parser.parse_args()
45
46 if (len(args) == 0) and (filter is None):
47         print "Either the username or '--filter' must be specified!"
48         parser.print_usage()
49         sys.exit(1)
50
51 days = opts.days
52 if days is None:
53         days = 0
54
55 if filter is None:
56         username = args[0]
57         filter = "(&(objectClass=user)(sAMAccountName=%s))" % (username)
58
59 lp = sambaopts.get_loadparm()
60 creds = credopts.get_credentials(lp)
61
62 samdb = SamDB(url=lp.get("sam database"), session_info=system_session(),
63               credentials=creds, lp=lp)
64 samdb.setexpiry(filter, days*24*3600, no_expiry_req=opts.noexpiry)