From 1ca6815fe0e20acd60466f5ddd89f629ae3f710b Mon Sep 17 00:00:00 2001 From: awarde96 Date: Mon, 22 Sep 2025 12:09:57 +0000 Subject: [PATCH] Add costing from polytope-mars for feature extraction requests at server level so request can be denied before going to polytope-mars --- polytope_server/common/datasource/polytope.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/polytope_server/common/datasource/polytope.py b/polytope_server/common/datasource/polytope.py index 07d9ed59..e857f972 100644 --- a/polytope_server/common/datasource/polytope.py +++ b/polytope_server/common/datasource/polytope.py @@ -26,6 +26,7 @@ import yaml from polytope_feature.utility.exceptions import PolytopeError from polytope_mars.api import PolytopeMars +from polytope_mars.utils.areas import request_cost from ..request import Request from ..schedule import SCHEDULE_READER @@ -90,6 +91,9 @@ def check_extra_roles(self, request: Request) -> bool: def retrieve(self, request): r = copy.deepcopy(request.coerced_request) + self.request_cost = request_cost(r) + logging.info(f"Request cost: {self.request_cost}") + # Check data released if SCHEDULE_READER is not None and self.obey_schedule: SCHEDULE_READER.check_released_polytope_request(r)