fractal-server 1.4.0a7__py3-none-any.whl → 1.4.0a9__py3-none-any.whl

fractal_server/__init__.py

@@ -1 +1 @@
-__VERSION__ = "1.4.0a7"
+__VERSION__ = "1.4.0a9"

fractal_server/app/db/__init__.py

@@ -63,7 +63,9 @@ class DB:
                 connect_args={"check_same_thread": False},
             )
         else:
-            engine_kwargs_async = {}
+            engine_kwargs_async = {
+                "pool_pre_ping": True,
+            }
             engine_kwargs_sync = {}
 
         cls._engine_async = create_async_engine(
@@ -80,11 +82,17 @@ class DB:
         )
 
         cls._async_session_maker = sessionmaker(
-            cls._engine_async, class_=AsyncSession, expire_on_commit=False
+            cls._engine_async,
+            class_=AsyncSession,
+            expire_on_commit=False,
+            future=True,
         )
 
         cls._sync_session_maker = sessionmaker(
-            bind=cls._engine_sync, autocommit=False, autoflush=False
+            bind=cls._engine_sync,
+            autocommit=False,
+            autoflush=False,
+            future=True,
         )
 
         @event.listens_for(cls._engine_sync, "connect")

fractal_server/app/routes/api/v1/job.py

@@ -15,8 +15,12 @@ from .....config import get_settings
 from .....syringe import Inject
 from ....db import AsyncSession
 from ....db import get_db
+from ....models import ApplyWorkflow
 from ....runner._common import SHUTDOWN_FILENAME
 from ....schemas import ApplyWorkflowRead
+from ....schemas import ApplyWorkflowUpdate
+from ....schemas import JobStatusType
+from ....security import current_active_superuser
 from ....security import current_active_user
 from ....security import User
 from ._aux_functions import _get_job_check_owner
@@ -196,3 +200,39 @@ async def stop_job(
         f.write(f"Trigger executor shutdown for {job.id=}, {project_id=}.")
 
     return Response(status_code=status.HTTP_204_NO_CONTENT)
+
+
+@router.patch(
+    "/job/{job_id}/",
+    response_model=ApplyWorkflowRead,
+)
+async def update_job(
+    job_update: ApplyWorkflowUpdate,
+    job_id: int,
+    user: User = Depends(current_active_superuser),
+    db: AsyncSession = Depends(get_db),
+) -> Optional[ApplyWorkflowRead]:
+    """
+    Change the status of an existing job.
+
+    This endpoint is only open to superusers, and it does not apply
+    project-based access-control to jobs.
+    """
+    job = await db.get(ApplyWorkflow, job_id)
+    if job is None:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail=f"Job {job_id} not found",
+        )
+
+    if job_update.status != JobStatusType.FAILED:
+        raise HTTPException(
+            status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
+            detail=f"Cannot set job status to {job_update.status}",
+        )
+
+    setattr(job, "status", job_update.status)
+    await db.commit()
+    await db.refresh(job)
+    await db.close()
+    return job

fractal_server/app/routes/auth.py

