diff --git a/src/config/schema-transformer/db.py b/src/config/schema-transformer/db.py index 4ba4b41c5e7..5d72cf137de 100644 --- a/src/config/schema-transformer/db.py +++ b/src/config/schema-transformer/db.py @@ -12,6 +12,7 @@ from cfgm_common.zkclient import IndexAllocator from cfgm_common.vnc_cassandra import VncCassandraClient from sandesh_common.vns.constants import SCHEMA_KEYSPACE_NAME +import uuid class SchemaTransformerDB(VncCassandraClient): @@ -214,8 +215,12 @@ def get_service_chain_ip(self, sc_name): return None, None def add_service_chain_ip(self, sc_name, ip, ipv6): - self._sc_ip_cf.insert(sc_name, {'ip_address': ip, - 'ipv6_address': ipv6}) + val = {} + if ip: + val['ip_address'] = ip + if ipv6: + val['ipv6_address'] = ipv6 + self._sc_ip_cf.insert(sc_name, val) def remove_service_chain_ip(self, sc_name): try: