vyos.base: use Warning() helper for op-mode commands

3 files changed, 6 insertions, 3 deletions

diff --git a/src/op_mode/restart_frr.py b/src/op_mode/restart_frr.py
index e5014452f..91b25567a 100755
--- a/src/op_mode/restart_frr.py
+++ b/src/op_mode/restart_frr.py
@@ -22,6 +22,7 @@ import psutil
 from logging.handlers import SysLogHandler
 from shutil import rmtree
 
+from vyos.base import Warning
 from vyos.util import call
 from vyos.util import ask_yes_no
 from vyos.util import process_named_running
@@ -163,7 +164,7 @@ if cmd_args.action == 'restart':
     if cmd_args.daemon != ['']:
         for daemon in cmd_args.daemon:
             if not process_named_running(daemon):
-                print('WARNING: some of listed daemons are not running!')
+                Warning('some of listed daemons are not running!')
 
     # run command to restart daemon
     for daemon in cmd_args.daemon:
diff --git a/src/op_mode/show_dhcp.py b/src/op_mode/show_dhcp.py
index cd6e8ed43..4b1758eea 100755
--- a/src/op_mode/show_dhcp.py
+++ b/src/op_mode/show_dhcp.py
@@ -26,6 +26,7 @@ from datetime import datetime
 
 from isc_dhcp_leases import Lease, IscDhcpLeases
 
+from vyos.base import Warning
 from vyos.config import Config
 from vyos.util import is_systemd_service_running
 
@@ -213,7 +214,7 @@ if __name__ == '__main__':
 
     # if dhcp server is down, inactive leases may still be shown as active, so warn the user.
     if not is_systemd_service_running('isc-dhcp-server.service'):
-        print("WARNING: DHCP server is configured but not started. Data may be stale.")
+        Warning('DHCP server is configured but not started. Data may be stale.')
 
     if args.leases:
         leases = get_leases(conf, lease_file, args.state, args.pool, args.sort)
diff --git a/src/op_mode/show_dhcpv6.py b/src/op_mode/show_dhcpv6.py
index 1f987ff7b..b34b730e6 100755
--- a/src/op_mode/show_dhcpv6.py
+++ b/src/op_mode/show_dhcpv6.py
@@ -26,6 +26,7 @@ from datetime import datetime
 
 from isc_dhcp_leases import Lease, IscDhcpLeases
 
+from vyos.base import Warning
 from vyos.config import Config
 from vyos.util import is_systemd_service_running
 
@@ -203,7 +204,7 @@ if __name__ == '__main__':
 
     # if dhcp server is down, inactive leases may still be shown as active, so warn the user.
     if not is_systemd_service_running('isc-dhcp-server6.service'):
-        print("WARNING: DHCPv6 server is configured but not started. Data may be stale.")
+        Warning('DHCPv6 server is configured but not started. Data may be stale.')
 
     if args.leases:
         leases = get_leases(conf, lease_file, args.state, args.pool, args.sort)