From 01ee5903be302f9807656c9405807c08ae41d30c Mon Sep 17 00:00:00 2001
From: Hironsan <hiroki.nakayama.py@gmail.com>
Date: Tue, 8 Aug 2023 16:14:41 +0900
Subject: [PATCH] Enable to bulk assign in frontend

---
 frontend/components/example/ActionMenu.vue    |   8 +-
 .../components/example/FormAssignment.vue     | 132 ++++++++++++++++++
 frontend/pages/projects/_id/dataset/index.vue |  12 ++
 .../example/apiAssignmentRepository.ts        |   5 +
 4 files changed, 156 insertions(+), 1 deletion(-)
 create mode 100644 frontend/components/example/FormAssignment.vue

diff --git a/frontend/components/example/ActionMenu.vue b/frontend/components/example/ActionMenu.vue
index 33b88cc5..47d39b8e 100644
--- a/frontend/components/example/ActionMenu.vue
+++ b/frontend/components/example/ActionMenu.vue
@@ -5,12 +5,13 @@
     @create="$emit('create')"
     @upload="$emit('upload')"
     @download="$emit('download')"
+    @assign="$emit('assign')"
   />
 </template>
 
 <script lang="ts">
 import Vue from 'vue'
-import { mdiUpload, mdiDownload } from '@mdi/js'
+import { mdiAccountCheck, mdiUpload, mdiDownload } from '@mdi/js'
 import ActionMenu from '~/components/utils/ActionMenu.vue'
 
 export default Vue.extend({
@@ -30,6 +31,11 @@ export default Vue.extend({
           title: this.$t('dataset.exportDataset'),
           icon: mdiDownload,
           event: 'download'
+        },
+        {
+          title: 'Assign to member',
+          icon: mdiAccountCheck,
+          event: 'assign'
         }
       ]
     }
diff --git a/frontend/components/example/FormAssignment.vue b/frontend/components/example/FormAssignment.vue
new file mode 100644
index 00000000..f249e7d2
--- /dev/null
+++ b/frontend/components/example/FormAssignment.vue
@@ -0,0 +1,132 @@
+<template>
+  <v-card>
+    <v-toolbar color="primary white--text" flat>
+      <v-toolbar-title>Assign examples to members</v-toolbar-title>
+    </v-toolbar>
+    <v-card-text>
+      <v-container fluid>
+        <v-row>
+          <v-card-title class="pb-0 pl-3">Select assignment strategy</v-card-title>
+          <v-col cols="12">
+            <v-select
+              v-model="selectedStrategy"
+              :items="strategies"
+              item-text="displayName"
+              item-value="value"
+              outlined
+              dense
+              hide-details
+            ></v-select>
+            {{ strategies.find((strategy) => strategy.value === selectedStrategy)?.description }}
+            Note that assigning to an administrator does not make sense.
+          </v-col>
+        </v-row>
+        <v-row>
+          <v-card-title class="pb-0 pl-3">Allocate weights</v-card-title>
+          <v-col v-for="(member, i) in members" :key="member.id" cols="12" class="pt-0 pb-0">
+            <v-subheader class="pl-0">{{ member.username }}</v-subheader>
+            <v-slider v-model="workloadAllocation[i]" :max="100" class="align-center">
+              <template #append>
+                <v-text-field
+                  v-model="workloadAllocation[i]"
+                  class="mt-0 pt-0"
+                  type="number"
+                  style="width: 60px"
+                ></v-text-field>
+              </template>
+            </v-slider>
+          </v-col>
+        </v-row>
+      </v-container>
+    </v-card-text>
+    <v-card-actions>
+      <v-spacer />
+      <v-btn class="text-capitalize" text color="primary" data-test="cancel-button" @click="cancel">
+        Cancel
+      </v-btn>
+      <v-btn
+        class="text-none"
+        text
+        :disabled="!validateWeight"
+        data-test="delete-button"
+        @click="agree"
+      >
+        Assign
+      </v-btn>
+    </v-card-actions>
+  </v-card>
+</template>
+
+<script lang="ts">
+import Vue from 'vue'
+import { MemberItem } from '~/domain/models/member/member'
+
+export default Vue.extend({
+  data() {
+    return {
+      members: [] as MemberItem[],
+      workloadAllocation: [] as number[],
+      selectedStrategy: 'weighted_sequential'
+    }
+  },
+
+  async fetch() {
+    this.members = await this.$repositories.member.list(this.projectId)
+    this.workloadAllocation = this.members.map(() => Math.round(100 / this.members.length))
+  },
+
+  computed: {
+    projectId() {
+      return this.$route.params.id
+    },
+
+    strategies() {
+      return [
+        {
+          displayName: 'Weighted sequential',
+          value: 'weighted_sequential',
+          description:
+            'Assign examples to members in order of their workload. The total weight must equal 100.'
+        },
+        {
+          displayName: 'Weighted random',
+          value: 'weighted_random',
+          description:
+            'Assign examples to members randomly based on their workload. The total weight must equal 100.'
+        },
+        {
+          displayName: 'Sampling without replacement',
+          value: 'sampling_without_replacement',
+          description: 'Assign examples to members randomly without replacement.'
+        }
+      ]
+    },
+
+    validateWeight(): boolean {
+      if (this.selectedStrategy === 'sampling_without_replacement') {
+        return true
+      } else {
+        return this.workloadAllocation.reduce((acc, cur) => acc + cur, 0) === 100
+      }
+    }
+  },
+
+  methods: {
+    agree() {
+      const workloads = this.workloadAllocation.map((weight, i) => ({
+        weight,
+        member_id: this.members[i].id
+      }))
+      console.log(workloads)
+      this.$repositories.assignment.bulkAssign(this.projectId, {
+        strategy_name: this.selectedStrategy,
+        workloads
+      })
+      this.$emit('assigned')
+    },
+    cancel() {
+      this.$emit('cancel')
+    }
+  }
+})
+</script>
diff --git a/frontend/pages/projects/_id/dataset/index.vue b/frontend/pages/projects/_id/dataset/index.vue
index ee2ebc61..a66c2f04 100644
--- a/frontend/pages/projects/_id/dataset/index.vue
+++ b/frontend/pages/projects/_id/dataset/index.vue
@@ -4,6 +4,7 @@
       <action-menu
         @upload="$router.push('dataset/import')"
         @download="$router.push('dataset/export')"
+        @assign="dialogAssignment = true"
       />
       <v-btn
         class="text-capitalize ms-2"
@@ -33,6 +34,9 @@
       <v-dialog v-model="dialogDeleteAll">
         <form-delete-bulk @cancel="dialogDeleteAll = false" @remove="removeAll" />
       </v-dialog>
+      <v-dialog v-model="dialogAssignment">
+        <form-assignment @assigned="assigned" @cancel="dialogAssignment = false" />
+      </v-dialog>
     </v-card-title>
     <image-list
       v-if="project.isImageProject"
@@ -83,6 +87,7 @@ import { mapGetters } from 'vuex'
 import Vue from 'vue'
 import { NuxtAppOptions } from '@nuxt/types'
 import DocumentList from '@/components/example/DocumentList.vue'
+import FormAssignment from '~/components/example/FormAssignment.vue'
 import FormDelete from '@/components/example/FormDelete.vue'
 import FormDeleteBulk from '@/components/example/FormDeleteBulk.vue'
 import ActionMenu from '~/components/example/ActionMenu.vue'
@@ -98,6 +103,7 @@ export default Vue.extend({
     AudioList,
     DocumentList,
     ImageList,
+    FormAssignment,
     FormDelete,
     FormDeleteBulk
   },
@@ -114,6 +120,7 @@ export default Vue.extend({
     return {
       dialogDelete: false,
       dialogDeleteAll: false,
+      dialogAssignment: false,
       item: {} as ExampleListDTO,
       selected: [] as ExampleDTO[],
       members: [] as MemberItem[],
@@ -204,6 +211,11 @@ export default Vue.extend({
     async unassign(assignmentId: string) {
       await this.$repositories.assignment.unassign(this.projectId, assignmentId)
       this.item = await this.$services.example.list(this.projectId, this.$route.query)
+    },
+
+    async assigned() {
+      this.dialogAssignment = false
+      this.item = await this.$services.example.list(this.projectId, this.$route.query)
     }
   }
 })
diff --git a/frontend/repositories/example/apiAssignmentRepository.ts b/frontend/repositories/example/apiAssignmentRepository.ts
index 08bc9d82..762924b5 100644
--- a/frontend/repositories/example/apiAssignmentRepository.ts
+++ b/frontend/repositories/example/apiAssignmentRepository.ts
@@ -15,4 +15,9 @@ export class APIAssignmentRepository {
     const url = `/projects/${projectId}/assignments/${assignmentId}`
     await this.request.delete(url)
   }
+
+  async bulkAssign(projectId: string, workloadAllocation: Object): Promise<void> {
+    const url = `/projects/${projectId}/assignments/bulk_assign`
+    await this.request.post(url, workloadAllocation)
+  }
 }