zou 0.20.38__py3-none-any.whl → 0.20.40__py3-none-any.whl

zou/cli.py

@@ -9,12 +9,13 @@ import traceback
 from sqlalchemy.exc import IntegrityError
 
 from zou.app.utils import dbhelpers, auth, commands
-from zou.app.services import persons_service, auth_service
+from zou.app.services import persons_service, auth_service, plugins_service
 from zou.app.services.exception import (
     IsUserLimitReachedException,
     PersonNotFoundException,
     TwoFactorAuthenticationNotEnabledException,
 )
+
 from zou.app import app, config
 
 from zou import __file__ as root_path
@@ -643,5 +644,172 @@ def renormalize_movie_preview_files(
     )
 
 
+@cli.command()
+@click.option(
+    "--path",
+    required=True,
+)
+@click.option(
+    "--force",
+    is_flag=True,
+    default=False,
+    show_default=True,
+)
+def install_plugin(path, force=False):
+    """
+    Install a plugin.
+    """
+    with app.app_context():
+        plugins_service.install_plugin(path, force)
+    print(f"Plugin {path} installed. Restart the server to apply changes.")
+
+
+@cli.command()
+@click.option(
+    "--id",
+    required=True,
+)
+def uninstall_plugin(id):
+    """
+    Uninstall a plugin.
+    """
+    with app.app_context():
+        plugins_service.uninstall_plugin(id)
+    print(f"Plugin {id} uninstalled.")
+
+
+@cli.command()
+@click.option(
+    "--path",
+    required=True,
+)
+@click.option(
+    "--id",
+    help="Plugin ID (must be unique).",
+    required=True,
+)
+@click.option(
+    "--name", help="Plugin name.", default="MyPlugin", show_default=True
+)
+@click.option(
+    "--description",
+    help="Plugin description.",
+    default="My plugin description.",
+    show_default=True,
+)
+@click.option(
+    "--version", help="Plugin version.", default="0.1.0", show_default=True
+)
+@click.option(
+    "--maintainer",
+    help="Plugin maintainer.",
+    default="Author <author@author.com>",
+    show_default=True,
+)
+@click.option(
+    "--website",
+    help="Plugin website.",
+    default="mywebsite.com",
+    show_default=True,
+)
+@click.option(
+    "--license",
+    help="Plugin license.",
+    default="GPL-3.0-only",
+    show_default=True,
+)
+@click.option(
+    "--force",
+    is_flag=True,
+    default=False,
+    show_default=True,
+)
+def create_plugin_skeleton(
+    path,
+    id,
+    name,
+    description,
+    version,
+    maintainer,
+    website,
+    license,
+    force=False,
+):
+    """
+    Create a plugin skeleton.
+    """
+    plugin_path = plugins_service.create_plugin_skeleton(
+        path,
+        id,
+        name,
+        description,
+        version,
+        maintainer,
+        website,
+        license,
+        force,
+    )
+    print(f"Plugin skeleton created in '{plugin_path}'.")
+
+
+@cli.command()
+@click.option(
+    "--path",
+    required=True,
+)
+@click.option(
+    "--output-path",
+    required=True,
+)
+@click.option(
+    "--force",
+    is_flag=True,
+    default=False,
+    show_default=True,
+)
+def create_plugin_package(
+    path,
+    output_path,
+    force=False,
+):
+    """
+    Create a plugin package.
+    """
+    plugin_path = plugins_service.create_plugin_package(
+        path, output_path, force
+    )
+    print(f"Plugin package created in '{plugin_path}'.")
+
+
+@cli.command()
+@click.option(
+    "--format",
+    "output_format",
+    type=click.Choice(["table", "json"], case_sensitive=False),
+    default="table",
+    show_default=True,
+    help="Output format: table or json.",
+)
+@click.option(
+    "--verbose",
+    is_flag=True,
+    default=False,
+    help="Show more plugin information.",
+)
+@click.option(
+    "--filter-field",
+    type=click.Choice(
+        ["plugin_id", "name", "maintainer", "license"], case_sensitive=False
+    ),
+    help="Field to filter by.",
+)
+@click.option("--filter-value", type=str, help="Value to search in the field.")
+def list_plugins(output_format, verbose, filter_field, filter_value):
+    """
+    List installed plugins.
+    """
+    commands.list_plugins(output_format, verbose, filter_field, filter_value)
+
+
 if __name__ == "__main__":
     cli()

