allow disabling of scrub from command line
This commit is contained in:
1 parent
3699e08010
commit
6af9025eea
1 file changed
+12
-5
+12
-5
@@ -106,10 +106,10 @@ def finish(is_success):
|
|||||||
sys.exit(0 if is_success else 1)
|
sys.exit(0 if is_success else 1)
|
||||||
|
|
||||||
|
|
||||||
def load_config(file):
|
def load_config(args):
|
||||||
global config
|
global config
|
||||||
parser = ConfigParser.RawConfigParser()
|
parser = ConfigParser.RawConfigParser()
|
||||||
parser.read(file)
|
parser.read(args.conf)
|
||||||
sections = ["snapraid", "logging", "email", "smtp", "scrub"]
|
sections = ["snapraid", "logging", "email", "smtp", "scrub"]
|
||||||
config = dict((x, defaultdict(lambda: "")) for x in sections)
|
config = dict((x, defaultdict(lambda: "")) for x in sections)
|
||||||
for section in parser.sections():
|
for section in parser.sections():
|
||||||
@@ -130,6 +130,9 @@ def load_config(file):
|
|||||||
config["scrub"]["enabled"] = (config["scrub"]["enabled"].lower() == "true")
|
config["scrub"]["enabled"] = (config["scrub"]["enabled"].lower() == "true")
|
||||||
config["email"]["short"] = (config["email"]["short"].lower() == "true")
|
config["email"]["short"] = (config["email"]["short"].lower() == "true")
|
||||||
|
|
||||||
|
if args.scrub is not None:
|
||||||
|
config["scrub"]["enabled"] = args.scrub
|
||||||
|
|
||||||
|
|
||||||
def setup_logger():
|
def setup_logger():
|
||||||
log_format = logging.Formatter(
|
log_format = logging.Formatter(
|
||||||
@@ -166,9 +169,13 @@ def setup_logger():
|
|||||||
|
|
||||||
def main():
|
def main():
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
parser.add_argument("-c", "--conf", nargs=1,
|
parser.add_argument("-c", "--conf",
|
||||||
default="snapraid-runner.conf",
|
default="snapraid-runner.conf",
|
||||||
help="Configuration file (default %(default)s)")
|
metavar="CONFIG",
|
||||||
|
help="Configuration file (default: %(default)s)")
|
||||||
|
parser.add_argument("--no-scrub", action='store_false',
|
||||||
|
dest='scrub', default=None,
|
||||||
|
help="Do not scrub (overrides config)")
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
if not os.path.exists(args.conf):
|
if not os.path.exists(args.conf):
|
||||||
@@ -177,7 +184,7 @@ def main():
|
|||||||
sys.exit(2)
|
sys.exit(2)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
load_config(args.conf)
|
load_config(args)
|
||||||
except:
|
except:
|
||||||
print("unexpected exception while loading config")
|
print("unexpected exception while loading config")
|
||||||
print traceback.format_exc()
|
print traceback.format_exc()
|
||||||
|
|||||||
Reference in new issue
Block a user