diff options
Diffstat (limited to 'src')
| -rwxr-xr-x | src/conf-mode/vyatta-update-crontab.py | 169 |
1 files changed, 169 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..381221bfb --- /dev/null +++ b/src/conf-mode/vyatta-update-crontab.py @@ -0,0 +1,169 @@ +#!/usr/bin/env python3 +# +# +# 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/>. +# +# + +#!/usr/bin/env python3 +import io +import re +import sys +sys.path.append("/usr/lib/python3/dist-packages/vyos/") + +from config import VyOSError, Config + + +class CronTask(Config): + def __init__(self, task): + + self.task = task + self.minutes = "*" + self.hours = "*" + self.days = "*" + self.user = "root" + self.executable = None + self.arguments = "" + self.interval = None + self.crontab_spec = None + self._cli_shell_api = "/bin/cli-shell-api" + + # Unused now + self.months = "*" + self.days_of_week = "*" + super(CronTask, self).set_level("system task-scheduler task") + try: + self.user = super(CronTask, self).return_value(" ".join([self.task, "user"])) + except VyOSError: + pass + + try: + byte_executable = super(CronTask, self).return_value(" ".join([self.task, "executable path"])) + self.executable = byte_executable.decode("utf-8") + except VyOSError: + raise VyOSError (task + "must define executable") + + try: + byte_arguments = super(CronTask, self).return_value(" ".join([self.task, "executable arguments"])) + self.executable = byte_arguments.decode("utf-8") + except VyOSError: + pass + + try: + self.interval = super(CronTask, self).return_value(" ".join([self.task, "interval"])) + except VyOSError: + self.interval = None + + try: + byte_crontab_spec = super(CronTask, self).return_value(" ".join([self.task, "crontab-spec"])) + self.crontab_spec = byte_crontab_spec + except: + self.crontab_spec = None + + +def get_config(): + conf = Config() + conf.set_level("system task-scheduler task") + tasks = conf.list_nodes("") + list_of_instanses=[] + for task in tasks: + list_of_instanses.append(CronTask(task.decode("utf-8"))) + return list_of_instanses + + +def verify(config): + for task in config: + if task.interval and task.crontab_spec: + raise VyOSError(task, "can not use interval and crontab-spec at the same time!") + + if task.interval: + result = re.search(b"(\d+)([mdh]?)", task.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!") + task.minutes = "*/" + str(value) + + elif suffix == b"h": + if value > 24: + raise VyOSError("Interval in hours must not exceed 24!") + task.minutes = "0" + task.hours = "*/" + str(value) + + elif suffix == b"d": + if value > 31: + raise VyOSError("Interval in days must not exceed 31!") + + task.minutes = "0" + task.hours = "0" + task.days = "*/" + str(value) + elif task.interval and task.crontab_spec: + raise VyOSError(task, "must define either interval or crontab-spec") + return None + + +def generate(config): + crontab = "/etc/cron.d/vyatta-crontab" + crontab_header = "### Added by /opt/vyatta/sbin/vyatta-update-crontab.py ###\n" + crontab_append = crontab_header + count = 0 + for task in config: + if task.interval: + crontab_string = "{minutes} {hours} {days} {months} {days_of_week} {user} {executable} {arguments}\n".format( + minutes=task.minutes, + hours=task.hours, + days=task.days, + months=task.months, + days_of_week=task.days_of_week, + user=task.user, + executable=task.executable, + arguments=task.arguments + ) + elif task.crontab_spec: + crontab_string = "{crontab_spec) {user} {executable} {arguments}\n".format( + crontab_spec=task.crontab_spec, + user=task.user, + executable=task.executable, + arguments=task.arguments + ) + 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") + + +def apply(config): + pass + + +if __name__ == '__main__': + try: + c = get_config() + verify(c) + generate(c) + apply(c) + except VyOSError: + sys.exit(0) + |