Skip to content

requests

celery.utils.requests

send_simulation_logfiles 

send_simulation_logfiles(
    simulation_id, update_key, logfiles
)

Sends simulation logfiles to Flask backend which will save it in database Returns True if successful, False otherwise

Source code in yaptide/celery/utils/requests.py 
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
def send_simulation_logfiles(simulation_id: int, update_key: str, logfiles: dict) -> bool:
    """
    Sends simulation logfiles to Flask backend which will save it in database
    Returns True if successful, False otherwise
    """
    flask_url = os.environ.get("BACKEND_INTERNAL_URL")
    if not flask_url:
        logging.warning("Flask URL not found via BACKEND_INTERNAL_URL")
        return False
    dict_to_send = {
        "simulation_id": simulation_id,
        "update_key": update_key,
        "logfiles": logfiles,
    }
    logging.info("Sending log files to flask via %s", flask_url)
    res: requests.Response = requests.Session().post(url=f"{flask_url}/logfiles", json=dict_to_send)
    if res.status_code != 202:
        logging.warning("Saving logfiles failed: %s", res.json()["message"])
        return False
    return True

send_simulation_results 

send_simulation_results(
    simulation_id, update_key, estimators
)

Sends simulation results to flask to save it in database

Source code in yaptide/celery/utils/requests.py 
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
def send_simulation_results(simulation_id: int, update_key: str, estimators: list) -> bool:
    """Sends simulation results to flask to save it in database"""
    flask_url = os.environ.get("BACKEND_INTERNAL_URL")
    if not flask_url:
        logging.warning("Flask URL not found via BACKEND_INTERNAL_URL")
        return False
    if not update_key:
        logging.warning("Update key not found, skipping update")
        return False
    dict_to_send = {
        "simulation_id": simulation_id,
        "update_key": update_key,
        "estimators": estimators,
    }
    logging.info("Sending results to flask via %s", flask_url)
    res: requests.Response = requests.Session().post(url=f"{flask_url}/results", json=dict_to_send)
    if res.status_code != 202:
        logging.warning("Saving results failed: %s", res.json().get("message"))
        return False
    return True

send_task_update 

send_task_update(
    simulation_id, task_id, update_key, update_dict
)

Sends task status to backend which will update the database

Source code in yaptide/celery/utils/requests.py 
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
def send_task_update(simulation_id: int, task_id: int, update_key: str, update_dict: dict) -> bool:
    """Sends task status to backend which will update the database"""
    flask_url = os.environ.get("BACKEND_INTERNAL_URL")
    if not flask_url:
        logging.warning("Flask URL not found via BACKEND_INTERNAL_URL")
        return False
    if not update_key:
        logging.warning("Update key not found, skipping update")
        return False
    dict_to_send = {
        "simulation_id": simulation_id,
        "task_id": task_id,
        "update_key": update_key,
        "update_dict": update_dict
    }
    logging.debug("Sending update %s to the backend %s", dict_to_send, flask_url)
    res: requests.Response = requests.Session().post(url=f"{flask_url}/tasks", json=dict_to_send)
    if res.status_code != 202:
        logging.warning("Update_dict: %s", update_dict)
        logging.warning("Task update for %s - Failed: %s", task_id, res.json().get("message"))
        return False
    return True