swag:auto-uptime-kuma Initial release of the mod

root/app/auto-uptime-kuma.py

@@ -0,0 +1,68 @@
+from swagDocker import SwagDocker
+from swagUptimeKuma import SwagUptimeKuma
+import sys
+import argparse
+import os
+
+
+def parseCommandLine():
+    """
+    Different application behavior if executed from CLI
+    """
+    parser = argparse.ArgumentParser()
+    parser.add_argument('-purge', action='store_true')
+    args = parser.parse_args()
+
+    if (args.purge == True):
+        swagUptimeKuma.purgeData()
+        swagUptimeKuma.disconnect()
+        sys.exit(0)
+
+
+def addOrUpdateMonitors(domainName, swagContainers):
+    for swagContainer in swagContainers:
+        containerConfig = swagDocker.parseContainerLabels(
+            swagContainer.labels, ".monitor.")
+        containerName = swagContainer.name
+        monitorData = swagUptimeKuma.parseMonitorData(
+            containerName, domainName, containerConfig)
+
+        if (not swagUptimeKuma.monitorExists(containerName)):
+            swagUptimeKuma.addMonitor(containerName, domainName, monitorData)
+        else:
+            swagUptimeKuma.updateMonitor(
+                containerName, domainName, monitorData)
+
+
+def getMonitorsToBeRemoved(swagContainers, apiMonitors):
+    # Monitors to be removed are those that no longer have an existing container
+    # Monitor <-> Container link is done by comparing the container name with the monitor swag tag value
+    existingMonitorNames = [swagUptimeKuma.getMonitorSwagTagValue(
+        monitor) for monitor in apiMonitors]
+    existingContainerNames = [container.name for container in swagContainers]
+
+    monitorsToBeRemoved = [
+        containerName for containerName in existingMonitorNames if containerName not in existingContainerNames]
+    return monitorsToBeRemoved
+
+
+if __name__ == "__main__":
+    url = os.environ['UPTIME_KUMA_URL']
+    username = os.environ['UPTIME_KUMA_USERNAME']
+    password = os.environ['UPTIME_KUMA_PASSWORD']
+    domainName = os.environ['URL']
+
+    swagDocker = SwagDocker("swag.uptime-kuma")
+    swagUptimeKuma = SwagUptimeKuma(url, username, password)
+
+    parseCommandLine()
+
+    swagContainers = swagDocker.getSwagContainers()
+
+    addOrUpdateMonitors(domainName, swagContainers)
+
+    monitorsToBeRemoved = getMonitorsToBeRemoved(
+        swagContainers, swagUptimeKuma.apiMonitors)
+    swagUptimeKuma.deleteMonitors(monitorsToBeRemoved)
+
+    swagUptimeKuma.disconnect()