diff --git a/snippets/explore_activities.py b/snippets/explore_activities.py index 6c56440..9614996 100644 --- a/snippets/explore_activities.py +++ b/snippets/explore_activities.py @@ -27,25 +27,16 @@ def main(): # ----------------- # Find project by name name_to_find = "212096 - 401 FIRST STREET (College Station)" # replace with your project name - projects = projects_api.get_projects() - project_id = projects[0]["id"] # default to first project - for project in projects: - if project["name"] == name_to_find: - print(f"Found Project: {project['id']} - {project['name']}") - project_id = project["id"] - break + project = projects_api.find_project_by_name(name=name_to_find) + project_id = project["id"] # Find scenario by name scenario_to_find = "Full Schedule" # replace with your scenario name - scenarios = scenarios_api.get_scenarios(project_id=project_id) - print(json.dumps(scenarios, indent=4)) - scenario_id = scenarios[0]["id"] # default to first scenario - for scenario in scenarios: - if scenario["name"] == scenario_to_find: - print(f"Found Scenario:") - print(json.dumps(scenario, indent=4)) - scenario_id = scenario["id"] - # don't break since there are might be multiple matching scenarios and we want to get the latest + matching_scenarios = scenarios_api.find_scenario_by_name( + project_id=project_id, + scenario_name=scenario_to_find + ) + scenario_id = matching_scenarios[-1].get("id") print("Get All Activities") activities = activity_api.get_activities( diff --git a/snippets/explore_scenarios.py b/snippets/explore_scenarios.py index f62e290..4fe194d 100644 --- a/snippets/explore_scenarios.py +++ b/snippets/explore_scenarios.py @@ -39,12 +39,12 @@ def main(): # View Scenario Details # --------------------- # Get latest scenario for "Full Schedule" - scenario_id = scenarios[0]["id"] # default to the first scenario in the list - for scenario in scenarios: - if scenario["name"] == "Full Schedule": - scenario_id = scenario["id"] - print("Scenario Summay: 'Full Schedule'") - print(json.dumps(scenario, indent=4)) + scenario_to_find = "Full Schedule" # replace with your scenario name + matching_scenarios = scenarios_api.find_scenario_by_name( + project_id=project_id, + scenario_name=scenario_to_find + ) + scenario_id = matching_scenarios[-1].get("id") print(f"Getting scenario details for Scenario {scenario_id}") scenario_details = scenarios_api.get_scenario_details( diff --git a/snippets/plot_earned_days.py b/snippets/plot_earned_days.py index 4df779f..e115961 100644 --- a/snippets/plot_earned_days.py +++ b/snippets/plot_earned_days.py @@ -25,25 +25,16 @@ def main(): # --------------------------------------------------------- # Find project by name name_to_find = "212096 - 401 FIRST STREET (College Station)" # replace with your project name - projects = projects_api.get_projects() - project_id = projects[0]["id"] # default to first project - for project in projects: - if project["name"] == name_to_find: - print(f"Found Project: {project['id']} - {project['name']}") - project_id = project["id"] - break + project = projects_api.find_project_by_name(name=name_to_find) + project_id = project["id"] # Find scenario by name scenario_to_find = "Full Schedule" # replace with your scenario name - scenarios = scenarios_api.get_scenarios(project_id=project_id) - print(json.dumps(scenarios, indent=4)) - scenario_id = scenarios[0]["id"] # default to first scenario - for scenario in scenarios: - if scenario["name"] == scenario_to_find: - print(f"Found Scenario:") - print(json.dumps(scenario, indent=4)) - scenario_id = scenario["id"] - # don't break since there are might be multiple matching scenarios and we want to get the latest + matching_scenarios = scenarios_api.find_scenario_by_name( + project_id=project_id, + scenario_name=scenario_to_find + ) + scenario_id = matching_scenarios[-1].get("id") # Check data complete_curve = scenarios_api.get_earned_schedule_curve(