36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import logging
|
|
import time
|
|
|
|
from labthings import fields
|
|
from labthings.extensions import BaseExtension
|
|
from labthings.views import ActionView
|
|
|
|
|
|
class DevToolsExtension(BaseExtension):
|
|
def __init__(self) -> None:
|
|
super().__init__(
|
|
"org.openflexure.dev.tools",
|
|
version="0.1.0",
|
|
description="Actions to cause various traumatic events in the microscope, used for testing.",
|
|
)
|
|
self.add_view(RaiseException, "/raise")
|
|
self.add_view(SleepFor, "/sleep")
|
|
|
|
|
|
class RaiseException(ActionView):
|
|
def post(self):
|
|
raise Exception("The developer raised an exception")
|
|
|
|
|
|
class SleepFor(ActionView):
|
|
schema = {"TimeAsleep": fields.Float()}
|
|
args = {"time": fields.Float(description="Time to sleep, in seconds", example=0.5)}
|
|
|
|
def post(self, args):
|
|
sleep_time: int = args.get("time", 0)
|
|
logging.info("Going to sleep for %s...", sleep_time)
|
|
start = time.time()
|
|
time.sleep(sleep_time)
|
|
end = time.time()
|
|
logging.info("Waking up!")
|
|
return {"TimeAsleep": (end - start)}
|