Convert snmp.yml to configdb

dockers/docker-snmp/Dockerfile.j2

@@ -68,6 +68,7 @@ RUN apt-get -y purge     \
     rm -rf /debs /python-wheels ~/.cache
 
 COPY ["start.sh", "/usr/bin/"]
+COPY ["snmp_yml_to_configdb.py", "/usr/bin/"]
 COPY ["supervisord.conf", "/etc/supervisor/conf.d/"]
 COPY ["*.j2", "/usr/share/sonic/templates/"]
 COPY ["files/supervisor-proc-exit-listener", "/usr/bin"]

dockers/docker-snmp/snmp_yml_to_configdb.py

@@ -0,0 +1,56 @@
+#!/usr/bin/env python3
+
+import os
+import sys 
+
+import yaml
+from sonic_py_common.logger import Logger
+from swsssdk import ConfigDBConnector
+
+db = ConfigDBConnector()
+db.connect()
+
+
+SYSLOG_IDENTIFIER = 'snmp_yml_to_configdb.py'
+logger = Logger(SYSLOG_IDENTIFIER)
+logger.set_min_log_priority_info()
+
+snmp_comm_config_db = db.get_table('SNMP_COMMUNITY')
+snmp_config_db_communities = snmp_comm_config_db.keys()
+snmp_general_config_db = db.get_table('SNMP')
+snmp_general_keys = snmp_general_config_db.keys()
+
+full_snmp_comm_list = ['snmp_rocommunity', 'snmp_rocommunities', 'snmp_rwcommunity', 'snmp_rwcommunities']
+
+if not os.path.exists('/etc/sonic/snmp.yml'):
+    logger.log_info('/etc/sonic/snmp.yml does not exist')
+    sys.exit(1)
+
+with open('/etc/sonic/snmp.yml', 'r') as yaml_file:
+    yaml_snmp_info = yaml.load(yaml_file, Loader=yaml.FullLoader)
+
+for comm_type in full_snmp_comm_list:
+    if comm_type in yaml_snmp_info.keys():
+        if comm_type.startswith('snmp_rocommunities'):
+            for community in yaml_snmp_info[comm_type]:
+                if community not in snmp_config_db_communities:
+                    db.set_entry('SNMP_COMMUNITY', community, {"TYPE": "RO"})
+        elif comm_type.startswith('snmp_rocommunity'):
+            community = yaml_snmp_info['snmp_rocommunity']
+            if community not in snmp_config_db_communities:
+                db.set_entry('SNMP_COMMUNITY', community, {"TYPE": "RO"})
+        elif comm_type.startswith('snmp_rwcommunities'):
+            for community in yaml_snmp_info[comm_type]:
+                if community not in snmp_config_db_communities:
+                    db.set_entry('SNMP_COMMUNITY', community, {"TYPE": "RW"})
+        elif comm_type.startswith('snmp_rwcommunity'):
+            community = yaml_snmp_info['snmp_rwcommunity']
+            if community not in snmp_config_db_communities:
+                db.set_entry('SNMP_COMMUNITY', community, {"TYPE": "RW"})
+
+if yaml_snmp_info.get('snmp_location'):
+    if 'LOCATION' not in snmp_general_keys:
+        db.set_entry('SNMP', 'LOCATION', {'Location': yaml_snmp_info['snmp_location']})
+else:
+    logger.log_info('snmp_location does not exist in snmp.yml file')
+    sys.exit(1)

dockers/docker-snmp/start.sh

@@ -2,6 +2,9 @@
 
 mkdir -p /etc/ssw /etc/snmp
 
+# Parse snmp.yml and insert the data in Config DB
+/usr/bin/snmp_yml_to_configdb.py
+
 SONIC_CFGGEN_ARGS=" \
     -d \
     -y /etc/sonic/sonic_version.yml \