update arguments

douban · Nov 3, 2021 · 2d1487b · 2d1487b
1 parent 34ee292
commit 2d1487b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/sa_tools_core/notify.py b/sa_tools_core/notify.py
@@ -99,7 +99,10 @@ def notify(args):
         if values:
             addrs = [i for v in values for i in re.split(r"[,\s]+", v)]
             try:
-                getattr(notifier, type_)(addrs, from_addr=args.from_addr, company=args.company)
+                if args.company:
+                    getattr(notifier, type_)(addrs, from_addr=args.from_addr, company=args.company)
+                else:
+                    getattr(notifier, type_)(addrs, from_addr=args.from_addr)
             except Exception as e:
                 logger.exception("Notifier.%s(%s) failed: %s", type_, addrs, e)
                 report()