From 488f00e56f8f5f6c5b7b3968ed9d95dd6e8edcb6 Mon Sep 17 00:00:00 2001
From: Rasmus Luha <rasmus.luha@ut.ee>
Date: Sat, 5 Apr 2025 01:03:31 +0300
Subject: init commit

---
 main.py | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 main.py

(limited to 'main.py')

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..06c56b5
--- /dev/null
+++ b/main.py
@@ -0,0 +1,30 @@
+from modules.nifi import core as nifi
+from modules.telegraf import core as telegraf
+
+AVAILABLE_PLATFORMS = {
+    "1": ("Nifi", nifi),
+    "2": ("Telegraf", telegraf)}
+
+
+def list_platforms():
+    print("Available platforms:")
+    for key, (name, _) in AVAILABLE_PLATFORMS.items():
+        print(f"{key}. {name}")
+
+
+def main():
+    list_platforms()
+    plat_choice = input("Palun vali platform (number): ").strip()
+
+    platform = AVAILABLE_PLATFORMS.get(plat_choice)
+    if not platform:
+        print("Ebaõnnestunud valik ...")
+        return
+
+    name, module = platform
+    module.introduction()
+    module.build_pipeline()
+
+
+if __name__ == "__main__":
+    main()
-- 
cgit v1.2.3