From 984725f4e6a616bf23661cf510dfc99e1c9254a1 Mon Sep 17 00:00:00 2001 From: Daniil Baturin Date: Thu, 31 May 2018 03:19:41 +0200 Subject: T629: replace the config-management migration script. --- src/migration-scripts/config-management/0-to-1 | 31 ++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100755 src/migration-scripts/config-management/0-to-1 (limited to 'src') diff --git a/src/migration-scripts/config-management/0-to-1 b/src/migration-scripts/config-management/0-to-1 new file mode 100755 index 000000000..344359110 --- /dev/null +++ b/src/migration-scripts/config-management/0-to-1 @@ -0,0 +1,31 @@ +#!/usr/bin/env python3 + +# Add commit-revisions option if it doesn't exist + +import sys + +from vyos.configtree import ConfigTree + +if (len(sys.argv) < 1): + print("Must specify file name!") + sys.exit(1) + +file_name = sys.argv[1] + +with open(file_name, 'r') as f: + config_file = f.read() + +config = ConfigTree(config_file) + +if config.exists(['system', 'config-management', 'commit-revisions']): + # Nothing to do + sys.exit(0) +else: + config.set(['system', 'config-management', 'commit-revisions'], value='200') + + try: + with open(file_name, 'w') as f: + f.write(config.to_string()) + except OSError as e: + print("Failed to save the modified config: {}".format(e)) + sys.exit(1) -- cgit v1.2.3