diff options
Diffstat (limited to 'src/conf-mode/vyos-update-crontab.py')
| -rwxr-xr-x | src/conf-mode/vyos-update-crontab.py | 21 |
1 files changed, 11 insertions, 10 deletions
diff --git a/src/conf-mode/vyos-update-crontab.py b/src/conf-mode/vyos-update-crontab.py index ac6ae96d3..2d15de8ea 100755 --- a/src/conf-mode/vyos-update-crontab.py +++ b/src/conf-mode/vyos-update-crontab.py @@ -20,7 +20,8 @@ import os import re import sys -from vyos.config import VyOSError, Config +from vyos.config import Config +from vyos.util import ConfigError crontab_file = "/etc/cron.d/vyos-crontab" @@ -75,34 +76,34 @@ def get_config(): def verify(tasks): for task in tasks: if not task["interval"] and not task["spec"]: - raise VyOSError("Invalid task {0}: must define either interval or crontab-spec".format(task["name"])) + raise ConfigError("Invalid task {0}: must define either interval or crontab-spec".format(task["name"])) if task["interval"]: if task["spec"]: - raise VyOSError("Invalid task {0}: cannot use interval and crontab-spec at the same time".format(task["name"])) + raise ConfigError("Invalid task {0}: cannot use interval and crontab-spec at the same time".format(task["name"])) if not re.match(r"^\d+[mdh]?$", task["interval"]): - raise(VyOSError("Invalid interval {0} in task {1}: interval should be a number optionally followed by m, h, or d".format(task["name"], task["interval"]))) + raise(ConfigError("Invalid interval {0} in task {1}: interval should be a number optionally followed by m, h, or d".format(task["name"], task["interval"]))) else: # Check if values are within allowed range value, suffix = split_interval(task["interval"]) if not suffix or suffix == "m": if value > 60: - raise VyOSError("Invalid task {0}: interval in minutes must not exceed 60".format(task["name"])) + raise ConfigError("Invalid task {0}: interval in minutes must not exceed 60".format(task["name"])) elif suffix == "h": if value > 24: - raise VyOSError("Invalid task {0}: interval in hours must not exceed 24".format(task["name"])) + raise ConfigError("Invalid task {0}: interval in hours must not exceed 24".format(task["name"])) elif suffix == "d": if value > 31: - raise VyOSError("Invalid task {0}: interval in days must not exceed 31".format(task["name"])) + raise ConfigError("Invalid task {0}: interval in days must not exceed 31".format(task["name"])) if not task["executable"]: - raise VyOSError("Invalid task {0}: executable is not defined".format(task["name"])) + raise ConfigError("Invalid task {0}: executable is not defined".format(task["name"])) else: # Check if executable exists and is executable if not (os.path.isfile(task["executable"]) and os.access(task["executable"], os.X_OK)): - raise VyOSError("Invalid task {0}: file {1} does not exist or is not executable".format(task["name"], task["executable"])) + raise ConfigError("Invalid task {0}: file {1} does not exist or is not executable".format(task["name"], task["executable"])) def generate(tasks): crontab_header = "### Generated by vyos-update-crontab.py ###\n" @@ -142,6 +143,6 @@ if __name__ == '__main__': verify(c) generate(c) apply(c) - except VyOSError as e: + except ConfigError as e: print(e) sys.exit(1) |