From: David Howells <dhowells@redhat.com>
Date: Wed, 30 Apr 2014 12:51:14 +0000 (+0100)
Subject: Implement "bos setrestart"
X-Git-Url: https://www.infradead.org/git/?a=commitdiff_plain;h=caa34d85ce746dc3f080467866268d334afd6f09;p=users%2Fdhowells%2Fkafs-utils.git
Implement "bos setrestart"
Signed-off-by: David Howells <dhowells@redhat.com>
---
diff --git a/suite/commands/bos/setrestart.py b/suite/commands/bos/setrestart.py
new file mode 100644
index 0000000..a3e3948
--- /dev/null
+++ b/suite/commands/bos/setrestart.py
@@ -0,0 +1,64 @@
+#
+# AFS Server management toolkit: Set server restart times
+# -*- coding: utf-8 -*-
+#
+
+__copyright__ = """
+Copyright (C) 2014 Red Hat, Inc. All Rights Reserved.
+Written by David Howells (dhowells@redhat.com)
+
+This program is free software; you can redistribute it and/or modify
+it under the terms of the GNU General Public Licence version 2 as
+published by the Free Software Foundation.
+
+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 Licence for more details.
+
+You should have received a copy of the GNU General Public Licence
+along with this program; if not, write to the Free Software
+Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+"""
+
+from exception import AFSArgumentError
+from afs.argparse import *
+from afs.lib.output import *
+from afs.lib.parse_setrestart_time import *
+import kafs
+
+help = "Set when the BOS Server restarts processes"
+
+command_arguments = [
+ [ "server", get_bosserver, "rs", "<machine name>" ],
+ [ "time", parse_restart_time, "rs", "<time to restart server>" ],
+ [ "general", get_dummy, "fn" ],
+ [ "newbinary", get_dummy, "fn" ],
+ [ "cell", get_cell, "os", "<cell name>" ],
+ [ "noauth", get_auth, "fn" ],
+ [ "localauth", get_auth, "fn" ],
+ [ "verbose", get_verbose, "fn" ],
+ [ "encrypt", get_dummy, "fn" ],
+]
+
+cant_combine_arguments = [
+ ( "general", "newbinary" ),
+ ( "cell", "localauth" ),
+ ( "noauth", "localauth" ),
+]
+
+description = r"""
+Set when the BOS Server restarts processes
+"""
+
+def main(params):
+ cell = params["cell"]
+ bos_conn = cell.open_bos_server(params["server"], params)
+
+ time = params["time"]
+
+ t = kafs.BOZO_RESTARTTIME_GENERAL
+ if "newbinary" in params:
+ t = kafs.BOZO_RESTARTTIME_NEWBIN
+
+ ret = kafs.BOZO_SetRestartTime(bos_conn, t, time)
diff --git a/suite/lib/parse_setrestart_time.py b/suite/lib/parse_setrestart_time.py
new file mode 100644
index 0000000..fec69f1
--- /dev/null
+++ b/suite/lib/parse_setrestart_time.py
@@ -0,0 +1,105 @@
+#
+# AFS Server management toolkit: Restart time parser
+# -*- coding: utf-8 -*-
+#
+
+__copyright__ = """
+Copyright (C) 2014 Red Hat, Inc. All Rights Reserved.
+Written by David Howells (dhowells@redhat.com)
+
+This program is free software; you can redistribute it and/or modify
+it under the terms of the GNU General Public Licence version 2 as
+published by the Free Software Foundation.
+
+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 Licence for more details.
+
+You should have received a copy of the GNU General Public Licence
+along with this program; if not, write to the Free Software
+Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+"""
+
+from exception import AFSArgumentError
+import kafs
+
+days = [ "sunday", "monday", "tuesday", "wednesday", "thursday", "friday", "saturday" ]
+
+###############################################################################
+#
+# Parse the bos setrestart -time argument
+#
+###############################################################################
+def parse_restart_time(switch, params):
+ at = kafs.bozo_netKTime()
+
+ time = params[0].strip().lower()
+ if time == "":
+ raise AFSArgumentError("Empty time string")
+ if time == "never":
+ at.mask = kafs.KTIME_NEVER
+ return at
+ if time == "now":
+ at.mask = kafs.KTIME_NOW
+ return at
+
+ if time.startswith("every "):
+ time = time[6:].lstrip()
+ elif time.startswith("at "):
+ time = time[2:].lstrip()
+
+ if time == "":
+ raise AFSArgumentError("Incomplete time string")
+ if time[0].isalpha():
+ ix = time.find(" ")
+ if ix < 3:
+ raise AFSArgumentError("Can't extract day name")
+ day = time[:ix]
+ time = time[ix + 1:].lstrip()
+ for i in range(0, 7):
+ if days[i].startswith(day):
+ at.day = i
+ at.mask |= kafs.KTIME_DAY
+ break
+ else:
+ raise AFSArgumentError("Unrecognised day name")
+
+ ix = time.find(" ")
+ if ix == -1:
+ # 24-hour clock
+ clock_type = 24
+ else:
+ # 12-hour clock
+ clock_type = 12
+ ampm = time[ix + 1:].lstrip()
+ if ampm == "am":
+ at.hour = 0
+ elif ampm == "pm":
+ at.hour = 12
+ else:
+ raise AFSArgumentError("Expect 'am' or 'pm' in time")
+ time = time[0:ix]
+
+ ix = time.find(":")
+ if ix == -1:
+ raise AFSArgumentError("Expect colon in time")
+ if ix <= 0 or ix > 2:
+ raise AFSArgumentError("Expect one or two hour digits in time")
+ if len(time) - (ix + 1) != 2:
+ raise AFSArgumentError("Expect two minute digits in time")
+
+ hour = int(time[0:ix])
+ if (clock_type == 12 and (hour < 1 or hour > 12) or
+ clock_type == 24 and (hour < 0 or hour > 23)):
+ raise AFSArgumentError("Hour {:d} out of range in time".format(hour))
+ if clock_type == 12 and hour == 12:
+ hour = 0 # 12am maps to 00:00 and 12pm to 12:00
+ at.hour += hour
+
+ at.min = int(time[ix + 1:])
+ if at.min > 59:
+ raise AFSArgumentError("Minute out of range in time")
+
+ at.mask |= kafs.KTIME_HOUR | kafs.KTIME_MIN
+ return at