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

* Append mask if user has not provided one
sonic-net · 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)