zou/event_stream.py

@@ -35,6 +35,9 @@ def _get_empty_room(current_frame=0):
         "current_preview_file_index": None,
         "current_frame": current_frame,
         "is_repeating": None,
+        "is_annotations_displayed": False,
+        "is_zoom_enabled": False,
+        "is_waveform_displayed": False,
         "is_laser_mode": None,
         "handle_in": None,
         "handle_out": None,
@@ -59,7 +62,7 @@ def _leave_room(room_id, user_id):
     room["people"] = list(set(room["people"]) - {user_id})
     if len(room["people"]) > 0:
         rooms_data[room_id] = room
-    else:
+    elif room_id in rooms_data:
         del rooms_data[room_id]
     _emit_people_updated(room_id, room["people"])
     return room
@@ -81,10 +84,17 @@ def _update_room_playing_status(data, room):
     room["is_playing"] = data.get("is_playing", False)
     room["is_repeating"] = data.get("is_repeating", False)
     room["is_laser_mode"] = data.get("is_laser_mode", False)
+    room["is_annotations_displayed"] = data.get(
+        "is_annotations_displayed", False
+    )
+    room["is_zoom_enabled"] = data.get("is_zoom_enabled", False)
+    room["is_waveform_displayed"] = data.get("is_waveform_displayed", False)
     room["current_entity_id"] = data.get("current_entity_id", None)
     room["current_entity_index"] = data.get("current_entity_index", None)
     room["current_preview_file_id"] = data.get("current_preview_file_id", None)
-    room["current_preview_file_index"] = data.get("current_preview_file_index", None)
+    room["current_preview_file_index"] = data.get(
+        "current_preview_file_index", None
+    )
     room["handle_in"] = data.get("handle_in", None)
     room["handle_out"] = data.get("handle_out", None)
     if "current_frame" in data:
@@ -193,19 +203,20 @@ def on_join(data):
     room["playlist_id"] = room_id
     rooms_data[room_id] = room
     _emit_people_updated(room_id, room["people"])
+    emit("preview-room:room-updated", room, room=room_id)
 
 
 @socketio.on("preview-room:leave", namespace="/events")
 @jwt_required()
 def on_leave(data):
     user_id = get_jwt_identity()
-    room_id = data["playlist_id"]
+    room_id = data.get("playlist_id", "")
     _leave_room(room_id, user_id)
 
 
-@socketio.on("preview-room:update-playing-status", namespace="/events")
+@socketio.on("preview-room:room-updated", namespace="/events")
 @jwt_required()
-def on_playing_status_updated(data, only_newcomer=False):
+def on_room_updated(data, only_newcomer=False):
     room, room_id = _get_room_from_data(data)
     rooms_data[room_id] = _update_room_playing_status(data, room)
     event_data = {"only_newcomer": only_newcomer, **rooms_data[room_id]}
@@ -240,6 +251,13 @@ def on_change_version(data):
     emit("preview-room:change-version", data, room=room_id)
 
 
