feat: notify scheduler when MIB starts

pyproject.toml

@@ -29,6 +29,7 @@ PyYAML = "^5.4.1"
 Flask = "^1.1.2"
 Flask-AutoIndex = "^0.6.6"
 types-PyYAML = "^5.4.3"
+checksumdir = "^1.2.0"
 
 [tool.poetry.dev-dependencies]
 mongomock = "^3.22.1"

splunk_connect_for_snmp_mib_server/mib_server.py

@@ -14,7 +14,10 @@
 #    limitations under the License.
 #   ########################################################################
 import logging
+import os
 
+import requests
+from checksumdir import dirhash
 from flask import Flask, request
 from flask_autoindex import AutoIndex
 
@@ -80,3 +83,8 @@ def get_profiles():
     def run_mib_server(self):
         # poetry run fails when debug=True
         self._flask_app.run(host="0.0.0.0", port=self._args.port)
+
+    def notify_startup(self):
+        profiles_hash = dirhash("profiles")
+        params = {"profiles_hash": profiles_hash}
+        requests.get(os.environ["SCHEDULER_REFRESH_URI"], params=params, timeout=3)

splunk_connect_for_snmp_mib_server/snmp_mib_server.py

@@ -15,6 +15,7 @@
 #   ########################################################################
 import argparse
 import logging
+import threading
 import time
 
 import yaml
@@ -77,7 +78,9 @@ def main():
 
     upload_mibs(server_config)
     mib_server = MibServer(args, server_config)
-    mib_server.run_mib_server()
+    threading.Thread(target=mib_server.run_mib_server).start()
+    time.sleep(5)
+    mib_server.notify_startup()
 
 
 if __name__ == "__main__":