replace calls to sys for arguments with argparse
This commit is contained in:
@@ -1,23 +1,43 @@
|
||||
import argparse
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
import json
|
||||
|
||||
|
||||
# Verify script arguments
|
||||
if len(sys.argv) != 3:
|
||||
sys.exit('Invalid number of arguments. Usage:\n updateSettings.py defaultSettingsFile.json outputSettingsFile.json')
|
||||
parser = argparse.ArgumentParser(
|
||||
description='Updates output settings file based on a default file',
|
||||
)
|
||||
|
||||
default_settings = sys.argv[1]
|
||||
transmission_settings = sys.argv[2]
|
||||
parser.add_argument(
|
||||
'input_file',
|
||||
type=str,
|
||||
help='Path to default settings json file',
|
||||
)
|
||||
|
||||
parser.add_argument(
|
||||
'output_file',
|
||||
type=str,
|
||||
help='Path to output settings json file',
|
||||
)
|
||||
|
||||
args = parser.parse_args()
|
||||
default_settings = args.input_file
|
||||
transmission_settings = args.output_file
|
||||
|
||||
# Fail if default settings file doesnt exist.
|
||||
if not os.path.isfile(default_settings):
|
||||
sys.exit('Invalid arguments, default settings file does not exist')
|
||||
sys.exit(
|
||||
'Invalid arguments, default settings file{file} does not exist'.format(
|
||||
file=default_settings,
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
# Define which file to base the config on
|
||||
if os.path.isfile(transmission_settings):
|
||||
configuration_baseline=transmission_settings
|
||||
configuration_baseline = transmission_settings
|
||||
else:
|
||||
configuration_baseline=default_settings
|
||||
configuration_baseline = default_settings
|
||||
|
||||
print('Using config baseline ' + configuration_baseline)
|
||||
|
||||
|
Reference in New Issue
Block a user