@@ -3,6 +3,11 @@ Definition of `/auth` routes.
 """
 from fastapi import APIRouter
 from fastapi import Depends
+from fastapi import HTTPException
+from fastapi import status
+from fastapi_users import exceptions
+from fastapi_users import schemas
+from fastapi_users.router.common import ErrorCode
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlmodel import select
 
@@ -13,12 +18,14 @@ from ..models.security import UserOAuth as User
 from ..schemas.user import UserCreate
 from ..schemas.user import UserRead
 from ..schemas.user import UserUpdate
+from ..schemas.user import UserUpdateStrict
 from ..security import cookie_backend
 from ..security import current_active_superuser
 from ..security import current_active_user
 from ..security import fastapi_users
+from ..security import get_user_manager
 from ..security import token_backend
-
+from ..security import UserManager
 
 router_auth = APIRouter()
 
@@ -33,18 +40,22 @@ router_auth.include_router(
     fastapi_users.get_register_router(UserRead, UserCreate),
     dependencies=[Depends(current_active_superuser)],
 )
-router_auth.include_router(
-    fastapi_users.get_reset_password_router(),
-)
-router_auth.include_router(
-    fastapi_users.get_verify_router(UserRead),
-)
 
-# Include users routes, after removing DELETE endpoint (ref
-# https://github.com/fastapi-users/fastapi-users/discussions/606)
 users_router = fastapi_users.get_users_router(UserRead, UserUpdate)
+
+# We remove `/auth/users/me` endpoints to implement our own
+# at `/auth/current-user/`.
+# We also remove `DELETE /auth/users/{user_id}`
+# (ref https://github.com/fastapi-users/fastapi-users/discussions/606)
 users_router.routes = [
-    route for route in users_router.routes if route.name != "users:delete_user"
+    route
+    for route in users_router.routes
+    if route.name
+    not in [
+        "users:current_user",
+        "users:delete_user",
+        "users:patch_current_user",
+    ]
 ]
 router_auth.include_router(
     users_router,
@@ -53,17 +64,37 @@ router_auth.include_router(
 )
 
 
-@router_auth.get("/whoami", response_model=UserRead)
-async def whoami(
-    user: User = Depends(current_active_user),
+@router_auth.patch("/current-user/", response_model=UserRead)
+async def patch_current_user(
+    user_update: UserUpdateStrict,
+    current_user: User = Depends(current_active_user),
+    user_manager: UserManager = Depends(get_user_manager),
 ):
+
+    update = UserUpdate(**user_update.dict(exclude_unset=True))
+
+    try:
+        user = await user_manager.update(update, current_user, safe=True)
+    except exceptions.InvalidPasswordException as e:
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail={
+                "code": ErrorCode.UPDATE_USER_INVALID_PASSWORD,
+                "reason": e.reason,
+            },
+        )
+    return schemas.model_validate(User, user)
+
+
+@router_auth.get("/current-user/", response_model=UserRead)
+async def get_current_user(user: User = Depends(current_active_user)):
     """
     Return current user
     """
     return user
 
 
-@router_auth.get("/userlist", response_model=list[UserRead])
+@router_auth.get("/users/", response_model=list[UserRead])
 async def list_users(
     user: User = Depends(current_active_superuser),
     db: AsyncSession = Depends(get_db),

fractal_server/app/schemas/__init__.py

@@ -3,6 +3,7 @@ Schemas for API request/response bodies
 """
 from .applyworkflow import ApplyWorkflowCreate  # noqa: F401
 from .applyworkflow import ApplyWorkflowRead  # noqa: F401
+from .applyworkflow import ApplyWorkflowUpdate  # noqa: F401
 from .applyworkflow import JobStatusType  # noqa: F401
 from .dataset import DatasetCreate  # noqa: F401
 from .dataset import DatasetRead  # noqa: F401

fractal_server/app/schemas/applyworkflow.py

@@ -146,3 +146,14 @@ class ApplyWorkflowRead(_ApplyWorkflowBase):
         d["start_timestamp"] = str(self.start_timestamp)
         d["end_timestamp"] = str(self.end_timestamp)
         return d
+
+
+class ApplyWorkflowUpdate(BaseModel):
+    """
+    Class for updating a job status.
+
+    Attributes:
+        status: New job status.
+    """
+
+    status: JobStatusType

fractal_server/app/schemas/user.py

@@ -1,6 +1,8 @@
 from typing import Optional
 
 from fastapi_users import schemas
+from pydantic import BaseModel
+from pydantic import Extra
 from pydantic import validator
 
 from ._validators import val_absolute_path
@@ -52,6 +54,31 @@ class UserUpdate(schemas.BaseUserUpdate):
         val_absolute_path("cache_dir")
     )
 
