FastAPI app

fastapi/README.md

@@ -0,0 +1,26 @@
+# SpiceQL FastAPI App
+
+## Create local instance
+
+### 1. Create conda environment
+Create the conda environment to run your local instance in:
+```
+conda create -n spiceql-api -f environment.yaml
+```
+
+### 2. Set environment variables
+Similarly to your SpiceQL conda environment, set `SPICEROOT` or `ISISDATA` to your ISIS data area. You may also need to set `SSPICE_DEBUG` to any value, like `True`.
+
+To set an environment variable within the scope of your conda environment:
+```
+conda activate spiceql-api
+conda env config vars set SPICEROOT=/path/to/isis_data
+```
+
+### 3. Run the app
+Inside the `app/` dir, run the following command:
+```
+uvicorn app.main:app --reload --port 8080
+```
+
+You can access the Swagger UI of all the endpoints at http://127.0.0.1:8080/docs.

fastapi/app/main.py

@@ -0,0 +1,227 @@
+"""Module providing SpiceQL endpoints"""
+
+from ast import literal_eval
+from typing import Annotated, Any
+from fastapi import FastAPI, Query
+from pydantic import BaseModel, Field
+import pyspiceql
+
+
+# Models
+class MessageItem(BaseModel):
+    message: str
+
+class ResultModel(BaseModel):
+    result: Any = Field(serialization_alias='return')
+
+class ErrorModel(BaseModel):
+    error: str
+
+class ResponseModel(BaseModel):
+    statusCode: int
+    body: ResultModel | ErrorModel
+
+# Create FastAPI instance
+app = FastAPI()
+
+# General endpoints
+@app.get("/")
+async def root():
+    return {"message": "Visit the /docs endpoint to see the Swagger UI."}
+
+@app.post("/customMessage")
+async def message(
+    message_item: MessageItem
+    ):
+    return {"message": message_item.message}
+
+
+# SpiceQL endpoints
+@app.get("/getTargetStates")
+async def getTargetStates(
+    ets: Annotated[list[float], Query()] | str,
+    target: str,
+    observer: str,
+    frame: str,
+    abcorr: str,
+    mission: str,
+    ckQuality: str = "",
+    spkQuality: str = "",
+    searchKernels: bool = False):
+    try:
+        if isinstance(ets, str):
+            ets = literal_eval(ets)
+        result = pyspiceql.getTargetStates(ets, target, observer, frame, abcorr, mission, ckQuality, spkQuality, searchKernels)
+        body = ResultModel(result=result)
+        return ResponseModel(statusCode=200, body=body)
+    except Exception as e:
+        body = ErrorModel(error=str(e))
+        return ResponseModel(statusCode=500, body=body)
+
+@app.get("/getTargetOrientations")
+async def getTargetOrientations(
+    ets: Annotated[list[float], Query()] | str,
+    toFrame: int,
+    refFrame: int,
+    mission: str,
+    ckQuality: str = "",
+    searchKernels: bool = False):
+    try:
+        if isinstance(ets, str):
+            ets = literal_eval(ets)
+        result = pyspiceql.getTargetOrientations(ets, toFrame, refFrame, mission, ckQuality, searchKernels)
+        body = ResultModel(result=result)  
+        return ResponseModel(statusCode=200, body=body)
+    except Exception as e:
+        body = ErrorModel(error=str(e))
+        return ResponseModel(statusCode=500, body=body)
+
+@app.get("/strSclkToEt")
+async def strSclkToEt(
+    frameCode: int,
+    sclk: str,
+    mission: str,
+    searchKernels: bool = False):
+    try:
+        result = pyspiceql.strSclkToEt(frameCode, sclk, mission, searchKernels)
+        body = ResultModel(result=result)
+        return ResponseModel(statusCode=200, body=body)
+    except Exception as e:
+        body = ErrorModel(error=str(e))
+        return ResponseModel(statusCode=500, body=body)
+
+@app.get("/doubleSclkToEt")
+async def doubleSclkToEt(
+    frameCode: int,
+    sclk: float,
+    mission: str,
+    searchKernels: bool = False):
+    try:
+        result = pyspiceql.doubleSclkToEt(frameCode, sclk, mission, searchKernels)
+        body = ResultModel(result=result)
+        return ResponseModel(statusCode=200, body=body)
+    except Exception as e:
+        body = ErrorModel(error=str(e))
+        return ResponseModel(statusCode=500, body=body)
+
+@app.get("/utcToEt")
+async def utcToEt(
+    utc: str,
+    searchKernels: bool = False):
+    try:
+        result = pyspiceql.utcToEt(utc, searchKernels)
+        body = ResultModel(result=result)
+        return ResponseModel(statusCode=200, body=body)
+    except Exception as e:
+        body = ErrorModel(error=str(e))
+        return ResponseModel(statusCode=500, body=body)
+
+@app.get("/translateNameToCode")
+async def translateNameToCode(
+    frame: str,
+    mission: str,
+    searchKernels: bool = False):
+    try:
+        result = pyspiceql.translateNameToCode(frame, mission, searchKernels)
+        body = ResultModel(result=result)
+        return ResponseModel(statusCode=200, body=body)
+    except Exception as e:
+        body = ErrorModel(error=str(e))
+        return ResponseModel(statusCode=500, body=body)
+
+@app.get("/translateCodeToName")
+async def translateCodeToName(
+    frame: int,
+    mission: str,
+    searchKernels: bool = False):
+    try:
+        result = pyspiceql.translateCodeToName(frame, mission, searchKernels)
+        body = ResultModel(result=result)
+        return ResponseModel(statusCode=200, body=body)
+    except Exception as e:
+        body = ErrorModel(error=str(e))
+        return ResponseModel(statusCode=500, body=body)
+
+@app.get("/getFrameInfo")
+async def getFrameInfo(
+    frame: int,
+    mission: str,
+    searchKernels: bool = False):
+    try:
+        result = pyspiceql.getFrameInfo(frame, mission, searchKernels)
+        body = ResultModel(result=result)
+        return ResponseModel(statusCode=200, body=body)
+    except Exception as e:
+        body = ErrorModel(error=str(e))
+        return ResponseModel(statusCode=500, body=body)
+
+@app.get("/getTargetFrameInfo")
+async def getTargetFrameInfo(
+    targetId: int,
+    mission: str,
+    searchKernels: bool = False):
+    try:
+        result = pyspiceql.getTargetFrameInfo(targetId, mission, searchKernels)
+        body = ResultModel(result=result)
+        return ResponseModel(statusCode=200, body=body)
+    except Exception as e:
+        body = ErrorModel(error=str(e))
+        return ResponseModel(statusCode=500, body=body)
+
+@app.get("/findMissionKeywords")
+async def findMissionKeywords(
+    key: str,
+    mission: str,
+    searchKernels: bool = False):
+    try:
+        result = pyspiceql.findMissionKeywords(key, mission, searchKernels)
+        body = ResultModel(result=result)
+        return ResponseModel(statusCode=200, body=body)
+    except Exception as e:
+        body = ErrorModel(error=str(e))
+        return ResponseModel(statusCode=500, body=body)
+
+@app.get("/findTargetKeywords")
+async def findTargetKeywords(
+    key: str,
+    mission: str,
+    searchKernels: bool = False):
+    try:
+        result = pyspiceql.findTargetKeywords(key, mission, searchKernels)
+        body = ResultModel(result=result)
+        return ResponseModel(statusCode=200, body=body)
+    except Exception as e:
+        body = ErrorModel(error=str(e))
+        return ResponseModel(statusCode=500, body=body)
+
+@app.get("/frameTrace")
+async def frameTrace(
+    et: float,
+    initialFrame: int,
+    mission: str,
+    ckQuality: str = "",
+    searchKernels: bool = False):
+    try:
+        result = pyspiceql.frameTrace(et, initialFrame, mission, ckQuality, searchKernels)
+        body = ResultModel(result=result)
+        return ResponseModel(statusCode=200, body=body)
+    except Exception as e:
+        body = ErrorModel(error=str(e))
+        return ResponseModel(statusCode=500, body=body)
+
+@app.get("/extractExactCkTimes")
+async def extractExactCkTimes(
+    observStart: float,
+    observEnd: float,
+    targetFrame: int,
+    mission: str,
+    ckQuality: str = "",
+    searchKernels: bool = False):
+    try:
+        result = pyspiceql.extractExactCkTimes(observStart, observEnd, targetFrame, mission, ckQuality, searchKernels)
+        body = ResultModel(result=result)
+        return ResponseModel(statusCode=200, body=body)
+    except Exception as e:
+        body = ErrorModel(error=str(e))
+        return ResponseModel(statusCode=500, body=body)
+

fastapi/environment.yml

@@ -0,0 +1,8 @@
+channels:
+  - conda-forge
+
+dependencies:
+  - fastapi
+  - pydantic==2.6.3
+  - spiceql
+  - uvicorn