-
Notifications
You must be signed in to change notification settings - Fork 19
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[monitorlib, uss_qualifier] Add flight_planning client and resource (#…
…284) * Add flight_planning client and resource * Fix message signing * Check execution style
- Loading branch information
1 parent
726592d
commit 8a39765
Showing
8 changed files
with
243 additions
and
37 deletions.
There are no files selected for viewing
175 changes: 175 additions & 0 deletions
175
monitoring/monitorlib/clients/flight_planning/client_v1.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,175 @@ | ||
| import uuid | ||
|
|
||
| from implicitdict import ImplicitDict | ||
| from monitoring.monitorlib.clients.flight_planning.client import ( | ||
| FlightPlannerClient, | ||
| PlanningActivityError, | ||
| ) | ||
| from monitoring.monitorlib.clients.flight_planning.test_preparation import ( | ||
| TestPreparationActivityResponse, | ||
| ) | ||
|
|
||
| from monitoring.monitorlib.clients.flight_planning.flight_info import ( | ||
| FlightInfo, | ||
| FlightID, | ||
| ExecutionStyle, | ||
| ) | ||
| from monitoring.monitorlib.clients.flight_planning.planning import ( | ||
| PlanningActivityResponse, | ||
| ) | ||
| from monitoring.monitorlib.fetch import query_and_describe | ||
| from monitoring.monitorlib.geotemporal import Volume4D | ||
| from monitoring.monitorlib.infrastructure import UTMClientSession | ||
|
|
||
| from uas_standards.interuss.automated_testing.flight_planning.v1 import api | ||
| from uas_standards.interuss.automated_testing.flight_planning.v1.constants import Scope | ||
|
|
||
|
|
||
| class V1FlightPlannerClient(FlightPlannerClient): | ||
| _session: UTMClientSession | ||
|
|
||
| def __init__(self, session: UTMClientSession): | ||
| self._session = session | ||
|
|
||
| def _inject( | ||
| self, | ||
| flight_plan_id: FlightID, | ||
| flight_info: FlightInfo, | ||
| execution_style: ExecutionStyle, | ||
| ) -> PlanningActivityResponse: | ||
| flight_plan = ImplicitDict.parse(flight_info, api.FlightPlan) | ||
| req = api.UpsertFlightPlanRequest( | ||
| flight_plan=flight_plan, | ||
| execution_style=execution_style, | ||
| request_id=str(uuid.uuid4()), | ||
| ) | ||
|
|
||
| op = api.OPERATIONS[api.OperationID.UpsertFlightPlan] | ||
| url = op.path.format(flight_plan_id=flight_plan_id) | ||
| query = query_and_describe( | ||
| self._session, op.verb, url, json=req, scope=Scope.Plan | ||
| ) | ||
| if query.status_code != 200 and query.status_code != 201: | ||
| raise PlanningActivityError( | ||
| f"Attempt to plan flight returned status {query.status_code} rather than 200 as expected", | ||
| query, | ||
| ) | ||
| try: | ||
| resp: api.UpsertFlightPlanResponse = ImplicitDict.parse( | ||
| query.response.json, api.UpsertFlightPlanResponse | ||
| ) | ||
| except ValueError as e: | ||
| raise PlanningActivityError( | ||
| f"Response to plan flight could not be parsed: {str(e)}", query | ||
| ) | ||
|
|
||
| response = PlanningActivityResponse( | ||
| flight_id=flight_plan_id, | ||
| queries=[query], | ||
| activity_result=resp.planning_result, | ||
| flight_plan_status=resp.flight_plan_status, | ||
| ) | ||
| return response | ||
|
|
||
| def try_plan_flight( | ||
| self, flight_info: FlightInfo, execution_style: ExecutionStyle | ||
| ) -> PlanningActivityResponse: | ||
| return self._inject(str(uuid.uuid4()), flight_info, execution_style) | ||
|
|
||
| def try_update_flight( | ||
| self, | ||
| flight_id: FlightID, | ||
| updated_flight_info: FlightInfo, | ||
| execution_style: ExecutionStyle, | ||
| ) -> PlanningActivityResponse: | ||
| return self._inject(flight_id, updated_flight_info, execution_style) | ||
|
|
||
| def try_end_flight( | ||
| self, flight_id: FlightID, execution_style: ExecutionStyle | ||
| ) -> PlanningActivityResponse: | ||
| if execution_style != ExecutionStyle.IfAllowed: | ||
| raise NotImplementedError( | ||
| "Only IfAllowed execution style is currently allowed" | ||
| ) | ||
| op = api.OPERATIONS[api.OperationID.DeleteFlightPlan] | ||
| url = op.path.format(flight_plan_id=flight_id) | ||
| query = query_and_describe(self._session, op.verb, url, scope=Scope.Plan) | ||
| if query.status_code != 200: | ||
| raise PlanningActivityError( | ||
| f"Attempt to delete flight plan returned status {query.status_code} rather than 200 as expected", | ||
| query, | ||
| ) | ||
| try: | ||
| resp: api.DeleteFlightPlanResponse = ImplicitDict.parse( | ||
| query.response.json, api.DeleteFlightPlanResponse | ||
| ) | ||
| except ValueError as e: | ||
| raise PlanningActivityError( | ||
| f"Response to delete flight plan could not be parsed: {str(e)}", query | ||
| ) | ||
|
|
||
| response = PlanningActivityResponse( | ||
| flight_id=flight_id, | ||
| queries=[query], | ||
| activity_result=resp.planning_result, | ||
| flight_plan_status=resp.flight_plan_status, | ||
| ) | ||
| return response | ||
|
|
||
| def report_readiness(self) -> TestPreparationActivityResponse: | ||
| op = api.OPERATIONS[api.OperationID.GetStatus] | ||
| query = query_and_describe( | ||
| self._session, op.verb, op.path, scope=Scope.DirectAutomatedTest | ||
| ) | ||
| if query.status_code != 200: | ||
| raise PlanningActivityError( | ||
| f"Attempt to get interface status returned status {query.status_code} rather than 200 as expected", | ||
| query, | ||
| ) | ||
| try: | ||
| resp: api.StatusResponse = ImplicitDict.parse( | ||
| query.response.json, api.StatusResponse | ||
| ) | ||
| except ValueError as e: | ||
| raise PlanningActivityError( | ||
| f"Response to get interface status could not be parsed: {str(e)}", query | ||
| ) | ||
|
|
||
| if resp.status == api.StatusResponseStatus.Ready: | ||
| errors = [] | ||
| elif resp.status == api.StatusResponseStatus.Starting: | ||
| errors = ["Flight planning v1 interface is still starting (not ready)"] | ||
| else: | ||
| errors = [f"Unrecognized status '{resp.status}'"] | ||
|
|
||
| return TestPreparationActivityResponse(errors=errors, queries=[query]) | ||
|
|
||
| def clear_area(self, area: Volume4D) -> TestPreparationActivityResponse: | ||
| req = api.ClearAreaRequest( | ||
| request_id=str(uuid.uuid4()), extent=area.to_interuss_scd_api() | ||
| ) | ||
|
|
||
| op = api.OPERATIONS[api.OperationID.ClearArea] | ||
| query = query_and_describe( | ||
| self._session, op.verb, op.path, json=req, scope=Scope.DirectAutomatedTest | ||
| ) | ||
| if query.status_code != 200: | ||
| raise PlanningActivityError( | ||
| f"Attempt to clear area returned status {query.status_code} rather than 200 as expected", | ||
| query, | ||
| ) | ||
| try: | ||
| resp: api.ClearAreaResponse = ImplicitDict.parse( | ||
| query.response.json, api.ClearAreaResponse | ||
| ) | ||
| except ValueError as e: | ||
| raise PlanningActivityError( | ||
| f"Response to clear area could not be parsed: {str(e)}", query | ||
| ) | ||
|
|
||
| if resp.outcome.success: | ||
| errors = None | ||
| else: | ||
| errors = [f"[{resp.outcome.timestamp}]: {resp.outcome.message}"] | ||
|
|
||
| return TestPreparationActivityResponse(errors=errors, queries=[query]) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.