diff options
Diffstat (limited to 'modules/nifi/core.py')
-rw-r--r-- | modules/nifi/core.py | 35 |
1 files changed, 33 insertions, 2 deletions
diff --git a/modules/nifi/core.py b/modules/nifi/core.py index bdac035..6e2f2e7 100644 --- a/modules/nifi/core.py +++ b/modules/nifi/core.py @@ -3,6 +3,7 @@ from rich.console import Console from common import core as common import sys +import json def introduction(): @@ -11,7 +12,27 @@ def introduction(): console.print(ascii_art, style="cyan") print("Valisid Nifi Platformi!\n") + +## TODO +def set_processor_property(pipeline, processor_name, property_key, property_value): + for processor in pipeline['flowContents']['processors']: + if processor['name'] == processor_name: + processor['properties'][property_key] = property_value + print(f"Updated '{property_key}' in processor '{processor_name}'") + return + print(f"Processor '{processor_name}' not found.") + + + + + + + def build_pipeline(): + + chosen_json_values = [] + + ##Getting API url and json values while True: api_url = input("Palun sisesta andmete API URL: ").strip() username = "placeholder" @@ -22,8 +43,18 @@ def build_pipeline(): username=input("Sisesta kasutajanimi: ") passwd=input("Sisesta parool: ") - if common.is_app_url_correct(api_url,needs_auth,username,passwd): - break # Exit loop if URL is correct + json_data, api_url_correct = common.is_app_url_correct(api_url,needs_auth,username,passwd) + + ## TODO itemite eemaldamise v6malus + if api_url_correct: + while True: + chosen_json_values.append(common.inspect_json_top_level(json_data)) + print("Oled hetkel valinud järgmised väärtused:", chosen_json_values) + choose_another = common.ask_binary_input(prompt="\nKas soovid (v)alida veel mõne väärtuse või liikuda (e)dasi?(v/e): ",valikud=["v","e"]).strip().lower() + + if choose_another == 'e': + return chosen_json_values + else: choice = common.ask_binary_input(prompt="\nKas soovid URL-i (m)uuta URL-i või (v)äljuda?(m/v): ",valikud=["m","v"]).strip().lower() if choice == 'v': |