summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorTaniadz <tdziubenko@ukr.net>2017-08-23 18:23:11 +0300
committerTaniadz <tdziubenko@ukr.net>2017-08-23 18:23:11 +0300
commit96aef9c3fe9c5341acd98f697a627c0d1d6fe33f (patch)
tree1094de82480b6312f7232ea166e49aefb9f808e6 /src
parent65a596b414756b8a0e49ea837db1e30215c13711 (diff)
downloadvyos-1x-96aef9c3fe9c5341acd98f697a627c0d1d6fe33f.tar.gz
vyos-1x-96aef9c3fe9c5341acd98f697a627c0d1d6fe33f.zip
adding vyatta-update-crontab.py
Diffstat (limited to 'src')
-rwxr-xr-xsrc/conf-mode/vyatta-update-crontab.py147
1 files changed, 147 insertions, 0 deletions
diff --git a/src/conf-mode/vyatta-update-crontab.py b/src/conf-mode/vyatta-update-crontab.py
new file mode 100755
index 000000000..605742828
--- /dev/null
+++ b/src/conf-mode/vyatta-update-crontab.py
@@ -0,0 +1,147 @@
+#!/usr/bin/env python3
+#
+# vyatta-update-ctontab.pl: crontab generator
+#
+# Maintainer: Daniil Baturin <daniil@baturin.org>
+#
+# Copyright (C) 2013 SO3Group
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License 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 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/>.
+#
+#
+
+import io
+import os
+import re
+import sys
+sys.path.append("/usr/lib/python3/dist-packages/vyos/")
+import config
+from config import VyOSError
+
+
+def update_crontab():
+ conf = config.Config()
+ count = 0
+
+
+ default_user = "root"
+ crontab = "/etc/cron.d/vyatta-crontab"
+ crontab_header = "### Added by /opt/vyatta/sbin/vyatta-update-crontab.py ###\n"
+ if not conf.exists("system task-scheduler task"):
+ os.remove(crontab)
+ return 0
+ crontab_append = crontab_header
+
+ conf.set_level("system task-scheduler task")
+ tasks = conf.list_nodes("")
+
+ for task in tasks:
+ task = task.decode("utf-8")
+ minutes = "*"
+ hours = "*"
+ days = "*"
+
+ # Unused now
+ months = "*"
+ days_of_week = "*"
+ try:
+ user = conf.return_value(" ".join([task, "user"]))
+ except VyOSError:
+ user = default_user
+
+ try:
+ executable = conf.return_value(" ".join([task, "executable path"]))
+ executable = executable.decode("utf-8")
+ except VyOSError:
+ raise VyOSError (task + "must define executable")
+
+ try:
+ arguments = conf.return_value(" ".join([task, "executable arguments"]))
+ except VyOSError:
+ arguments = ""
+
+ try:
+ interval = conf.return_value(" ".join([task, "interval"]))
+ except VyOSError:
+ interval = None
+
+ try:
+ crontab_spec = conf.return_value(" ".join([task, "crontab-spec"]))
+ except:
+ crontab_spec = None
+
+ if interval and crontab_spec:
+ raise VyOSError(task, "can not use interval and crontab-spec at the same time!")
+
+ if interval:
+ result = re.search(b"(\d+)([mdh]?)", interval)
+ value = int(result.group(1))
+ suffix = result.group(2)
+
+
+ if not suffix or suffix == b"m":
+ if value > 60:
+ raise VyOSError("Interval in minutes must not exceed 60!")
+ minutes = "*/" + str(value)
+
+ elif suffix == b"h":
+ if value > 24:
+ raise VyOSError("Interval in hours must not exceed 24!")
+ minutes = "0"
+ hours = "*/" + str(value)
+
+ elif suffix == b"d":
+ if value > 31:
+ raise VyOSError("Interval in days must not exceed 31!")
+
+ minutes = "0"
+ hours = "0"
+ days = "*/" + str(value)
+ crontab_string = "{minutes} {hours} {days} {months} {days_of_week} {user} {executable} {arguments}\n".format(
+ minutes=minutes,
+ hours=hours,
+ days=days,
+ months=months,
+ days_of_week=days_of_week,
+ user=user,
+ executable=executable,
+ arguments=arguments
+ )
+ elif crontab_spec:
+ crontab_string = "{crontab_spec) {user} {executable} {arguments}\n".format(
+ crontab_spec=crontab_spec,
+ user=user,
+ executable=executable,
+ arguments=arguments
+ )
+ else:
+ raise VyOSError(task, "must define either interval or crontab-spec")
+ crontab_append = crontab_append + crontab_string
+ count = count + 1
+
+ if count > 0:
+ try:
+ f = io.open(crontab, "w")
+ f.write(crontab_append)
+ f.close()
+ except IOError:
+ print("Could not open /etc/crontab for write")
+ else:
+ os.remove(crontab)
+
+if __name__ == '__main__':
+ try:
+ update_crontab()
+
+ except VyOSError:
+ sys.exit(0) \ No newline at end of file