draft query

Author: pcrespov

services/web/server/src/simcore_service_webserver/projects/_jobs_repository.py

@@ -1,16 +1,39 @@
 import logging
+from typing import cast
 
+import sqlalchemy as sa
+from models_library.products import ProductName
 from models_library.projects import ProjectID
+from models_library.users import UserID
+from pydantic import TypeAdapter
+from simcore_postgres_database.models.groups import user_to_groups
+from simcore_postgres_database.models.project_to_groups import project_to_groups
+from simcore_postgres_database.models.projects import projects
 from simcore_postgres_database.models.projects_to_jobs import projects_to_jobs
-from simcore_postgres_database.utils_repos import transaction_context
+from simcore_postgres_database.models.projects_to_products import projects_to_products
+from simcore_postgres_database.utils_repos import (
+    get_columns_from_db_model,
+    pass_or_acquire_connection,
+    transaction_context,
+)
 from sqlalchemy.dialects.postgresql import insert as pg_insert
 from sqlalchemy.ext.asyncio import AsyncConnection
 
 from ..db.base_repository import BaseRepository
+from .models import ProjectDBGet, ProjectJobDBGet
 
 _logger = logging.getLogger(__name__)
 
 
+_PROJECT_JOB_DB_COLS = [
+    *get_columns_from_db_model(
+        projects,
+        ProjectDBGet,
+    ),
+    projects_to_jobs.c.job_parent_resource_name,  # Add job_parent_resource_name
+]
+
+
 class ProjectJobsRepository(BaseRepository):
 
     async def set_project_as_job(
@@ -34,3 +57,87 @@ async def set_project_as_job(
             )
 
             await conn.execute(stmt)
+
+    async def list_projects_marked_as_jobs(
+        self,
+        connection: AsyncConnection | None = None,
+        *,
+        product_name: ProductName,
+        user_id: UserID,
+        offset: int = 0,
+        limit: int = 10,
+        job_parent_resource_name_filter: str | None = None,
+    ) -> tuple[int, list[ProjectJobDBGet]]:
+        """
+        Lists projects marked as jobs for a specific user and product
+        """
+
+        # Step 1: Get group IDs associated with the user
+        user_groups_query = (
+            sa.select(user_to_groups.c.gid)
+            .where(user_to_groups.c.uid == user_id)
+            .subquery()
+        )
+
+        # Step 2: Create access_query to filter projects based on product_name and read access
+        access_query = (
+            sa.select(projects_to_jobs.c.project_uuid)
+            .select_from(
+                projects_to_jobs.join(
+                    projects_to_products,
+                    projects_to_jobs.c.project_uuid
+                    == projects_to_products.c.project_uuid,
+                ).join(
+                    project_to_groups,
+                    projects_to_jobs.c.project_uuid == project_to_groups.c.project_uuid,
+                )
+            )
+            .where(
+                projects_to_products.c.product_name == product_name,
+                project_to_groups.c.gid.in_(sa.select(user_groups_query.c.gid)),
+                project_to_groups.c.read.is_(True),
+            )
+        )
+
+        # Apply job_parent_resource_name_filter if provided
+        if job_parent_resource_name_filter:
+            access_query = access_query.where(
+                projects_to_jobs.c.job_parent_resource_name.like(
+                    f"%{job_parent_resource_name_filter}%"
+                )
+            )
+
+        # Convert access_query to a subquery
+        base_query = access_query.subquery()
+
+        # Step 3: Query to get the total count
+        total_query = sa.select(sa.func.count()).select_from(base_query)
+
+        # Step 4: Query to get the paginated list with full selection
+        list_query = (
+            sa.select(*_PROJECT_JOB_DB_COLS)
+            .select_from(
+                base_query.join(
+                    projects,
+                    projects.c.uuid == base_query.c.project_uuid,
+                )
+            )
+            .order_by(
+                projects.c.creation_date.desc(),  # latests first
+                projects.c.id.desc(),
+            )
+            .limit(limit)
+            .offset(offset)
+        )
+
+        # Step 5: Execute queries
+        async with pass_or_acquire_connection(self.engine, connection) as conn:
+            total_count = await conn.scalar(total_query)
+            assert isinstance(total_count, int)  # nosec
+
+            result = await conn.execute(list_query)
+            projects_list = TypeAdapter(list[ProjectJobDBGet]).validate_python(
+                result.fetchall()
+            )
+
+            return cast(int, total_count), projects_list

services/web/server/src/simcore_service_webserver/projects/_jobs_service.py

@@ -6,6 +6,7 @@
 from models_library.projects import ProjectID
 from models_library.users import UserID
 from pydantic import AfterValidator, validate_call
+from simcore_service_webserver.projects.models import ProjectDBGet
 
 from ._access_rights_service import check_user_project_permission
 from ._jobs_repository import ProjectJobsRepository
@@ -45,3 +46,27 @@ async def set_project_as_job(
     await repo.set_project_as_job(
         project_uuid=project_uuid, job_parent_resource_name=job_parent_resource_name
     )
+
+
+@validate_call(config={"arbitrary_types_allowed": True})
+async def list_my_projects_marked_as_jobs(
+    app: web.Application,
+    *,
+    product_name: ProductName,
+    user_id: UserID,
+    offset: int = 0,
+    limit: int = 10,
+    job_parent_resource_name_filter: str | None = None,
+) -> tuple[int, list[ProjectDBGet]]:
+    """
+    Lists paginated projects marked as jobs for the given user and product.
+    Optionally filters by job_parent_resource_name using SQL-like wildcard patterns.
+    """
+    repo = ProjectJobsRepository.create_from_app(app)
+    return await repo.list_projects_marked_as_jobs(
+        user_id=user_id,
+        product_name=product_name,
+        offset=offset,
+        limit=limit,
+        job_parent_resource_name_filter=job_parent_resource_name_filter,
+    )

services/web/server/src/simcore_service_webserver/projects/models.py

@@ -71,6 +71,10 @@ class ProjectDBGet(BaseModel):
     )
 
 
+class ProjectJobDBGet(ProjectDBGet):
+    job_parent_resource_name: str
+
+
 class ProjectWithTrashExtra(ProjectDBGet):
     # This field is not part of the tables
     trashed_by_primary_gid: GroupID | None = None