Refactor server into submodule

This commit is contained in:
Richard Bowman 2024-08-09 00:49:24 +01:00
parent 3abddc3b39
commit c32b44d1a6
4 changed files with 78 additions and 71 deletions

View file

@ -0,0 +1,43 @@
from __future__ import annotations
from labthings_fastapi.thing_server import ThingServer
from labthings_sangaboard import SangaboardThing
from labthings_picamera2.thing import StreamingPiCamera2
from ..things.autofocus import AutofocusThing
from ..things.camera_stage_mapping import CameraStageMapper
from ..things.system_control import SystemControlThing
from ..things.settings_manager import SettingsManager
from ..things.auto_recentre_stage import RecentringThing
from ..things.smart_scan import SmartScanThing, BackgroundDetectThing
from ..things.test import APITestThing
from .serve_static_files import add_static_files
from .legacy_api import add_v2_endpoints
from ..logging import configure_logging, retrieve_log
configure_logging()
thing_server = ThingServer()
thing_server.add_thing(StreamingPiCamera2(), "/camera/")
thing_server.add_thing(SangaboardThing(), "/stage/")
thing_server.add_thing(RecentringThing(), "/auto_recentre_stage/")
thing_server.add_thing(AutofocusThing(), "/autofocus/")
thing_server.add_thing(CameraStageMapper(), "/camera_stage_mapping/")
thing_server.add_thing(SystemControlThing(), "/system_control/")
thing_server.add_thing(SettingsManager(), "/settings/")
thing_server.add_thing(SmartScanThing("application/openflexure-stitching/.venv/bin/openflexure-stitch"), "/smart_scan/")
thing_server.add_thing(BackgroundDetectThing(), "/background_detect/")
thing_server.add_thing(APITestThing(), "/api_test/")
try:
add_static_files(thing_server.app)
except RuntimeError:
print("Failed to add static files - you will have to do without them!")
add_v2_endpoints(thing_server) # Add the v2 endpoints for compatibility with OpenFlexure Connect
# Add an endpoint to get the log
thing_server.app.get("/log/")(retrieve_log)
app = thing_server.app

View file

@ -0,0 +1,35 @@
from . import ThingServer
from fastapi import Response
from socket import gethostname
def add_v2_endpoints(thing_server: ThingServer):
app = thing_server.app
# TODO: update openflexure connect to make this unnecessary!!
# The endpoints below fool OpenFlexure Connect into thinking we are a
# v2 microscope, so we show up correctly.
# This is necessary until Connect is rebuilt.
@app.get("/routes")
def routes_stub() -> dict[str, dict]:
"""A stub list of routes, used by OF Connect to identify the microscope"""
fake_routes = [
"/api/v2/",
"/api/v2/streams/snapshot",
"/api/v2/instrument/settings/name"
]
return {url: {"url": url, "methods": ["GET"]} for url in fake_routes}
class JPEGResponse(Response):
media_type = "image/jpeg"
@app.get("/api/v2/streams/snapshot")
@app.head("/api/v2/streams/snapshot")
async def thumbnail() -> JPEGResponse:
"""A low-resolution snapshot, for compatibility with OF connect"""
blob = await thing_server.things["/camera/"].lores_mjpeg_stream.grab_frame()
return JPEGResponse(blob)
@app.get("/api/v2/instrument/settings/name")
def get_hostname() -> str:
"""Get the hostname of the device, for compatibility with OF connect"""
return gethostname()

View file

@ -0,0 +1,35 @@
from fastapi.responses import FileResponse, RedirectResponse
from fastapi.staticfiles import StaticFiles
from fastapi import FastAPI
import os
def add_static_file(app: FastAPI, fname: str, folder: str):
print(f"Adding route for /{fname}")
p=os.path.join(folder, fname)
app.get(
f"/{fname}",
response_class=FileResponse,
include_in_schema=False
)(lambda: FileResponse(p))
def add_static_files(app: FastAPI):
#with importlib.resources.as_file(openflexure_microscope_server) as p:
# static_path = p.join("/static/")
#TODO: don't hard code this!
static_path = "/var/openflexure/application/openflexure-microscope-server/src/openflexure_microscope_server/static"
if not os.path.isdir(static_path):
raise RuntimeError("Can't find static files :(")
@app.get("/", response_class=RedirectResponse)
async def redirect_fastapi():
return "/index.html"
for fname in os.listdir(static_path):
fpath = os.path.join(static_path, fname)
if os.path.isfile(fpath):
add_static_file(app, fname, static_path)
elif os.path.isdir(fpath):
app.mount(
f"/{fname}/",
StaticFiles(directory=fpath),
name=f"static_{fname}",
)