Remove unused files

dockerfiles/dataExporter/code/awinlib/hdd.py

@@ -0,0 +1,49 @@
+import os
+
+from .utilities import log, runCmd
+
+
+def findHddIdInNt():
+    return {}
+
+def findHddIdInPosix():
+    hddDict = {}
+    results = runCmd("blkid").split("\n")
+    for line in results:
+        if line == "":
+            continue
+
+        hddInfoList = line.split(" ")
+        if len(hddInfoList) > 1 and hddInfoList[0].startswith("/dev/sd"):
+            hddName = hddInfoList[0].replace(":", "")
+            hddDict[hddName] = {}
+            for hddInfoline in hddInfoList[1:]:
+                kvList = hddInfoline.split("=")
+                if len(kvList) > 1:
+                    hddDict[hddName][kvList[0].replace("\"", "")] = kvList[1].replace("\"", "")
+    return hddDict
+
+
+def findHddId():
+    hddDict = {}
+    if os.name == "nt":
+        hddDict = findHddIdInNt()
+    elif os.name == "posix":
+        hddDict = findHddIdInPosix()
+    return hddDict
+
+
+def findHddTemp(hddDict):
+    for hddName in sorted(hddDict.keys()):
+        cmd = f"smartctl -A {hddName} | grep Temperature_Celsius"
+        hddTempString = runCmd(cmd)
+        hddTempList = hddTempString.split(" ")
+        hddTempList = [item for item in hddTempList if item != ""]
+        if len(hddTempList) > 9:
+            hddDict[hddName]["temp"] = int(hddTempList[9])
+
+
+def getHddTemp():
+    hddDict = findHddId()
+    findHddTemp(hddDict)
+    return hddDict

dockerfiles/dataExporter/code/prometheus_data_exporter.py

@@ -0,0 +1,88 @@
+import argparse
+import time
+
+import schedule
+from prometheus_client import Gauge, start_http_server
+
+from awinlib.PrometheusDataExporter import (
+    CpuFanTempDataExporter,
+    HddTempDataExporter,
+    KasaPowerStripDataExporter,
+    TapoT315DataExporter,
+)
+from awinlib.utilities import datetimeNow, loadJson, log
+
+ENABLE_EXPORTER = True
+DEFAULT_PORT = 8087
+DEFAULT_POLLING_INTERVAL = 60
+DEFAULT_SECRET_FILEPATH = "~/.secret"
+DEFAULT_CONFIGS = {
+    "port": DEFAULT_PORT,
+    "polling_interval": DEFAULT_POLLING_INTERVAL,
+    "secret_filepath": DEFAULT_SECRET_FILEPATH
+}
+
+
+def job_5s(gaugeDict, exporters):
+    exportData(gaugeDict, exporters)
+
+
+def job_60s(gaugeDict, exporters):
+    exportData(gaugeDict, exporters)
+
+
+def exportData(gaugeDict, exporters):
+    for exporter in exporters:
+        data = exporter.export()
+        for d in data:
+            gaugeName = d["name"]
+            gaugeDesc = d["description"]
+            gaugeValue = d["value"]
+            log(f"Set {gaugeName}: {gaugeValue}")
+            if gaugeName not in gaugeDict:
+                gaugeDict[gaugeName] = Gauge(gaugeName, gaugeDesc)
+
+            if ENABLE_EXPORTER:
+                gaugeDict[gaugeName].set(gaugeValue)
+
+
+def main(configs):
+    log(f"Start: {datetimeNow()}")
+    secrect = loadJson(configs.get("secret_filepath", DEFAULT_POLLING_INTERVAL))
+    if secrect is None:
+        log(f'Cannot read secret file({configs.get("secret_filepath", DEFAULT_POLLING_INTERVAL)})')
+        return
+
+    hddTempExporter = HddTempDataExporter()
+    cpuFanTempExporter = CpuFanTempDataExporter()
+    powerStripLivingroomExporter = KasaPowerStripDataExporter("livingroom", "192.168.1.203")
+    powerStripAtticExporter = KasaPowerStripDataExporter("attic", "192.168.1.202")
+    t315Atticexporter = TapoT315DataExporter(secrect["ip"], secrect["username"], secrect["password"], "鐵皮屋-溫濕度感測器", "attic-sensors")
+    gaugeDict = {}
+
+    schedule.every( 5).seconds.do(exportData, gaugeDict=gaugeDict, exporters=(hddTempExporter, cpuFanTempExporter, powerStripLivingroomExporter, powerStripAtticExporter))
+    schedule.every(60).seconds.do(exportData, gaugeDict=gaugeDict, exporters=(t315Atticexporter,))
+
+    ## Start Prometheus server
+    prometheusServerPort = configs.get("port", DEFAULT_PORT)
+    log(f"Start Prometheus server on port {prometheusServerPort}")
+    start_http_server(prometheusServerPort) if ENABLE_EXPORTER else None
+    while True:
+        try:
+            schedule.run_pending()
+            time.sleep(1)
+        except KeyboardInterrupt:
+            log("User stopped process.")
+            break
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument("-p", "--port", type=int, default=DEFAULT_PORT, help="The port of exportor")
+    parser.add_argument("-i", "--polling_interval", type=int, default=DEFAULT_POLLING_INTERVAL, help="Polling interval for collect HDD temperature")
+    parser.add_argument("-f", "--secrect_file", default=".secret", help="The file that contains the secrets")
+    args = parser.parse_args()
+    DEFAULT_CONFIGS["port"] = args.port
+    DEFAULT_CONFIGS["polling_interval"] = args.polling_interval
+    DEFAULT_CONFIGS["secret_filepath"] = args.secrect_file
+    main(DEFAULT_CONFIGS)