product repo

Author: pcrespov

services/web/server/src/simcore_service_webserver/products/_api.py

@@ -1,6 +1,7 @@
 from decimal import Decimal
+from itertools import product
 from pathlib import Path
-from typing import cast
+from typing import Any, cast
 
 import aiofiles
 from aiohttp import web
@@ -12,7 +13,11 @@
 from ._db import ProductRepository
 from ._events import APP_PRODUCTS_TEMPLATES_DIR_KEY
 from ._model import Product
-from .errors import BelowMinimumPaymentError, ProductPriceNotDefinedError
+from .errors import (
+    BelowMinimumPaymentError,
+    ProductNotFoundError,
+    ProductPriceNotDefinedError,
+)
 
 
 def get_product_name(request: web.Request) -> str:
@@ -56,6 +61,16 @@ async def get_current_product_credit_price_info(
     )
 
 
+async def get_product_ui(
+    repo: ProductRepository, product_name: ProductName
+) -> dict[str, Any]:
+    ui = await repo.get_product_ui(product_name=product_name)
+    if ui is not None:
+        return ui
+
+    raise ProductNotFoundError(product_name=product_name)
+
+
 async def get_credit_amount(
     app: web.Application,
     *,

services/web/server/src/simcore_service_webserver/products/_db.py

@@ -1,6 +1,6 @@
 import logging
 from decimal import Decimal
-from typing import AsyncIterator, NamedTuple
+from typing import Any, AsyncIterator, NamedTuple
 
 import sqlalchemy as sa
 from aiopg.sa.connection import SAConnection
@@ -151,3 +151,11 @@ async def get_product_template_content(
                 .where(products.c.name == product_name)
             )
             return f"{content}" if content else None
+
+    async def get_product_ui(self, product_name: ProductName) -> dict[str, Any] | None:
+        async with self.engine.acquire() as conn:
+            result = await conn.execute(
+                sa.select(products.c.ui).where(products.c.name == product_name)
+            )
+            row: RowProxy | None = await result.first()
+            return dict(**row.ui) if row else None

services/web/server/src/simcore_service_webserver/products/_handlers.py

@@ -13,6 +13,7 @@
 from pydantic import Field
 from servicelib.aiohttp.requests_validation import parse_request_path_parameters_as
 from servicelib.request_keys import RQT_USERID_KEY
+from simcore_service_webserver.products._db import ProductRepository
 
 from .._constants import RQ_PRODUCT_KEY
 from .._meta import API_VTAG as VTAG
@@ -86,7 +87,11 @@ async def _get_current_product_ui(request: web.Request):
     req_ctx = _ProductsRequestContext.model_validate(request)
     product_name = req_ctx.product_name
 
-    data = ProductUIGet(product_name=product_name, ui={})
+    ui = await api.get_product_ui(
+        ProductRepository.create_from_request(request), product_name=product_name
+    )
+
+    data = ProductUIGet(product_name=product_name, ui=ui)
     return envelope_json_response(data)
 
 

services/web/server/src/simcore_service_webserver/products/api.py

@@ -7,6 +7,7 @@
     get_product_name,
     get_product_stripe_info,
     get_product_template_path,
+    get_product_ui,
     list_products,
 )
 from ._model import Product
@@ -17,6 +18,7 @@
     "get_product_name",
     "get_product_stripe_info",
     "get_product_template_path",
+    "get_product_ui",
     "get_product",
     "list_products",
     "Product",

services/web/server/src/simcore_service_webserver/products/errors.py

@@ -10,6 +10,10 @@ class ProductError(WebServerBaseError, ValueError):
     ...
 
 
+class ProductNotFoundError(ProductError):
+    msg_template = "Undefined product '{product_name}'"
+
+
 class ProductPriceNotDefinedError(ProductError):
     msg_template = "Product price not defined. {reason}"