fractal-server 2.17.1a1__py3-none-any.whl → 2.18.0__py3-none-any.whl

fractal_server/migrations/versions/7910eed4cf97_user_project_dirs_and_usergroup_viewer_.py

@@ -0,0 +1,60 @@
+"""User project_dirs and UserGroup viewer paths
+
+Revision ID: 7910eed4cf97
+Revises: bc0e8b3327a7
+Create Date: 2025-11-27 16:02:51.824653
+
+"""
+
+import sqlalchemy as sa
+from alembic import op
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision = "7910eed4cf97"
+down_revision = "bc0e8b3327a7"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("user_oauth", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column(
+                "project_dirs",
+                postgresql.ARRAY(sa.String()),
+                server_default="{}",
+                nullable=False,
+            )
+        )
+        batch_op.alter_column(
+            "project_dir", existing_type=sa.VARCHAR(), nullable=True
+        )
+
+    with op.batch_alter_table("usergroup", schema=None) as batch_op:
+        batch_op.drop_column("viewer_paths")
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("usergroup", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column(
+                "viewer_paths",
+                postgresql.JSONB(astext_type=sa.Text()),
+                server_default=sa.text("'[]'::json"),
+                autoincrement=False,
+                nullable=False,
+            )
+        )
+
+    with op.batch_alter_table("user_oauth", schema=None) as batch_op:
+        batch_op.alter_column(
+            "project_dir", existing_type=sa.VARCHAR(), nullable=False
+        )
+        batch_op.drop_column("project_dirs")
+
+    # ### end Alembic commands ###

fractal_server/migrations/versions/791ce783d3d8_add_indices.py

@@ -5,8 +5,8 @@ Revises: 969d84257cac
 Create Date: 2025-06-03 09:32:30.757651
 
 """
-from alembic import op
 
+from alembic import op
 
 # revision identifiers, used by Alembic.
 revision = "791ce783d3d8"

fractal_server/migrations/versions/83bc2ad3ffcc_2_17_0.py

@@ -5,6 +5,7 @@ Revises: 981d588fe248
 Create Date: 2025-10-30 14:16:53.639006
 
 """
+
 import sqlalchemy as sa
 import sqlmodel
 from alembic import op

fractal_server/migrations/versions/84bf0fffde30_add_dumps_to_applyworkflow.py

@@ -5,6 +5,7 @@ Revises: 99ea79d9e5d2
 Create Date: 2023-10-26 16:11:44.061971
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 

fractal_server/migrations/versions/88270f589c9b_add_prevent_new_submissions.py