+    @validator(
+        "is_active",
+        "is_verified",
+        "is_superuser",
+        "email",
+        "password",
+        always=False,
+    )
+    def cant_set_none(cls, v, field):
+        if v is None:
+            raise ValueError(f"Cannot set {field.name}=None")
+        return v
+
+
+class UserUpdateStrict(BaseModel, extra=Extra.forbid):
+    """
+    Attributes that every user can self-edit
+    """
+
+    cache_dir: Optional[str]
+
+    _cache_dir = validator("cache_dir", allow_reuse=True)(
+        val_absolute_path("cache_dir")
+    )
+
 
 class UserCreate(schemas.BaseUserCreate):
     """

fractal_server/app/security/__init__.py

@@ -35,8 +35,6 @@ from typing import Type
 
 from fastapi import Depends
 from fastapi import HTTPException
-from fastapi import Request
-from fastapi import Response
 from fastapi import status
 from fastapi_users import BaseUserManager
 from fastapi_users import FastAPIUsers
@@ -46,6 +44,7 @@ from fastapi_users.authentication import BearerTransport
 from fastapi_users.authentication import CookieTransport
 from fastapi_users.authentication import JWTStrategy
 from fastapi_users.db.base import BaseUserDatabase
+from fastapi_users.exceptions import InvalidPasswordException
 from fastapi_users.models import ID
 from fastapi_users.models import OAP
 from fastapi_users.models import UP
@@ -176,17 +175,17 @@ async def get_user_db(
 
 
 class UserManager(IntegerIDMixin, BaseUserManager[User, int]):
-    async def on_after_login(
-        self,
-        user: User,
-        request: Optional[Request] = None,
-        response: Optional[Response] = None,
-    ) -> None:
-        """
-        Perform logic after user login.
-        *You should overload this method to add your own logic.*
-        """
-        pass
+    async def validate_password(self, password: str, user: User) -> None:
+        # check password length
+        min_length, max_length = 4, 100
+        if len(password) < min_length:
+            raise InvalidPasswordException(
+                f"The password is too short (minimum length: {min_length})."
+            )
+        elif len(password) > max_length:
+            raise InvalidPasswordException(
+                f"The password is too long (maximum length: {min_length})."
+            )
 
 
 async def get_user_manager(

{fractal_server-1.4.0a7.dist-info → fractal_server-1.4.0a9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fractal-server
-Version: 1.4.0a7
+Version: 1.4.0a9
 Summary: Server component of the Fractal analytics platform
 Home-page: https://github.com/fractal-analytics-platform/fractal-server
 License: BSD-3-Clause
@@ -12,12 +12,13 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Provides-Extra: gunicorn
 Provides-Extra: postgres
 Provides-Extra: slurm
 Requires-Dist: aiosqlite (>=0.17.0,<0.18.0)
 Requires-Dist: alembic (>=1.9.1,<2.0.0)
-Requires-Dist: asyncpg (>=0.27.0,<0.28.0) ; extra == "postgres"
+Requires-Dist: asyncpg (>=0.29.0,<0.30.0) ; extra == "postgres"
 Requires-Dist: cloudpickle (>=2.2.1,<2.3.0) ; extra == "slurm"
 Requires-Dist: clusterfutures (>=0.5,<0.6) ; extra == "slurm"
 Requires-Dist: fastapi (>=0.103.0,<0.104.0)
@@ -26,8 +27,8 @@ Requires-Dist: gunicorn (>=20.1.0,<21.0.0) ; extra == "gunicorn"
 Requires-Dist: psycopg2 (>=2.9.5,<3.0.0) ; extra == "postgres"
 Requires-Dist: pydantic (>=1.10.8,<2)
 Requires-Dist: python-dotenv (>=0.20.0,<0.21.0)
-Requires-Dist: sqlalchemy[asyncio] (>=1.4,<2.0)
-Requires-Dist: sqlmodel (>=0.0.11,<0.0.12)
+Requires-Dist: sqlalchemy[asyncio] (>=2.0.23,<2.1)
+Requires-Dist: sqlmodel (>=0.0.12,<0.0.13)
 Requires-Dist: uvicorn (>=0.20.0,<0.21.0)
 Project-URL: Changelog, https://github.com/fractal-analytics-platform/fractal-server/blob/main/CHANGELOG.md
 Project-URL: Documentation, https://fractal-analytics-platform.github.io/fractal-server

{fractal_server-1.4.0a7.dist-info → fractal_server-1.4.0a9.dist-info}/RECORD

@@ -1,8 +1,8 @@
-fractal_server/__init__.py,sha256=N2MagRx5dIspHCzA8pKzYWpIRqjQogIOXNJvb_S8HJ8,24
+fractal_server/__init__.py,sha256=IneygUCsbuvssacaRSvCjbRcp6R_sPRDaCi7xtY7xSY,24
 fractal_server/__main__.py,sha256=znijcImbcEC4P26ICOhEJ9VY3_5vWdMwQcl-WP25sYA,2202
 fractal_server/alembic.ini,sha256=MWwi7GzjzawI9cCAK1LW7NxIBQDUqD12-ptJoq5JpP0,3153
 fractal_server/app/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-fractal_server/app/db/__init__.py,sha256=KWmQiO9CZ3f16RpKcnLa1BvpTwFv7WUpeMn6LmAe6KA,3629
+fractal_server/app/db/__init__.py,sha256=gJCmPYNVg8UEjvtPWLupM3yzZRpt3wvHjZuUsn1DBXk,3781
 fractal_server/app/models/__init__.py,sha256=RuxWH8fsmkTWsjLhYjrxSt-mvk74coCilAQlX2Q6OO0,353
 fractal_server/app/models/dataset.py,sha256=fcZkb2y7PXlFJAyZtndJ7Gf4c8VpkWjZe47iMybv1aE,2109
 fractal_server/app/models/job.py,sha256=DbX7OMx88eC-232C_OdYOpNeyN0tma7p8J3x7HB43os,2768
@@ -18,13 +18,13 @@ fractal_server/app/routes/api/__init__.py,sha256=EVyZrEq3I_1643QGTPCC5lgCp4xH_au
 fractal_server/app/routes/api/v1/__init__.py,sha256=V4nhYyMIqhlJxbotLTYikq_ghb6KID0ZKOOYaOq7C-g,944
 fractal_server/app/routes/api/v1/_aux_functions.py,sha256=qaeTsfSk78FHSsMVNkGdPIPBjeKOEd1sq0KhY0SQ1l0,10095
 fractal_server/app/routes/api/v1/dataset.py,sha256=_P8SUXEs5Gh-a82QR06xyZ_F6eX6R_CxmBduNXEeERY,14968
-fractal_server/app/routes/api/v1/job.py,sha256=07nrsWvPcV-DkFybiMKgHLVDp_CFeFx6pnEKnBNrvL4,5382
+fractal_server/app/routes/api/v1/job.py,sha256=9jEoIYnUvtc3oB52yRhFd_NYJk8tGK3qorDOjcgiN1I,6554
 fractal_server/app/routes/api/v1/project.py,sha256=uP-p9rghQBQ_vRZ2hUw_bT21WVLpqdgJNdxN6z367YI,11417
 fractal_server/app/routes/api/v1/task.py,sha256=iT45B_NXvBilp-aclWuONsNAWvpczrEWO21K9DOJsZ0,5504
 fractal_server/app/routes/api/v1/task_collection.py,sha256=bMnhrMkI_45Bttd6J9j_CEmpjQZCV5mTy8_j1Ca8xQg,11724
 fractal_server/app/routes/api/v1/workflow.py,sha256=rXU-wyU8SFJ7t6tNl2dCiugXzMjFpYV_ljTcYQrC8NE,9819
 fractal_server/app/routes/api/v1/workflowtask.py,sha256=ZXmZBymcQgt838Bywoxcy6M0l1z2uatHJxq_e45-8T0,5432
-fractal_server/app/routes/auth.py,sha256=XEerBzlHE7b4XfB6Bk0qYj6edRM6yFZr3UosLzjfvt8,3816
+fractal_server/app/routes/auth.py,sha256=aBR0ZICrbJpOgPO972HN5lJ1YqfkrO0avwdrSRvI9aQ,4873
 fractal_server/app/runner/.gitignore,sha256=ytzN_oyHWXrGU7iFAtoHSTUbM6Rn6kG0Zkddg0xZk6s,16
 fractal_server/app/runner/__init__.py,sha256=WNMxT9XFEocPOdJRN6FfMek-LecOpBYXfJxmxupMRUE,13597
 fractal_server/app/runner/_common.py,sha256=p8T-fUD357vaBXPwNPZOHptRRT3c3scnhzgHYoxFNro,22649
@@ -43,9 +43,9 @@ fractal_server/app/runner/_slurm/executor.py,sha256=ao5YuWtjsIfTYUucE1SvNS8a99Sg
 fractal_server/app/runner/_slurm/remote.py,sha256=wLziIsGdSMiO-jIXM8x77JRK82g_2hx0iBKTiMghuIo,5852
 fractal_server/app/runner/common.py,sha256=nz0ZuIro0iwZm-OV-e-Y-PrtgKcLK0d7BrzebWyEWEk,9496
 fractal_server/app/runner/handle_failed_job.py,sha256=PKgJARHjXyv33sDsl7oTINdcTu7EwmFmIkp38RqAE3Q,4641
-fractal_server/app/schemas/__init__.py,sha256=th51Dzbe94xatOIMVheqlq0rVFy8oI1CHRfzCbjiTSg,1859
+fractal_server/app/schemas/__init__.py,sha256=UciI1WXCtZbGj9rSBFN388_oaJACzNY0yJulxbMR17w,1920
 fractal_server/app/schemas/_validators.py,sha256=7YEbgrnGRpzkLMfZzQNfczEmcNnO__SmVOaBHhzaiXE,1819
-fractal_server/app/schemas/applyworkflow.py,sha256=zzQy_vxWjzUE0WnZHQWQBU7GB0a8ZkBczZidl32WOY0,3911
+fractal_server/app/schemas/applyworkflow.py,sha256=8GLvfhfsH3CbFShkiihqYhQ89gA0vbvx-DT9zGHWGRo,4080
 fractal_server/app/schemas/dataset.py,sha256=PPqGTsRQ5JEwkiM4NcjPYFckxnCdi_Zov-bWXDm1LUk,3092
 fractal_server/app/schemas/json_schemas/manifest.json,sha256=yXYKHbYXPYSkSXMTLfTpfCUGBtmQuPTk1xuSXscdba4,1787
 fractal_server/app/schemas/manifest.py,sha256=xxTd39dAXMK9Ox1y-p3gbyg0zd5udW99pV4JngCUGwM,3819
@@ -53,9 +53,9 @@ fractal_server/app/schemas/project.py,sha256=qfVqDeQPIyb1QiT6oUJNcF6gXcSfjsj3BJE
 fractal_server/app/schemas/state.py,sha256=CS8Rs5qF21TsnqmyzUHLqRaX1b61Oc6Yra6POYpYSQY,762
 fractal_server/app/schemas/task.py,sha256=2TBE5Ne9tO_-a2-Es0PRXMT8ZddSInTOPMor7u8-gx0,3671
 fractal_server/app/schemas/task_collection.py,sha256=mPk6E1LK2UvnHkhIQWHmTztsVT99iHZn-UZy7mGNjUk,2965
-fractal_server/app/schemas/user.py,sha256=zhB-2WfJ30hNcHaW2V126v5i7rHl66fX_SRmIWCQrjM,1587
+fractal_server/app/schemas/user.py,sha256=vxoxojcnCngjGlAzNtd1rp4zw_mWsBije0AJR2le1Ik,2187
 fractal_server/app/schemas/workflow.py,sha256=oFoO62JH5hfMJjKoicdpyC5hd2O9XgqoAm5RN9YjXAI,4238
-fractal_server/app/security/__init__.py,sha256=dnLkDpzduB9pS5dNeUZhEb0z6ZDb7dPreE57uCtxOBE,7998
+fractal_server/app/security/__init__.py,sha256=_h50FYHMBn70_F2qV54rewairdllmwEK_Z9jU6e7fWM,8189
 fractal_server/config.py,sha256=zekTDA_FhQG_RYgOWEIxT6KyJKRpMge-pB-iYRr4sIY,14447
 fractal_server/logger.py,sha256=keri8i960WHT8Zz9Rm2MwfnrA2dw9TsrfCmojqtGDLs,4562
 fractal_server/main.py,sha256=N82d-bq0TxOItUBwo5MZO693oL9lklu5-chk2McB8Eo,6662
@@ -76,8 +76,8 @@ fractal_server/syringe.py,sha256=3qSMW3YaMKKnLdgnooAINOPxnCOxP7y2jeAQYB21Gdo,278
 fractal_server/tasks/__init__.py,sha256=Wzuxf5EoH1v0fYzRpAZHG_S-Z9f6DmbIsuSvllBCGvc,72
 fractal_server/tasks/collection.py,sha256=POKvQyS5G5ySybH0r0v21I_ZQ5AREe9kAqr_uFfGyaU,17627
 fractal_server/utils.py,sha256=b7WwFdcFZ8unyT65mloFToYuEDXpQoHRcmRNqrhd_dQ,2115
-fractal_server-1.4.0a7.dist-info/LICENSE,sha256=QKAharUuhxL58kSoLizKJeZE3mTCBnX6ucmz8W0lxlk,1576
-fractal_server-1.4.0a7.dist-info/METADATA,sha256=TSSUg_-pl8slsQ6suxXi5J2aR9uiThGFOQX7Jiad4WA,3773
-fractal_server-1.4.0a7.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-fractal_server-1.4.0a7.dist-info/entry_points.txt,sha256=8tV2kynvFkjnhbtDnxAqImL6HMVKsopgGfew0DOp5UY,58
-fractal_server-1.4.0a7.dist-info/RECORD,,
+fractal_server-1.4.0a9.dist-info/LICENSE,sha256=QKAharUuhxL58kSoLizKJeZE3mTCBnX6ucmz8W0lxlk,1576
+fractal_server-1.4.0a9.dist-info/METADATA,sha256=ES4pTFqvvpmuYSrpQlW0YyOIFN2M2RvoxAQYAJAwTvc,3827
+fractal_server-1.4.0a9.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+fractal_server-1.4.0a9.dist-info/entry_points.txt,sha256=8tV2kynvFkjnhbtDnxAqImL6HMVKsopgGfew0DOp5UY,58
+fractal_server-1.4.0a9.dist-info/RECORD,,

{fractal_server-1.4.0a7.dist-info → fractal_server-1.4.0a9.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.6.1
+Generator: poetry-core 1.8.1
 Root-Is-Purelib: true
 Tag: py3-none-any