diff options
author | Douglas Bagnall <douglas.bagnall@catalyst.net.nz> | 2018-10-19 16:37:08 +1300 |
---|---|---|
committer | Douglas Bagnall <dbagnall@samba.org> | 2019-01-08 23:55:32 +0100 |
commit | b275baebd7e7fa8576cb49737a886f02e3615640 (patch) | |
tree | 312552b8b9f5c4a90932d15ec94f4f163381e44f /script/traffic_learner | |
parent | 3b3b4fa6f960d2bc06d830266671b582ddf186a1 (diff) | |
download | samba-b275baebd7e7fa8576cb49737a886f02e3615640.tar.gz |
traffic_learner: use samba.logger, not print(file=sys.stderr)
Signed-off-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
Diffstat (limited to 'script/traffic_learner')
-rwxr-xr-x | script/traffic_learner | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/script/traffic_learner b/script/traffic_learner index b3570d45eea..b46521f70cd 100755 --- a/script/traffic_learner +++ b/script/traffic_learner @@ -23,6 +23,10 @@ import argparse sys.path.insert(0, "bin/python") from samba.emulate import traffic +from samba.logger import get_samba_logger +logger = get_samba_logger(name=__name__, level=20) +error = logger.error +info = logger.info def main(): parser = argparse.ArgumentParser(description=__doc__, @@ -37,12 +41,12 @@ def main(): args = parser.parse_args() if args.out is None: - print("No output file was specified to write the model to.", file=sys.stdout) - print("Please specify a filename using the --out option.", file=sys.stdout) + error("No output file was specified to write the model to.") + error("Please specify a filename using the --out option.") return 1 if args.SUMMARY_FILE is sys.stdin: - print("reading from STDIN...", file=sys.stderr) + info("reading from STDIN...") (conversations, interval, @@ -51,7 +55,7 @@ def main(): dns_mode=args.dns_mode) model = traffic.TrafficModel() - print("learning model", sys.stderr) + info("learning model") if args.dns_mode == 'count': model.learn(conversations, dns_counts) else: |