@@ -0,0 +1,39 @@
+"""add_prevent_new_submissions
+
+Revision ID: 88270f589c9b
+Revises: f0702066b007
+Create Date: 2025-12-02 12:34:11.028259
+
+"""
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "88270f589c9b"
+down_revision = "f0702066b007"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("resource", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column(
+                "prevent_new_submissions",
+                sa.BOOLEAN(),
+                server_default="false",
+                nullable=False,
+            )
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("resource", schema=None) as batch_op:
+        batch_op.drop_column("prevent_new_submissions")
+
+    # ### end Alembic commands ###

fractal_server/migrations/versions/8e8f227a3e36_update_taskv2_post_2_7_0.py

@@ -5,10 +5,10 @@ Revises: 034a469ec2eb
 Create Date: 2024-10-29 09:01:33.075251
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 
-
 # revision identifiers, used by Alembic.
 revision = "8e8f227a3e36"
 down_revision = "034a469ec2eb"
@@ -31,9 +31,7 @@ def downgrade() -> None:
     # ### commands auto generated by Alembic - please adjust! ###
     with op.batch_alter_table("taskv2", schema=None) as batch_op:
         batch_op.add_column(
-            sa.Column(
-                "owner", sa.VARCHAR(), autoincrement=False, nullable=True
-            )
+            sa.Column("owner", sa.VARCHAR(), autoincrement=False, nullable=True)
         )
         batch_op.alter_column(
             "taskgroupv2_id", existing_type=sa.INTEGER(), nullable=True

fractal_server/migrations/versions/8f79bd162e35_add_docs_info_and_docs_link_to_task_.py

@@ -5,11 +5,11 @@ Revises: a7f4d6137b53
 Create Date: 2023-08-01 14:13:15.322398
 
 """
+
 import sqlalchemy as sa
 import sqlmodel
 from alembic import op
 
-
 # revision identifiers, used by Alembic.
 revision = "8f79bd162e35"
 down_revision = "a7f4d6137b53"

fractal_server/migrations/versions/94a47ea2d3ff_remove_cache_dir_slurm_user_and_slurm_.py

@@ -5,6 +5,7 @@ Revises: 9c5ae74c9b98
 Create Date: 2024-09-25 09:33:18.014831
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 from sqlalchemy.dialects import postgresql

fractal_server/migrations/versions/969d84257cac_add_historyrun_task_id.py

@@ -5,10 +5,10 @@ Revises: c90a7c76e996
 Create Date: 2025-05-13 11:17:22.089308
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 
-
 # revision identifiers, used by Alembic.
 revision = "969d84257cac"
 down_revision = "c90a7c76e996"

fractal_server/migrations/versions/97f444d47249_add_applyworkflow_project_dump.py

@@ -5,10 +5,10 @@ Revises: 71eefd1dd202
 Create Date: 2023-12-05 15:36:48.573358
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 
-
 # revision identifiers, used by Alembic.
 revision = "97f444d47249"
 down_revision = "71eefd1dd202"

fractal_server/migrations/versions/981d588fe248_add_executor_error_log.py

@@ -5,11 +5,11 @@ Revises: 0f5f85bb2ae7
 Create Date: 2025-09-15 08:31:10.363449
 
 """
+
 import sqlalchemy as sa
 import sqlmodel
 from alembic import op
 
-
 # revision identifiers, used by Alembic.
 revision = "981d588fe248"
 down_revision = "0f5f85bb2ae7"

fractal_server/migrations/versions/99ea79d9e5d2_add_dataset_history.py

@@ -5,10 +5,10 @@ Revises: 8f79bd162e35
 Create Date: 2023-10-16 09:45:15.132185
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 
-
 # revision identifiers, used by Alembic.
 revision = "99ea79d9e5d2"
 down_revision = "8f79bd162e35"
@@ -20,9 +20,7 @@ def upgrade() -> None:
     # ### commands auto generated by Alembic - please adjust! ###
     with op.batch_alter_table("dataset", schema=None) as batch_op:
         batch_op.add_column(
-            sa.Column(
-                "history", sa.JSON(), server_default="[]", nullable=False
-            )
+            sa.Column("history", sa.JSON(), server_default="[]", nullable=False)
         )
 
     # ### end Alembic commands ###

fractal_server/migrations/versions/9c5ae74c9b98_add_user_settings_table.py

@@ -5,11 +5,11 @@ Revises: d9a140db5d42
 Create Date: 2024-09-24 12:01:13.393326
 
 """
+
 import sqlalchemy as sa
 import sqlmodel
 from alembic import op
 
-
 # revision identifiers, used by Alembic.
 revision = "9c5ae74c9b98"
 down_revision = "d9a140db5d42"

fractal_server/migrations/versions/9db60297b8b2_set_ondelete.py

@@ -5,8 +5,8 @@ Revises: e81103413827
 Create Date: 2025-04-07 13:13:14.596394
 
 """
-from alembic import op
 
+from alembic import op
 
 # revision identifiers, used by Alembic.
 revision = "9db60297b8b2"

fractal_server/migrations/versions/9fd26a2b0de4_add_workflow_timestamp_created.py

@@ -5,13 +5,13 @@ Revises: efa89c30e0a4
 Create Date: 2024-01-11 09:31:20.950090
 
 """
+
 from datetime import datetime
 from datetime import timezone
 
 import sqlalchemy as sa
 from alembic import op
 
-
 # revision identifiers, used by Alembic.
 revision = "9fd26a2b0de4"
 down_revision = "4cedeb448a53"

fractal_server/migrations/versions/a7f4d6137b53_add_workflow_dump_to_applyworkflow.py

@@ -5,10 +5,10 @@ Revises: 70e77f1c38b0
 Create Date: 2023-07-24 16:53:02.569582
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 
-
 # revision identifiers, used by Alembic.
 revision = "a7f4d6137b53"
 down_revision = "70e77f1c38b0"

fractal_server/migrations/versions/af1ef1c83c9b_add_accounting_tables.py

@@ -5,6 +5,7 @@ Revises: 1eac13a26c83
 Create Date: 2025-02-17 14:22:32.701581
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 from sqlalchemy.dialects import postgresql

fractal_server/migrations/versions/af8673379a5c_drop_old_filter_columns.py

@@ -5,6 +5,7 @@ Revises: db09233ad13a
 Create Date: 2025-01-30 14:44:04.302795
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 from sqlalchemy.dialects import postgresql

fractal_server/migrations/versions/b1e7f7a1ff71_task_group_for_pixi.py

@@ -5,11 +5,11 @@ Revises: 791ce783d3d8
 Create Date: 2025-05-29 16:31:17.565973
 
 """
+
 import sqlalchemy as sa
 import sqlmodel
 from alembic import op
 
-
 # revision identifiers, used by Alembic.
 revision = "b1e7f7a1ff71"
 down_revision = "791ce783d3d8"

fractal_server/migrations/versions/b3ffb095f973_json_to_jsonb.py

@@ -5,6 +5,7 @@ Revises: b1e7f7a1ff71
 Create Date: 2025-06-19 10:12:06.699107
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 from sqlalchemy.dialects import postgresql

fractal_server/migrations/versions/bc0e8b3327a7_project_sharing.py

@@ -0,0 +1,72 @@
+"""Project sharing
+
+Revision ID: bc0e8b3327a7
+Revises: e0e717ae2f26
+Create Date: 2025-11-20 11:40:03.796112
+
+"""
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "bc0e8b3327a7"
+down_revision = "e0e717ae2f26"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("linkuserprojectv2", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column(
+                "is_owner", sa.BOOLEAN(), server_default="true", nullable=False
+            )
+        )
+        batch_op.add_column(
+            sa.Column(
+                "is_verified",
+                sa.BOOLEAN(),
+                server_default="true",
+                nullable=False,
+            )
+        )
+        batch_op.add_column(
+            sa.Column(
+                "permissions", sa.String(), server_default="rwx", nullable=False
+            )
+        )
+        batch_op.create_index(
+            "ix_linkuserprojectv2_one_owner_per_project",
+            ["project_id"],
+            unique=True,
+            postgresql_where=sa.text("is_owner IS true"),
+        )
+
+        # ### end Alembic commands ###
+
+        # Manually add check constraints
+        batch_op.create_check_constraint(
+            "owner_is_verified", "NOT (is_owner AND NOT is_verified)"
+        )
+        batch_op.create_check_constraint(
+            "owner_full_permissions", "NOT (is_owner AND permissions <> 'rwx')"
+        )
+        batch_op.create_check_constraint(
+            "valid_permissions", "permissions IN ('r', 'rw', 'rwx')"
+        )
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("linkuserprojectv2", schema=None) as batch_op:
+        batch_op.drop_index(
+            "ix_linkuserprojectv2_one_owner_per_project",
+            postgresql_where=sa.text("is_owner IS true"),
+        )
+        batch_op.drop_column("permissions")
+        batch_op.drop_column("is_verified")
+        batch_op.drop_column("is_owner")
+
+    # ### end Alembic commands ###

fractal_server/migrations/versions/c90a7c76e996_job_id_in_history_run.py

@@ -5,10 +5,10 @@ Revises: f37aceb45062
 Create Date: 2025-04-16 10:44:30.219309
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 
-
 # revision identifiers, used by Alembic.
 revision = "c90a7c76e996"
 down_revision = "f37aceb45062"

fractal_server/migrations/versions/caba9fb1ea5e_drop_useroauth_user_settings_id.py

@@ -5,10 +5,10 @@ Revises: 83bc2ad3ffcc
 Create Date: 2025-11-11 16:38:27.243693
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 
-
 # revision identifiers, used by Alembic.
 revision = "caba9fb1ea5e"
 down_revision = "83bc2ad3ffcc"

fractal_server/migrations/versions/d256a7379ab8_taskgroup_activity_and_venv_info_to_.py

@@ -5,6 +5,7 @@ Revises: 19eca0dd47a9
 Create Date: 2024-11-20 15:01:52.659832
 
 """
+
 import sqlalchemy as sa
 import sqlmodel
 from alembic import op
@@ -33,16 +34,10 @@ def upgrade() -> None:
         sa.Column(
             "version", sqlmodel.sql.sqltypes.AutoString(), nullable=False
         ),
-        sa.Column(
-            "status", sqlmodel.sql.sqltypes.AutoString(), nullable=False
-        ),
-        sa.Column(
-            "action", sqlmodel.sql.sqltypes.AutoString(), nullable=False
-        ),
+        sa.Column("status", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+        sa.Column("action", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
         sa.Column("log", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
-        sa.Column(
-            "timestamp_ended", sa.DateTime(timezone=True), nullable=True
-        ),
+        sa.Column("timestamp_ended", sa.DateTime(timezone=True), nullable=True),
         sa.ForeignKeyConstraint(
             ["taskgroupv2_id"],
             ["taskgroupv2.id"],

fractal_server/migrations/versions/d4fe3708d309_make_applyworkflow_workflow_dump_non_.py

@@ -5,6 +5,7 @@ Revises: e75cac726012
 Create Date: 2023-11-30 14:58:23.917824
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 

fractal_server/migrations/versions/da2cb2ac4255_user_group_viewer_paths.py

@@ -5,10 +5,10 @@ Revises: 94a47ea2d3ff
 Create Date: 2024-09-26 14:39:26.135101
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 
-
 # revision identifiers, used by Alembic.
 revision = "da2cb2ac4255"
 down_revision = "94a47ea2d3ff"

fractal_server/migrations/versions/db09233ad13a_split_filters_and_keep_old_columns.py

@@ -5,6 +5,7 @@ Revises: 316140ff7ee1
 Create Date: 2025-01-14 14:50:46.007222
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 from sqlalchemy.dialects import postgresql

fractal_server/migrations/versions/e0e717ae2f26_delete_linkuserproject_ondelete_project.py

@@ -0,0 +1,50 @@
+"""delete LinkUserProject ondelete Project
+
+Revision ID: e0e717ae2f26
+Revises: 40d6d6511b20
+Create Date: 2025-11-17 14:33:56.348071
+
+"""
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "e0e717ae2f26"
+down_revision = "40d6d6511b20"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("linkuserprojectv2", schema=None) as batch_op:
+        batch_op.drop_constraint(
+            batch_op.f("fk_linkuserprojectv2_project_id_projectv2"),
+            type_="foreignkey",
+        )
+        batch_op.create_foreign_key(
+            batch_op.f("fk_linkuserprojectv2_project_id_projectv2"),
+            "projectv2",
+            ["project_id"],
+            ["id"],
+            ondelete="CASCADE",
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("linkuserprojectv2", schema=None) as batch_op:
+        batch_op.drop_constraint(
+            batch_op.f("fk_linkuserprojectv2_project_id_projectv2"),
+            type_="foreignkey",
+        )
+        batch_op.create_foreign_key(
+            batch_op.f("fk_linkuserprojectv2_project_id_projectv2"),
+            "projectv2",
+            ["project_id"],
+            ["id"],
+        )
+
+    # ### end Alembic commands ###

fractal_server/migrations/versions/e75cac726012_make_applyworkflow_start_timestamp_not_.py

@@ -5,6 +5,7 @@ Revises: 84bf0fffde30
 Create Date: 2023-10-30 15:51:18.808789
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 

fractal_server/migrations/versions/e81103413827_add_job_type_filters.py

@@ -5,10 +5,10 @@ Revises: 47351f8c7ebc
 Create Date: 2025-03-26 11:10:41.748248
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 
-
 # revision identifiers, used by Alembic.
 revision = "e81103413827"
 down_revision = "47351f8c7ebc"

fractal_server/migrations/versions/efa89c30e0a4_add_project_timestamp_created.py

@@ -5,6 +5,7 @@ Revises: 97f444d47249
 Create Date: 2023-12-20 10:06:52.139917
 
 """
+
 from datetime import datetime
 from datetime import timezone
 

fractal_server/migrations/versions/f0702066b007_one_submitted_job_per_dataset.py

@@ -0,0 +1,40 @@
+"""One submitted Job per Dataset
+
+Revision ID: f0702066b007
+Revises: 7910eed4cf97
+Create Date: 2025-12-01 20:54:03.137093
+
+"""
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "f0702066b007"
+down_revision = "7910eed4cf97"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("jobv2", schema=None) as batch_op:
+        batch_op.create_index(
+            "ix_jobv2_one_submitted_job_per_dataset",
+            ["dataset_id"],
+            unique=True,
+            postgresql_where=sa.text("status = 'submitted'"),
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("jobv2", schema=None) as batch_op:
+        batch_op.drop_index(
+            "ix_jobv2_one_submitted_job_per_dataset",
+            postgresql_where=sa.text("status = 'submitted'"),
+        )
+
+    # ### end Alembic commands ###

fractal_server/migrations/versions/f37aceb45062_make_historyunit_logfile_required.py

@@ -5,10 +5,10 @@ Revises: 9db60297b8b2
 Create Date: 2025-04-14 13:49:40.910342
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 
-
 # revision identifiers, used by Alembic.
 revision = "f37aceb45062"
 down_revision = "9db60297b8b2"

fractal_server/migrations/versions/f384e1c0cf5d_drop_task_default_args_columns.py

@@ -5,6 +5,7 @@ Revises: 4c308bcaea2b
 Create Date: 2023-06-06 15:10:51.838607
 
 """
+
 import sqlalchemy as sa
 from alembic import op
 from sqlalchemy.dialects import sqlite

fractal_server/migrations/versions/fbce16ff4e47_new_history_items.py

@@ -5,6 +5,7 @@ Revises: af1ef1c83c9b
 Create Date: 2025-03-14 15:25:01.083619
 
 """
+
 import sqlalchemy as sa
 import sqlmodel
 from alembic import op
@@ -37,9 +38,7 @@ def upgrade() -> None:
         sa.Column(
             "timestamp_started", sa.DateTime(timezone=True), nullable=False
         ),
-        sa.Column(
-            "status", sqlmodel.sql.sqltypes.AutoString(), nullable=False
-        ),
+        sa.Column("status", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
         sa.Column("num_available_images", sa.Integer(), nullable=False),
         sa.ForeignKeyConstraint(
             ["dataset_id"],
@@ -59,12 +58,8 @@ def upgrade() -> None:
         "historyunit",
         sa.Column("id", sa.Integer(), nullable=False),
         sa.Column("history_run_id", sa.Integer(), nullable=False),
-        sa.Column(
-            "logfile", sqlmodel.sql.sqltypes.AutoString(), nullable=True
-        ),
-        sa.Column(
-            "status", sqlmodel.sql.sqltypes.AutoString(), nullable=False
-        ),
+        sa.Column("logfile", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
+        sa.Column("status", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
         sa.Column("zarr_urls", postgresql.ARRAY(sa.String()), nullable=True),
         sa.ForeignKeyConstraint(
             ["history_run_id"],