implement rut part of checkout/release license

Author: matusdrobuliak66

packages/models-library/src/models_library/resource_tracker_licensed_items_usages.py

@@ -1,17 +1,4 @@
 from typing import TypeAlias
 from uuid import UUID
 
-from models_library.api_schemas_resource_usage_tracker.licensed_items_usages import (
-    LicenseCheckoutID,
-)
-from models_library.products import ProductName
-from pydantic import BaseModel, ConfigDict
-
 LicensedItemUsageID: TypeAlias = UUID
-
-
-class LicenseCheckoutCreate(BaseModel):
-    checkout_id: LicenseCheckoutID
-    product_name: ProductName
-
-    model_config = ConfigDict(from_attributes=True)

services/resource-usage-tracker/src/simcore_service_resource_usage_tracker/api/rpc/_licensed_items_usages.py

@@ -0,0 +1,49 @@
+from fastapi import FastAPI
+from models_library.api_schemas_resource_usage_tracker.licensed_items_usages import (
+    LicensedItemUsageGet,
+    LicenseItemCheckoutGet,
+)
+from models_library.licensed_items import LicensedItemID
+from models_library.products import ProductName
+from models_library.resource_tracker import ServiceRunId
+from models_library.resource_tracker_licensed_items_usages import LicenseCheckoutID
+from models_library.users import UserID
+from models_library.wallets import WalletID
+from servicelib.rabbitmq import RPCRouter
+
+from ...services import licensed_items_usages
+
+router = RPCRouter()
+
+
+@router.expose(reraise_if_error_type=())
+async def checkout_licensed_item(
+    app: FastAPI,
+    *,
+    licensed_item_id: LicensedItemID,
+    wallet_id: WalletID,
+    product_name: ProductName,
+    num_of_seats: int,
+    service_run_id: ServiceRunId,
+    user_id: UserID,
+    user_email: str,
+) -> LicenseItemCheckoutGet:
+    return await licensed_items_usages.checkout_licensed_item(
+        db_engine=app.state.engine,
+        licensed_item_id=licensed_item_id,
+        wallet_id=wallet_id,
+        product_name=product_name,
+        num_of_seats=num_of_seats,
+        service_run_id=service_run_id,
+        user_id=user_id,
+        user_email=user_email,
+    )
+
+
+@router.expose(reraise_if_error_type=())
+async def release_licensed_item(
+    app: FastAPI, *, checkout_id: LicenseCheckoutID, product_name: ProductName
+) -> LicensedItemUsageGet:
+    return await licensed_items_usages.release_licensed_item(
+        db_engine=app.state.engine, checkout_id=checkout_id, product_name=product_name
+    )

services/resource-usage-tracker/src/simcore_service_resource_usage_tracker/services/licensed_items_usages.py

@@ -3,17 +3,15 @@
 
 from fastapi import Depends
 from models_library.api_schemas_resource_usage_tracker.licensed_items_usages import (
+    LicenseCheckoutID,
     LicensedItemsUsagesPage,
     LicensedItemUsageGet,
     LicenseItemCheckoutGet,
 )
 from models_library.licensed_items import LicensedItemID
 from models_library.products import ProductName
 from models_library.resource_tracker import ServiceRunId, ServiceRunStatus
-from models_library.resource_tracker_licensed_items_usages import (
-    LicenseCheckoutCreate,
-    LicensedItemUsageID,
-)
+from models_library.resource_tracker_licensed_items_usages import LicensedItemUsageID
 from models_library.rest_ordering import OrderBy
 from models_library.users import UserID
 from models_library.wallets import WalletID
@@ -160,13 +158,14 @@ async def checkout_licensed_item(
 async def release_licensed_item(
     db_engine: Annotated[AsyncEngine, Depends(get_resource_tracker_db_engine)],
     *,
-    data: LicenseCheckoutCreate,
+    checkout_id: LicenseCheckoutID,
+    product_name: ProductName,
 ) -> LicensedItemUsageGet:
 
     licensed_item_usage_db: LicensedItemUsageDB = await licensed_items_usages_db.update(
         db_engine,
-        licensed_item_usage_id=data.checkout_id,
-        product_name=data.product_name,
+        licensed_item_usage_id=checkout_id,
+        product_name=product_name,
         stopped_at=datetime.now(tz=UTC),
     )