Filtering options, closes #7

Author: jsbroks

app/api/datasets.py

@@ -1,3 +1,4 @@
+from flask import request
 from flask_restplus import Namespace, Resource, reqparse
 from flask_login import login_required, current_user
 from werkzeug.datastructures import FileStorage
@@ -26,7 +27,7 @@
 page_data = reqparse.RequestParser()
 page_data.add_argument('page', default=1, type=int)
 page_data.add_argument('limit', default=20, type=int)
-page_data.add_argument('folder', required=False, default='', help='Folder for data')
+page_data.add_argument('folder', default='', help='Folder for data')
 
 delete_data = reqparse.RequestParser()
 delete_data.add_argument('fully', default=False, type=bool,
@@ -217,6 +218,25 @@ def get(self, dataset_id):
         page = args['page']
         folder = args['folder']
 
+        args = dict(request.args)
+        if args.get('limit') != None:
+            del args['limit']
+        if args.get('page') != None:
+            del args['page']
+        if args.get('folder') != None:
+            del args['folder']
+
+        query = {}
+        for key, value in args.items():
+            lower = value.lower()
+            if lower in ["true", "false"]:
+                value = json.loads(lower)
+            
+            if len(lower) != 0:
+                query[key] = value
+        
+        # print(query, flush=True)
+        
         # Check if dataset exists
         dataset = current_user.datasets.filter(id=dataset_id, deleted=False).first()
         if dataset is None:
@@ -233,9 +253,9 @@ def get(self, dataset_id):
         if not os.path.exists(directory):
             return {'message': 'Directory does not exist.'}, 400
 
-        images = ImageModel.objects(dataset_id=dataset_id, path__startswith=directory, deleted=False) \
+        images = ImageModel.objects(dataset_id=dataset_id, path__startswith=directory, deleted=False, **query) \
             .order_by('file_name').only('id', 'file_name', 'annotating')
-
+        
         pagination = Pagination(images.count(), limit, page)
         images = query_util.fix_ids(images[pagination.start:pagination.end])
 

app/api/images.py

@@ -19,7 +19,6 @@
 image_all.add_argument('page', default=1, type=int)
 image_all.add_argument('perPage', default=50, type=int, required=False)
 
-
 image_upload = reqparse.RequestParser()
 image_upload.add_argument('image', location='files',
                           type=FileStorage, required=True,
@@ -47,7 +46,7 @@ def get(self):
         args = image_all.parse_args()
         per_page = args['perPage']
         page = args['page']-1
-        fields = args.get('fields', "")
+        fields = args.get('fields', '')
 
         images = current_user.images.filter(deleted=False)
         total = images.count()

client/src/components/annotator/panels/PanelButton.vue renamed to client/src/components/PanelButton.vue

@@ -0,0 +1,53 @@
+<template>
+  <button
+    class="btn btn-outline-light tool-input-button"
+    @click="$emit('click', current.value)"
+  >
+    {{ current.name }}
+  </button>
+</template>
+
+<script>
+export default {
+  name: "PanelButton",
+  props: {
+    name: {
+      type: String,
+      required: true
+    },
+    options: {
+      type: Array,
+      required: true
+    },
+    default: {
+      type: Number,
+      default: 0
+    }
+  },
+  data() {
+    return {
+      currentIndex: this.default
+    }
+  }
+  computed: {
+    current() {
+      return this.options[this.currentIndex]
+    }
+  }
+};
+</script>
+
+<style scoped>
+.tool-input-button {
+  height: 20px;
+  border-color: #4b5162;
+  padding: 0 0 0 11px;
+  font-size: 12px;
+  width: 100%;
+}
+
+.tool-input-button:hover {
+  border-color: lightgray;
+  background-color: white;
+}
+</style>

client/src/components/PanelButtonSelect.vue

@@ -3,7 +3,11 @@
     <div class="input-group-prepend tool-option-pre">
       <span class="input-group-text tool-option-font">{{ name }}</span>
     </div>
-    <input v-model="localValue" class="form-control tool-option-input" />
+    <input
+      v-model="localValue"
+      class="form-control tool-option-input"
+      @keyup.enter="$emit('submit')"
+    />
   </div>
 </template>
 

client/src/components/annotator/panels/PanelInputNumber.vue renamed to client/src/components/PanelInputNumber.vue

@@ -267,14 +267,14 @@ export default {
         this.$emit("click", this.index);
       }
     },
-    onMouseEnter(event) {
+    onMouseEnter() {
       if (this.compoundPath == null) return;
 
       this.compoundPath.selected = true;
     },
-    onMouseLeave(event) {
+    onMouseLeave() {
       if (this.compoundPath == null) return;
-      
+
       this.compoundPath.selected = false;
     },
     getCompoundPath() {

client/src/components/annotator/panels/PanelInputString.vue renamed to client/src/components/PanelInputString.vue

@@ -15,8 +15,8 @@
 </template>
 
 <script>
-import PanelInputString from "@/components/annotator/panels/PanelInputString";
-import PanelInputNumber from "@/components/annotator/panels/PanelInputNumber";
+import PanelInputString from "@/components/PanelInputString";
+import PanelInputNumber from "@/components/PanelInputNumber";
 
 export default {
   name: "BrushPanel",

client/src/components/annotator/panels/PanelToggle.vue renamed to client/src/components/PanelToggle.vue

@@ -15,8 +15,8 @@
 </template>
 
 <script>
-import PanelInputString from "@/components/annotator/panels/PanelInputString";
-import PanelInputNumber from "@/components/annotator/panels/PanelInputNumber";
+import PanelInputString from "@/components/PanelInputString";
+import PanelInputNumber from "@/components/PanelInputNumber";
 
 export default {
   name: "EraserPanel",