From b2e8ec86abe8089ed5fbd1655677889b6691397f Mon Sep 17 00:00:00 2001
From: Rasmus Luha <rasmus.luha@ut.ee>
Date: Tue, 29 Apr 2025 00:25:22 +0300
Subject: add cml variables, and platform option to config

---
 modules/telegraf/core.py           | 10 ++++++----
 modules/telegraf/telegraf_utils.py | 39 ++++++++++++++++++++++++++++++++++++--
 2 files changed, 43 insertions(+), 6 deletions(-)

(limited to 'modules/telegraf')

diff --git a/modules/telegraf/core.py b/modules/telegraf/core.py
index 3febf7b..ffef7d5 100644
--- a/modules/telegraf/core.py
+++ b/modules/telegraf/core.py
@@ -25,10 +25,12 @@ def introduction():
 ###########################
 
 
-def modify_template(new_pipeline_path, api_url):
-    #telegraf_utils.modify_output("templates/basic_ETL.toml", "urls", "testingIfWorks")
+def modify_template(new_pipeline_path, api_url, schedulingPeriod):
 
-    ## Api Url editing
+    ## Pipeline intervall
+    telegraf_utils.modify_agent(new_pipeline_path,"interval", schedulingPeriod)
+
+    ## API url 
     telegraf_utils.modify_input(new_pipeline_path,"urls", [api_url])
 
 
@@ -69,7 +71,7 @@ def build_pipeline():
     shutil.copy(f"modules/telegraf/templates/{template_name}", new_pipeline_path)
 
 
-    modify_template(new_pipeline_path, api_url)
+    modify_template(new_pipeline_path, api_url, schedulingPeriod)
     #telegraf.modify_output("templates/basic_ETL.toml", "urls", "testingIfWorks")
 
 
diff --git a/modules/telegraf/telegraf_utils.py b/modules/telegraf/telegraf_utils.py
index d6ed376..ffaddfd 100644
--- a/modules/telegraf/telegraf_utils.py
+++ b/modules/telegraf/telegraf_utils.py
@@ -4,6 +4,43 @@ def modify_input(new_pipeline_path, key, value):
     data = toml.load(new_pipeline_path)
     pluggin = data["inputs"]["http"][0]
 
+    if key in pluggin:
+
+        #print(f"Before: {key} = {pluggin[key]}")
+        pluggin[key] = value
+        #print(f"After:  {key} = {pluggin[key]}")
+
+
+    with open(new_pipeline_path, "w") as f:
+        toml.dump(data, f)
+
+
+##modify_input("templates/basic_ETL.toml", "test_pipers.toml, "urls", ["stillTesting"])
+
+
+## TODO
+def modify_agent(new_pipeline_path, key, value):
+    data = toml.load(new_pipeline_path)
+    pluggin = data["agent"]
+
+    if key in pluggin:
+
+        #print(f"Before: {key} = {pluggin[key]}")
+        pluggin[key] = value
+        #print(f"After:  {key} = {pluggin[key]}")
+
+
+    with open(new_pipeline_path, "w") as f:
+        toml.dump(data, f)
+
+
+
+
+## TODO
+def modify_output(new_pipeline_path, key, value):
+    data = toml.load(new_pipeline_path)
+    pluggin = data["outputs"]["influxdb"][0]
+
     if key in pluggin:
 
         print(f"Before: {key} = {pluggin[key]}")
@@ -13,5 +50,3 @@ def modify_input(new_pipeline_path, key, value):
 
     with open(new_pipeline_path, "w") as f:
         toml.dump(data, f)
-
-#modify_input("templates/basic_ETL.toml", "test_pipers.toml, "urls", ["stillTesting"])
-- 
cgit v1.2.3