From 53bc8022e3beb1f817aec0d263c4d3bb379dcaf7 Mon Sep 17 00:00:00 2001
From: Viacheslav Hletenko <v.gletenko@vyos.io>
Date: Thu, 25 Aug 2022 12:26:15 +0000
Subject: op-mode: T4645: Show nat source stat missing argument --family

As we use in commit 8d4205a9 argument '--family' for the
function '_get_raw_data_rules(direction, family)' we must use it
and for 'nat.py show_statistics' as it get raw data from the same
function
---
 op-mode-definitions/nat.xml.in | 2 +-
 src/op_mode/nat.py             | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/op-mode-definitions/nat.xml.in b/op-mode-definitions/nat.xml.in
index e89c3801f..dbc06b930 100644
--- a/op-mode-definitions/nat.xml.in
+++ b/op-mode-definitions/nat.xml.in
@@ -22,7 +22,7 @@
                 <properties>
                   <help>Show statistics for configured source NAT rules</help>
                 </properties>
-                <command>${vyos_op_scripts_dir}/nat.py show_statistics --direction source</command>
+                <command>${vyos_op_scripts_dir}/nat.py show_statistics --direction source --family inet</command>
               </node>
               <node name="translations">
                 <properties>
diff --git a/src/op_mode/nat.py b/src/op_mode/nat.py
index 2dffc378b..dec04aa48 100755
--- a/src/op_mode/nat.py
+++ b/src/op_mode/nat.py
@@ -188,8 +188,8 @@ def show_rules(raw: bool, direction: str, family: str):
         return _get_formatted_output_rules(nat_rules, direction, family)
 
 
-def show_statistics(raw: bool, direction: str):
-    nat_statistics = _get_raw_data_rules(direction)
+def show_statistics(raw: bool, direction: str, family: str):
+    nat_statistics = _get_raw_data_rules(direction, family)
     if raw:
         return nat_statistics
     else:
-- 
cgit v1.2.3