Append default mask field if user has not provided one (sonic-net#972)

* Append mask if user has not provided one
stepanblyschak · Jul 1, 2020 · fd52e93 · fd52e93
1 parent aae89b7
commit fd52e93
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/config/main.py b/config/main.py
@@ -2045,7 +2045,9 @@ def add(ctx, interface_name, ip_addr, gw):
             ctx.fail("'interface_name' is None!")
 
     try:
-        ipaddress.ip_network(unicode(ip_addr), strict=False)
+        net = ipaddress.ip_network(unicode(ip_addr), strict=False)
+        if '/' not in ip_addr:
+            ip_addr = str(net)
 
         if interface_name == 'eth0':
 
@@ -2102,7 +2104,9 @@ def remove(ctx, interface_name, ip_addr):
             ctx.fail("'interface_name' is None!")
 
     try:
-        ipaddress.ip_network(unicode(ip_addr), strict=False)
+        net = ipaddress.ip_network(unicode(ip_addr), strict=False)
+        if '/' not in ip_addr:
+            ip_addr = str(net)
 
         if interface_name == 'eth0':
             config_db.set_entry("MGMT_INTERFACE", (interface_name, ip_addr), None)