init commit

1 files changed, 30 insertions, 0 deletions

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()