feat: adicionar método que insere folha criando árvore caso não exista

backend/bu_service/app/adapter/tlmanager_adapter.py

@@ -1,11 +1,13 @@
-import requests
 import os
+
 import dotenv
+import requests
 
 dotenv.load_dotenv()
 
 TL_MANAGER_URL = os.getenv('TL_MANAGER_URL')
 
+
 def get_trees():
     return requests.get(f'{TL_MANAGER_URL}/').json()
 

backend/bu_service/app/controllers/bu_controller.py

@@ -1,4 +1,5 @@
 import logging
+
 from fastapi import APIRouter, UploadFile, File
 
 from app.services import bu_service
@@ -12,5 +13,5 @@ async def create(file: UploadFile = File(...)):
     file_content = await file.read()
 
     logging.info(f"Received file {file_name}")
-    bu_service.insert(file_content, file_name)
-    return {"status": "ok"}
+    bu_parsed = bu_service.insert(file_content, file_name)
+    return {"status": "ok", "bu": bu_parsed.dict(exclude={"bu"})}

backend/bu_service/app/services/tlmanager_service.py

@@ -0,0 +1,22 @@
+import logging
+
+from app.adapter.tlmanager_adapter import *
+
+TREE_DEFAULT_COMMITMENT_SIZE = os.getenv('TREE_DEFAULT_COMMITMENT_SIZE')
+
+
+def insert_creating_tree_if_not_exists(tree_name, data, commitment_size=TREE_DEFAULT_COMMITMENT_SIZE):
+    insert_leaf_response = insert_leaf(tree_name, data)
+    logging.info(insert_leaf_response.json())
+    if insert_leaf_response.status_code != 200 and insert_leaf_response.json().get('message') == "Tree does not exist":
+        logging.info(f"Creating tree with name {tree_name}")
+        create_tree(tree_name, commitment_size)
+
+        logging.info(f"Trying to insert leaf for tree {tree_name} again")
+        insert_leaf_response = insert_leaf(tree_name, data)
+        logging.info(insert_leaf_response.json())
+
+        if insert_leaf_response.status_code != 200:
+            raise Exception("Failed to insert leaf")
+
+    return insert_leaf_response