astrox-python 0.1.1__py3-none-any.whl → 0.2.0__py3-none-any.whl

astrox/__init__.py

@@ -4,7 +4,7 @@ Functions are organized by domain modules and can use package-level
 configuration without requiring explicit client management:
 
     import astrox
-    from astrox import access, coverage, components, lighting, orbits, propagator
+    from astrox import access, celestial, catalog, components, coverage, lighting, orbits, propagator, terrain
 
     astrox.configure(base_url="http://custom:8765", timeout=120)
     orbit = orbits.keplerian(...)
@@ -26,7 +26,21 @@ Raw route access is available for advanced callers:
 
 from importlib.metadata import PackageNotFoundError, version
 
-from astrox import access, coverage, components, lighting, orbits, propagator, rocket
+from astrox import (
+    access,
+    astrogator,
+    catalog,
+    cat,
+    celestial,
+    conjunction,
+    components,
+    coverage,
+    lighting,
+    orbits,
+    propagator,
+    rocket,
+    terrain,
+)
 from astrox._http import Client, configure, get_session, raw
 
 try:
@@ -39,11 +53,17 @@ __all__ = [
     "configure",
     "get_session",
     "access",
+    "astrogator",
+    "catalog",
+    "cat",
+    "celestial",
+    "conjunction",
     "coverage",
     "components",
     "lighting",
     "orbits",
     "propagator",
     "rocket",
+    "terrain",
     "raw",
 ]

astrox/astrogator/__init__.py

@@ -0,0 +1,8 @@
+"""Pythonic Astrogator mission-control sequence API."""
+
+from . import _models, _results
+from ._api import run_mcs
+from ._models import *  # noqa: F403
+from ._results import *  # noqa: F403
+
+__all__ = [*_models.__all__, *_results.__all__, "run_mcs"]  # noqa: PLE0604

astrox/astrogator/_api.py

@@ -0,0 +1,60 @@
+"""Public Astrogator RunMCS endpoint."""
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+from astrox._http import raw
+from astrox.propagator import HpopConfig
+
+from ._models import (
+    EngineConstant,
+    EntityPath,
+    MCSSegment,
+    _mission_wire,
+)
+from ._results import RunMCSResult, run_mcs_result_from_wire
+
+
+def _entities_to_wire(values: Sequence[EntityPath] | None) -> list[dict[str, object]] | None:
+    if values is None:
+        return None
+    if isinstance(values, (str, bytes)) or not isinstance(values, Sequence):
+        raise TypeError("entities must be a sequence of astrogator EntityPath values")
+    items = tuple(values)
+    if not all(isinstance(item, EntityPath) for item in items):
+        raise TypeError("entities must be a sequence of astrogator EntityPath values")
+    return [item.to_wire() for item in items]
+
+
+def run_mcs(
+    main_sequence: Sequence[MCSSegment],
+    *,
+    central_body: str = "Earth",
+    out_czml_frame_name: str = "INERTIAL",
+    compute_czml_positions: bool | None = None,
+    entities: Sequence[EntityPath] | None = None,
+    propagators: Sequence[HpopConfig] | None = None,
+    engine_models: Sequence[EngineConstant] | None = None,
+    text: str | None = None,
+) -> RunMCSResult:
+    """Run an ASTROX Astrogator mission-control sequence."""
+
+    payload = _mission_wire(
+        central_body=central_body,
+        main_sequence=main_sequence,
+        compute_czml_positions=compute_czml_positions,
+        out_czml_frame_name=out_czml_frame_name,
+        text=text,
+        propagators=propagators,
+        engine_models=engine_models,
+    )
+    entity_values = _entities_to_wire(entities)
+    if entity_values is not None:
+        payload["Entities"] = entity_values
+
+    result = raw.post("/Astrogator/RunMCS", json=payload)
+    return run_mcs_result_from_wire(result)
+
+
+__all__ = ["run_mcs"]