zou 0.20.83__py3-none-any.whl → 0.20.85__py3-none-any.whl

zou/app/blueprints/source/shotgun/status.py

@@ -1,4 +1,5 @@
 from flask import current_app
+from flask_jwt_extended import jwt_required
 
 from zou.app.models.task_status import TaskStatus
 from zou.app.utils import colors
@@ -10,6 +11,88 @@ from zou.app.blueprints.source.shotgun.base import (
 
 
 class ImportShotgunStatusResource(BaseImportShotgunResource):
+    @jwt_required()
+    def post(self):
+        """
+        Import shotgun task statuses
+        ---
+        description: Import Shotgun task statuses. Send a list of Shotgun
+          status entries in the JSON body. Returns created or updated task
+          statuses with colors.
+        tags:
+          - Import
+        requestBody:
+          required: true
+          content:
+            application/json:
+              schema:
+                type: array
+                items:
+                  type: object
+                  properties:
+                    id:
+                      type: integer
+                      description: Shotgun ID of the status
+                      example: 12345
+                    name:
+                      type: string
+                      description: Status name
+                      example: "In Progress"
+                    code:
+                      type: string
+                      description: Status short code
+                      example: "IP"
+                    bg_color:
+                      type: string
+                      description: Background color in RGB format
+                      example: "255,128,0"
+              example:
+                - id: 12345
+                  name: "In Progress"
+                  code: "IP"
+                  bg_color: "255,128,0"
+        responses:
+          200:
+            description: Task statuses imported successfully
+            content:
+              application/json:
+                schema:
+                  type: array
+                  items:
+                    type: object
+                    properties:
+                      id:
+                        type: string
+                        format: uuid
+                        description: Task status unique identifier
+                        example: a24a6ea4-ce75-4665-a070-57453082c25
+                      name:
+                        type: string
+                        description: Status name
+                        example: "In Progress"
+                      short_name:
+                        type: string
+                        description: Status short code
+                        example: "IP"
+                      color:
+                        type: string
+                        description: Status color in hex format
+                        example: "#FF8000"
+                      created_at:
+                        type: string
+                        format: date-time
+                        description: Creation timestamp
+                        example: "2024-01-15T10:30:00Z"
+                      updated_at:
+                        type: string
+                        format: date-time
+                        description: Update timestamp
+                        example: "2024-01-15T11:00:00Z"
+          400:
+            description: Invalid request body or data format error
+        """
+        return super().post()
+
     def import_entry(self, data):
         task_status = self.get_existing_status(data)
         if task_status is None:
@@ -44,3 +127,49 @@ class ImportShotgunStatusResource(BaseImportShotgunResource):
 class ImportRemoveShotgunStatusResource(ImportRemoveShotgunBaseResource):
     def __init__(self):
         ImportRemoveShotgunBaseResource.__init__(self, TaskStatus)
+
+    @jwt_required()
+    def post(self):
+        """
+        Remove shotgun task status
+        ---
+        description: Remove a Shotgun task status from the database. Provide
+          the Shotgun entry ID in the JSON body.
+        tags:
+          - Import
+        requestBody:
+          required: true
+          content:
+            application/json:
+              schema:
+                type: object
+                required:
+                  - id
+                properties:
+                  id:
+                    type: integer
+                    description: Shotgun ID of the task status to remove
+                    example: 12345
+              example:
+                id: 12345
+        responses:
+          200:
+            description: Removal result returned
+            content:
+              application/json:
+                schema:
+                  type: object
+                  properties:
+                    success:
+                      type: boolean
+                      description: Whether the removal was successful
+                      example: true
+                    removed_instance_id:
+                      type: string
+                      format: uuid
+                      description: ID of the removed task status, if found
+                      example: a24a6ea4-ce75-4665-a070-57453082c25
+          400:
+            description: Invalid request body or instance not found
+        """
+        return super().post()

zou/app/blueprints/source/shotgun/steps.py

@@ -1,5 +1,6 @@
 from flask import current_app
 from flask_restful import Resource
+from flask_jwt_extended import jwt_required
 
 from zou.app.models.department import Department
 from zou.app.models.task_type import TaskType
@@ -16,6 +17,97 @@ class ImportShotgunStepsResource(BaseImportShotgunResource):
     def __init__(self):
         Resource.__init__(self)
 
+    @jwt_required()
+    def post(self):
+        """
+        Import shotgun steps
+        ---
+        description: Import Shotgun steps (task types). Send a list of
+          Shotgun step entries in the JSON body. Returns created or updated
+          task types with departments.
+        tags:
+          - Import
+        requestBody:
+          required: true
+          content:
+            application/json:
+              schema:
+                type: array
+                items:
+                  type: object
+                  properties:
+                    id:
+                      type: integer
+                      description: Shotgun ID of the step
+                      example: 12345
+                    code:
+                      type: string
+                      description: Step code (used to extract department)
+                      example: "Animation Modeling"
+                    short_name:
+                      type: string
+                      description: Step short name
+                      example: "mod"
+                    color:
+                      type: string
+                      description: Color in RGB format
+                      example: "255,128,0"
+                    entity_type:
+                      type: string
+                      description: Entity type this step applies to
+                      example: "Asset"
+              example:
+                - id: 12345
+                  code: "Animation Modeling"
+                  short_name: "mod"
+                  color: "255,128,0"
+                  entity_type: "Asset"
+        responses:
+          200:
+            description: Task types imported successfully
+            content:
+              application/json:
+                schema:
+                  type: array
+                  items:
+                    type: object
+                    properties:
+                      id:
+                        type: string
+                        format: uuid
+                        description: Task type unique identifier
+                        example: a24a6ea4-ce75-4665-a070-57453082c25
+                      name:
+                        type: string
+                        description: Task type name
+                        example: "Animation Modeling"
+                      short_name:
+                        type: string
+                        description: Task type short name
+                        example: "mod"
+                      color:
+                        type: string
+                        description: Task type color in hex format
+                        example: "#FF8000"
+                      for_entity:
+                        type: string
+                        description: Entity type
+                        example: "Asset"
+                      created_at:
+                        type: string
+                        format: date-time
+                        description: Creation timestamp
+                        example: "2024-01-15T10:30:00Z"
+                      updated_at:
+                        type: string
+                        format: date-time
+                        description: Update timestamp
+                        example: "2024-01-15T11:00:00Z"
+          400:
+            description: Invalid request body or data format error
+        """
+        return super().post()
+
     def extract_data(self, sg_step):
         color = self.extract_color(sg_step)
         department_name = self.extract_department_name(sg_step)
@@ -85,3 +177,49 @@ class ImportShotgunStepsResource(BaseImportShotgunResource):
 class ImportRemoveShotgunStepResource(ImportRemoveShotgunBaseResource):
     def __init__(self):
         ImportRemoveShotgunBaseResource.__init__(self, TaskType)
+
+    @jwt_required()
+    def post(self):
+        """
+        Remove shotgun step
+        ---
+        description: Remove a Shotgun step (task type) from the database.
+          Provide the Shotgun entry ID in the JSON body.
+        tags:
+          - Import
+        requestBody:
+          required: true
+          content:
+            application/json:
+              schema:
+                type: object
+                required:
+                  - id
+                properties:
+                  id:
+                    type: integer
+                    description: Shotgun ID of the step to remove
+                    example: 12345
+              example:
+                id: 12345
+        responses:
+          200:
+            description: Removal result returned
+            content:
+              application/json:
+                schema:
+                  type: object
+                  properties:
+                    success:
+                      type: boolean
+                      description: Whether the removal was successful
+                      example: true
+                    removed_instance_id:
+                      type: string
+                      format: uuid
+                      description: ID of the removed step, if found
+                      example: a24a6ea4-ce75-4665-a070-57453082c25
+          400:
+            description: Invalid request body or instance not found
+        """
+        return super().post()

zou/app/blueprints/source/shotgun/tasks.py

@@ -1,4 +1,5 @@
 from flask import current_app
+from flask_jwt_extended import jwt_required
 
 from zou.app.models.task_type import TaskType
 from zou.app.models.task_status import TaskStatus
@@ -15,6 +16,158 @@ from zou.app.blueprints.source.shotgun.base import (
 
 
 class ImportShotgunTasksResource(BaseImportShotgunResource):
+    @jwt_required()
+    def post(self):
+        """
+        Import shotgun tasks
+        ---
+        description: Import Shotgun tasks. Send a list of Shotgun task
+          entries in the JSON body. Only tasks with steps and projects are
+          imported. Returns created or updated tasks with assignees.
+        tags:
+          - Import
+        requestBody:
+          required: true
+          content:
+            application/json:
+              schema:
+                type: array
+                items:
+                  type: object
+                  properties:
+                    id:
+                      type: integer
+                      description: Shotgun ID of the task
+                      example: 12345
+                    cached_display_name:
+                      type: string
+                      description: Task display name
+                      example: "Modeling for Asset"
+                    start_date:
+                      type: string
+                      format: date
+                      description: Start date
+                      example: "2024-01-15"
+                    due_date:
+                      type: string
+                      format: date
+                      description: Due date
+                      example: "2024-02-15"
+                    sg_sort_order:
+                      type: integer
+                      description: Sort order
+                      example: 1
+                    duration:
+                      type: integer
+                      description: Duration in days
+                      example: 30
+                    step:
+                      type: object
+                      description: Step (task type) information
+                      properties:
+                        id:
+                          type: integer
+                          example: 11111
+                    project:
+                      type: object
+                      description: Project information
+                      properties:
+                        id:
+                          type: integer
+                          example: 22222
+                    entity:
+                      type: object
+                      description: Entity information
+                      properties:
+                        id:
+                          type: integer
+                          example: 33333
+                        type:
+                          type: string
+                          example: "Asset"
+                    sg_status_list:
+                      type: string
+                      description: Status short name
+                      example: "IP"
+                    created_by:
+                      type: object
+                      description: Creator information
+                      properties:
+                        id:
+                          type: integer
+                          example: 44444
+                    task_assignees:
+                      type: array
+                      description: Task assignees
+                      items:
+                        type: object
+                        properties:
+                          id:
+                            type: integer
+                            example: 55555
+              example:
+                - id: 12345
+                  cached_display_name: "Modeling for Asset"
+                  start_date: "2024-01-15"
+                  due_date: "2024-02-15"
+                  sg_sort_order: 1
+                  duration: 30
+                  step:
+                    id: 11111
+                  project:
+                    id: 22222
+                  entity:
+                    id: 33333
+                    type: "Asset"
+                  sg_status_list: "IP"
+                  created_by:
+                    id: 44444
+                  task_assignees:
+                    - id: 55555
+        responses:
+          200:
+            description: Tasks imported successfully
+            content:
+              application/json:
+                schema:
+                  type: array
+                  items:
+                    type: object
+                    properties:
+                      id:
+                        type: string
+                        format: uuid
+                        description: Task unique identifier
+                        example: a24a6ea4-ce75-4665-a070-57453082c25
+                      name:
+                        type: string
+                        description: Task name
+                        example: "Modeling for Asset"
+                      start_date:
+                        type: string
+                        format: date
+                        description: Start date
+                        example: "2024-01-15"
+                      due_date:
+                        type: string
+                        format: date
+                        description: Due date
+                        example: "2024-02-15"
+                      created_at:
+                        type: string
+                        format: date-time
+                        description: Creation timestamp
+                        example: "2024-01-15T10:30:00Z"
+                      updated_at:
+                        type: string
+                        format: date-time
+                        description: Update timestamp
+                        example: "2024-01-15T11:00:00Z"
+          400:
+            description: Invalid request body or data format error
+        """
+        return super().post()
+
     def prepare_import(self):
         self.project_ids = Project.get_id_map()
         self.person_ids = Person.get_id_map()
@@ -122,6 +275,52 @@ class ImportRemoveShotgunTaskResource(ImportRemoveShotgunBaseResource):
     def __init__(self):
         ImportRemoveShotgunBaseResource.__init__(self, Task, self.delete_func)
 
+    @jwt_required()
+    def post(self):
+        """
+        Remove shotgun task
+        ---
+        description: Remove a Shotgun task from the database. Provide the
+          Shotgun entry ID in the JSON body.
+        tags:
+          - Import
+        requestBody:
+          required: true
+          content:
+            application/json:
+              schema:
+                type: object
+                required:
+                  - id
+                properties:
+                  id:
+                    type: integer
+                    description: Shotgun ID of the task to remove
+                    example: 12345
+              example:
+                id: 12345
+        responses:
+          200:
+            description: Removal result returned
+            content:
+              application/json:
+                schema:
+                  type: object
+                  properties:
+                    success:
+                      type: boolean
+                      description: Whether the removal was successful
+                      example: true
+                    removed_instance_id:
+                      type: string
+                      format: uuid
+                      description: ID of the removed task, if found
+                      example: a24a6ea4-ce75-4665-a070-57453082c25
+          400:
+            description: Invalid request body or instance not found
+        """
+        return super().post()
+
     def delete_func(self, entity):
         deletion_service.remove_task(entity.id)
         tasks_service.clear_task_cache(str(entity.id))

zou/app/blueprints/source/shotgun/team.py

@@ -1,4 +1,5 @@
 from flask import current_app
+from flask_jwt_extended import jwt_required
 
 from zou.app.models.project import Project
 from zou.app.models.project import ProjectPersonLink
@@ -14,6 +15,91 @@ class ImportShotgunProjectConnectionsResource(BaseImportShotgunResource):
     def __init__(self):
         BaseImportShotgunResource.__init__(self)
 
+    @jwt_required()
+    def post(self):
+        """
+        Import shotgun project connections
+        ---
+        description: Import Shotgun project-user connections. Send a list of
+          Shotgun project connection entries in the JSON body. Returns
+          projects with team members added.
+        tags:
+          - Import
+        requestBody:
+          required: true
+          content:
+            application/json:
+              schema:
+                type: array
+                items:
+                  type: object
+                  properties:
+                    id:
+                      type: integer
+                      description: Shotgun ID of the connection
+                      example: 12345
+                    project:
+                      type: object
+                      description: Project information
+                      properties:
+                        id:
+                          type: integer
+                          example: 11111
+                    user:
+                      type: object
+                      description: User information
+                      properties:
+                        id:
+                          type: integer
+                          example: 22222
+              example:
+                - id: 12345
+                  project:
+                    id: 11111
+                  user:
+                    id: 22222
+        responses:
+          200:
+            description: Project connections imported successfully
+            content:
+              application/json:
+                schema:
+                  type: array
+                  items:
+                    type: object
+                    properties:
+                      id:
+                        type: string
+                        format: uuid
+                        description: Project unique identifier
+                        example: a24a6ea4-ce75-4665-a070-57453082c25
+                      name:
+                        type: string
+                        description: Project name
+                        example: "My Project"
+                      team:
+                        type: array
+                        description: Team member IDs
+                        items:
+                          type: string
+                          format: uuid
+                        example:
+                          - b35b7fb5-df86-5776-b181-68564193d36
+                      created_at:
+                        type: string
+                        format: date-time
+                        description: Creation timestamp
+                        example: "2024-01-15T10:30:00Z"
+                      updated_at:
+                        type: string
+                        format: date-time
+                        description: Update timestamp
+                        example: "2024-01-15T11:00:00Z"
+          400:
+            description: Invalid request body or data format error
+        """
+        return super().post()
+
     def prepare_import(self):
         pass
 
@@ -55,3 +141,49 @@ class ImportRemoveShotgunProjectConnectionResource(
 ):
     def __init__(self):
         ImportRemoveShotgunBaseResource.__init__(self, ProjectPersonLink)
+
+    @jwt_required()
+    def post(self):
+        """
+        Remove shotgun project connection
+        ---
+        description: Remove a Shotgun project-user connection from the
+          database. Provide the Shotgun entry ID in the JSON body.
+        tags:
+          - Import
+        requestBody:
+          required: true
+          content:
+            application/json:
+              schema:
+                type: object
+                required:
+                  - id
+                properties:
+                  id:
+                    type: integer
+                    description: Shotgun ID of the connection to remove
+                    example: 12345
+              example:
+                id: 12345
+        responses:
+          200:
+            description: Removal result returned
+            content:
+              application/json:
+                schema:
+                  type: object
+                  properties:
+                    success:
+                      type: boolean
+                      description: Whether the removal was successful
+                      example: true
+                    removed_instance_id:
+                      type: string
+                      format: uuid
+                      description: ID of the removed connection, if found
+                      example: a24a6ea4-ce75-4665-a070-57453082c25
+          400:
+            description: Invalid request body or instance not found
+        """
+        return super().post()