summaryrefslogtreecommitdiff
path: root/modules/nifi/nifi_utils.py
diff options
context:
space:
mode:
authorRasmus Luha <rasmus.luha@ut.ee>2025-05-11 23:30:25 +0300
committerRasmus Luha <rasmus.luha@ut.ee>2025-05-11 23:30:25 +0300
commitaad1cf42f85eb1a2e2a34bf997a03f2941ec99de (patch)
tree0a20e566a0cf7b42765caf000bdfd04ae7099d67 /modules/nifi/nifi_utils.py
parent0cdd22fd88b00bc939eac7824bb1063f675cba84 (diff)
add comments
Diffstat (limited to 'modules/nifi/nifi_utils.py')
-rw-r--r--modules/nifi/nifi_utils.py24
1 files changed, 24 insertions, 0 deletions
diff --git a/modules/nifi/nifi_utils.py b/modules/nifi/nifi_utils.py
index 6028003..0dbd029 100644
--- a/modules/nifi/nifi_utils.py
+++ b/modules/nifi/nifi_utils.py
@@ -8,6 +8,15 @@ import urllib3
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
def update_template(file_path, dot_path, new_key, new_value):
+ """
+ Parameeterid:
+ file_path: malli asukoht
+ dot_path: väärtuse asuhot, mida soovitakse muuta mallis
+ new_key: võti mida soovitatakse soovitud asukohas muuta
+ new_value: väärtus, mis soovitud võtmele antakse
+
+ Uuendab etteantud väärtuste põhjal mallis võtme väärtuse.
+ """
# Step 2: Load the copied JSON
with open(file_path, "r") as f:
@@ -35,6 +44,14 @@ def update_template(file_path, dot_path, new_key, new_value):
#print("✅ Changes saved.")
def set_database_credentials(file_path,dot_path):
+ """
+ Parameeterid:
+ file_path: malli asukoht
+ dot_path: väärtuse asuhot, mida soovitakse muuta mallis
+
+ Uuendab andmebaasi autentimisväärtused
+ """
+
## Update URL
db_full_url=config.DB_URL+"/write?db="+config.DB_NAME
update_template(file_path, dot_path, "HTTP URL", db_full_url)
@@ -53,6 +70,9 @@ def set_database_credentials(file_path,dot_path):
# -H "Content-Type: application/x-www-form-urlencoded" -d 'username=lab08nifiuser&password=tartunifi2023')
def get_access_token():
+ """
+ Tagastab Nifi platvormi autentimiseks vajaliku andmekonveieri.
+ """
token_resp = requests.post(
f"{config.NIFI_HOST}/nifi-api/access/token",
headers={"Content-Type": "application/x-www-form-urlencoded"},
@@ -70,6 +90,10 @@ def get_access_token():
def upload_nifi_pipeline(token, pipeline_path, processorGroup_name, username=config.NIFI_USER, password=config.NIFI_PASS, nifi_url=config.NIFI_HOST, position_x=0, position_y=0, client_id = "unique-client-id-123"):
+ """
+ Paigaldab andmekonveieri nifi platvormile
+ """
+
with open(pipeline_path, "r") as json_file:
files = {