import logging
from pywps import FORMATS, ComplexOutput, LiteralInput, Process
from pywps.app.Common import Metadata
from pywps.app.exceptions import ProcessError
from clisops.parameter import time_parameter
from ..dashboard import Dashboard
from ..usage import Combine
LOGGER = logging.getLogger()
[docs]
class DashboardProcess(Process):
def __init__(self):
inputs = [
LiteralInput(
"site",
"Compute Site",
abstract="The name of the compute site" "Example: cds",
data_type="string",
min_occurs=1,
max_occurs=1,
default="local",
allowed_values=["local", "ipsl", "dkrz", "all"],
),
LiteralInput(
"time",
"Time Period",
abstract="The time period for usage collection separated by /"
"Example: 2024-06-01/2024-06-30",
data_type="string",
min_occurs=0,
max_occurs=1,
),
]
outputs = [
ComplexOutput(
"dashboard",
"Dashboard",
abstract="Dashboard of OGC:WPS metrics.",
as_reference=True,
supported_formats=[FORMATS.TEXT],
),
]
super().__init__(
self._handler,
identifier="dashboard",
title="Dashboard",
abstract="Run dashboard.",
metadata=[
Metadata("ROOK", "https://github.com/roocs/rook"),
],
version="0.3",
inputs=inputs,
outputs=outputs,
store_supported=True,
status_supported=True,
)
def _handler(self, request, response):
response.update_status("Dashboard started.", 0)
if "time" in request.inputs:
time = request.inputs["time"][0].data
time_start, time_end = time_parameter.TimeParameter(time).get_bounds()
else:
time = None
time_start = time_end = None
try:
site = request.inputs["site"][0].data
usage = Combine(site=site)
fusage, fdownloads = usage.collect(
time_start=time_start, time_end=time_end, outdir=self.workdir
)
response.update_status("Combine completed.", 90)
dashboard = Dashboard(output_dir=self.workdir)
dashboard.load(url=fusage, filter="orchestrate")
dashboard.load_downloads(url=fdownloads)
response.outputs["dashboard"].file = dashboard.write()
response.update_status("Dashboard completed.", 99)
except Exception as e:
raise ProcessError(f"{e}")
return response