+@socketio.on("preview-room:panzoom-changed", namespace="/events")
+@jwt_required()
+def on_change_version(data):
+    room_id = data["playlist_id"]
+    emit("preview-room:panzoom-changed", data, room=room_id)
+
+
 if __name__ == "__main__":
     socketio.run(
         app,

zou/migrations/versions/2762a797f1f9_add_people_salary_information.py

@@ -0,0 +1,52 @@
+"""add people salary information
+
+Revision ID: 2762a797f1f9
+Revises: 307edd8c639d
+Create Date: 2025-04-01 18:02:39.343857
+
+"""
+
+from alembic import op
+import sqlalchemy as sa
+import sqlalchemy_utils
+
+
+# revision identifiers, used by Alembic.
+revision = "2762a797f1f9"
+down_revision = "307edd8c639d"
+branch_labels = None
+depends_on = None
+
+from zou.app.models.person import POSITION_TYPES, SENIORITY_TYPES
+
+
+def upgrade():
+    with op.batch_alter_table("person", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column(
+                "position",
+                sqlalchemy_utils.types.choice.ChoiceType(POSITION_TYPES),
+                nullable=True,
+            )
+        )
+        batch_op.add_column(
+            sa.Column(
+                "seniority",
+                sqlalchemy_utils.types.choice.ChoiceType(SENIORITY_TYPES),
+                nullable=True,
+            )
+        )
+        batch_op.add_column(
+            sa.Column("daily_salary", sa.Integer(), nullable=True)
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("person", schema=None) as batch_op:
+        batch_op.drop_column("daily_salary")
+        batch_op.drop_column("seniority")
+        batch_op.drop_column("position")
+    # ### end Alembic commands ###

zou/migrations/versions/45f739ef962a_add_people_salary_scale_table.py

@@ -0,0 +1,70 @@
+"""add people salary scale table
+
+Revision ID: 45f739ef962a
+Revises: 2762a797f1f9
+Create Date: 2025-04-02 10:11:12.544243
+
+"""
+
+from alembic import op
+import sqlalchemy as sa
+import sqlalchemy_utils
+import sqlalchemy_utils
+import uuid
+
+# revision identifiers, used by Alembic.
+revision = "45f739ef962a"
+down_revision = "2762a797f1f9"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "salary_scale",
+        sa.Column(
+            "department_id",
+            sqlalchemy_utils.types.uuid.UUIDType(binary=False),
+            default=uuid.uuid4,
+            nullable=False,
+        ),
+        sa.Column("senior_supervisor_salary", sa.Integer(), nullable=False),
+        sa.Column("mid_supervisor_salary", sa.Integer(), nullable=False),
+        sa.Column("junior_supervisor_salary", sa.Integer(), nullable=False),
+        sa.Column("senior_lead_salary", sa.Integer(), nullable=False),
+        sa.Column("mid_lead_salary", sa.Integer(), nullable=False),
+        sa.Column("junior_lead_salary", sa.Integer(), nullable=False),
+        sa.Column("senior_artist_salary", sa.Integer(), nullable=False),
+        sa.Column("mid_artist_salary", sa.Integer(), nullable=False),
+        sa.Column("junior_artist_salary", sa.Integer(), nullable=False),
+        sa.Column(
+            "id",
+            sqlalchemy_utils.types.uuid.UUIDType(binary=False),
+            default=uuid.uuid4,
+            nullable=False,
+        ),
+        sa.Column("created_at", sa.DateTime(), nullable=True),
+        sa.Column("updated_at", sa.DateTime(), nullable=True),
+        sa.ForeignKeyConstraint(
+            ["department_id"],
+            ["department.id"],
+        ),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    with op.batch_alter_table("salary_scale", schema=None) as batch_op:
+        batch_op.create_index(
+            batch_op.f("ix_salary_scale_department_id"),
+            ["department_id"],
+            unique=False,
+        )
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("salary_scale", schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f("ix_salary_scale_department_id"))
+
+    op.drop_table("salary_scale")
+    # ### end Alembic commands ###

zou/migrations/versions/4aab1f84ad72_introduce_plugin_table.py

@@ -0,0 +1,68 @@
+"""Introduce Plugin table
+
+Revision ID: 4aab1f84ad72
+Revises: d25118cddcaa
+Create Date: 2025-04-14 15:18:29.346896
+
+"""
+
+from alembic import op
+import sqlalchemy as sa
+import sqlalchemy_utils
+import sqlalchemy_utils
+import uuid
+
+# revision identifiers, used by Alembic.
+revision = "4aab1f84ad72"
+down_revision = "d25118cddcaa"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "plugin",
+        sa.Column("plugin_id", sa.String(length=80), nullable=False),
+        sa.Column("name", sa.String(length=80), nullable=False),
+        sa.Column("description", sa.Text(), nullable=True),
+        sa.Column("version", sa.String(length=50), nullable=False),
+        sa.Column("maintainer_name", sa.String(length=200), nullable=False),
+        sa.Column(
+            "maintainer_email",
+            sqlalchemy_utils.types.email.EmailType(length=255),
+            nullable=True,
+        ),
+        sa.Column(
+            "website", sqlalchemy_utils.types.url.URLType(), nullable=True
+        ),
+        sa.Column("license", sa.String(length=80), nullable=False),
+        sa.Column(
+            "id",
+            sqlalchemy_utils.types.uuid.UUIDType(binary=False),
+            default=uuid.uuid4,
+            nullable=False,
+        ),
+        sa.Column("created_at", sa.DateTime(), nullable=True),
+        sa.Column("updated_at", sa.DateTime(), nullable=True),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    with op.batch_alter_table("plugin", schema=None) as batch_op:
+        batch_op.create_index(
+            batch_op.f("ix_plugin_name"), ["name"], unique=False
+        )
+        batch_op.create_index(
+            batch_op.f("ix_plugin_plugin_id"), ["plugin_id"], unique=True
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("plugin", schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f("ix_plugin_plugin_id"))
+        batch_op.drop_index(batch_op.f("ix_plugin_name"))
+
+    op.drop_table("plugin")
+    # ### end Alembic commands ###

zou/migrations/versions/7a16258f2fab_add_currency_field_to_budgets.py

@@ -0,0 +1,33 @@
+"""add currency field to budgets
+
+Revision ID: 7a16258f2fab
+Revises: 8ab98c178903
+Create Date: 2025-04-21 22:27:22.703525
+
+"""
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = "7a16258f2fab"
+down_revision = "8ab98c178903"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("budget", schema=None) as batch_op:
+        batch_op.add_column(sa.Column("currency", sa.String(length=3)))
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("task", schema=None) as batch_op:
+        batch_op.alter_column(
+            "difficulty", existing_type=sa.INTEGER(), nullable=True
+        )
+    # ### end Alembic commands ###

zou/migrations/versions/83e2f33a9b14_add_project_bugdet_table.py

@@ -0,0 +1,57 @@
+"""add project bugdet table
+
+Revision ID: 83e2f33a9b14
+Revises: 45f739ef962a
+Create Date: 2025-04-09 20:17:08.331320
+
+"""
+
+from alembic import op
+import sqlalchemy as sa
+import sqlalchemy_utils
+import sqlalchemy_utils
+import uuid
+
+# revision identifiers, used by Alembic.
+revision = "83e2f33a9b14"
+down_revision = "45f739ef962a"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "budget",
+        sa.Column(
+            "project_id",
+            sqlalchemy_utils.types.uuid.UUIDType(binary=False),
+            default=uuid.uuid4,
+            nullable=False,
+        ),
+        sa.Column("revision", sa.Integer(), nullable=False),
+        sa.Column("name", sa.String(length=255), nullable=False),
+        sa.Column(
+            "id",
+            sqlalchemy_utils.types.uuid.UUIDType(binary=False),
+            default=uuid.uuid4,
+            nullable=False,
+        ),
+        sa.Column("created_at", sa.DateTime(), nullable=True),
+        sa.Column("updated_at", sa.DateTime(), nullable=True),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    with op.batch_alter_table("budget", schema=None) as batch_op:
+        batch_op.create_index(
+            batch_op.f("ix_budget_project_id"), ["project_id"], unique=False
+        )
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("budget", schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f("ix_budget_project_id"))
+
+    op.drop_table("budget")
+    # ### end Alembic commands ###

zou/migrations/versions/8ab98c178903_add_budget_entry_table.py

@@ -0,0 +1,123 @@
+"""add budget entry table
+
+Revision ID: 8ab98c178903
+Revises: 83e2f33a9b14
+Create Date: 2025-04-15 16:56:25.451489
+
+"""
+
+from alembic import op
+import sqlalchemy as sa
+import sqlalchemy_utils
+import sqlalchemy_utils
+import uuid
+
+# revision identifiers, used by Alembic.
+revision = "8ab98c178903"
+down_revision = "83e2f33a9b14"
+branch_labels = None
+depends_on = None
+
+from zou.app.models.person import POSITION_TYPES, SENIORITY_TYPES
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "budget_entry",
+        sa.Column(
+            "budget_id",
+            sqlalchemy_utils.types.uuid.UUIDType(binary=False),
+            default=uuid.uuid4,
+            nullable=False,
+        ),
+        sa.Column(
+            "department_id",
+            sqlalchemy_utils.types.uuid.UUIDType(binary=False),
+            default=uuid.uuid4,
+            nullable=False,
+        ),
+        sa.Column(
+            "person_id",
+            sqlalchemy_utils.types.uuid.UUIDType(binary=False),
+            default=uuid.uuid4,
+            nullable=True,
+        ),
+        sa.Column("start_date", sa.Date(), nullable=False),
+        sa.Column("months_duration", sa.Integer(), nullable=False),
+        sa.Column("daily_salary", sa.Float(), nullable=False),
+        sa.Column(
+            "position",
+            sqlalchemy_utils.types.choice.ChoiceType(POSITION_TYPES),
+            nullable=True,
+        ),
+        sa.Column(
+            "seniority",
+            sqlalchemy_utils.types.choice.ChoiceType(SENIORITY_TYPES),
+            nullable=True,
+        ),
+        sa.Column(
+            "id",
+            sqlalchemy_utils.types.uuid.UUIDType(binary=False),
+            default=uuid.uuid4,
+            nullable=False,
+        ),
+        sa.Column("created_at", sa.DateTime(), nullable=True),
+        sa.Column("updated_at", sa.DateTime(), nullable=True),
+        sa.ForeignKeyConstraint(
+            ["budget_id"],
+            ["budget.id"],
+        ),
+        sa.ForeignKeyConstraint(
+            ["department_id"],
+            ["department.id"],
+        ),
+        sa.ForeignKeyConstraint(
+            ["person_id"],
+            ["person.id"],
+        ),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    with op.batch_alter_table("budget_entry", schema=None) as batch_op:
+        batch_op.create_index(
+            batch_op.f("ix_budget_entry_budget_id"),
+            ["budget_id"],
+            unique=False,
+        )
+        batch_op.create_index(
+            batch_op.f("ix_budget_entry_department_id"),
+            ["department_id"],
+            unique=False,
+        )
+        batch_op.create_index(
+            batch_op.f("ix_budget_entry_person_id"),
+            ["person_id"],
+            unique=False,
+        )
+
+    with op.batch_alter_table("budget", schema=None) as batch_op:
+        batch_op.alter_column(
+            "project_id", existing_type=sa.UUID(), nullable=True
+        )
+        batch_op.create_foreign_key(
+            "budget_project_id_fkey", "project", ["project_id"], ["id"]
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("budget", schema=None) as batch_op:
+        batch_op.drop_constraint("budget_project_id_fkey", type_="foreignkey")
+        batch_op.alter_column(
+            "project_id", existing_type=sa.UUID(), nullable=False
+        )
+
+    with op.batch_alter_table("budget_entry", schema=None) as batch_op:
+        batch_op.drop_index(batch_op.f("ix_budget_entry_person_id"))
+        batch_op.drop_index(batch_op.f("ix_budget_entry_department_id"))
+        batch_op.drop_index(batch_op.f("ix_budget_entry_budget_id"))
+
+    op.drop_table("budget_entry")
+    # ### end Alembic commands ###