dashboard: add VmDescFilter and 'shared with me' search choice

circle/dashboard/forms.py

@@ -1531,6 +1531,7 @@ class VmResourcesForm(forms.ModelForm):
 vm_search_choices = (
     ("owned", _("owned")),
     ("shared", _("shared")),
+    ("shared_with_me", _("shared with me")),
     ("all", _("all")),
 )
 

circle/dashboard/views/util.py

@@ -46,6 +46,7 @@ from common.models import HumanReadableException, HumanReadableObject
 from ..models import GroupProfile, Profile
 from ..forms import TransferOwnershipForm
 
+
 logger = logging.getLogger(__name__)
 saml_available = hasattr(settings, "SAML_CONFIG")
 
@@ -169,6 +170,26 @@ class FilterMixin(object):
         return queryset
 
 
+class VmDescFilterMixin(FilterMixin):
+
+    def create_acl_queryset(self, model):
+        cleaned_data = self.search_form.cleaned_data
+        stype = cleaned_data.get('stype', 'all')
+        queryset = super(VmDescFilterMixin, self).create_acl_queryset(model)
+        user = self.request.user
+        if stype == 'owned':
+            queryset = queryset.filter(owner=user)
+        elif stype == 'shared':
+            queryset = model.get_objects_with_level(
+                'owner', self.request.user)
+            pk_list = [record.pk for record in queryset
+                       if record.object_level_set.count() > 1]
+            queryset = queryset.filter(pk__in=pk_list)
+        elif stype == 'shared_with_me':
+            queryset = queryset.exclude(owner=user)
+        return queryset
+
+
 class CheckedDetailView(LoginRequiredMixin, DetailView):
     read_level = 'user'
 

circle/dashboard/views/vm.py

@@ -56,7 +56,7 @@ from vm.models import (
 )
 from .util import (
     CheckedDetailView, AjaxOperationMixin, OperationView, AclUpdateView,
-    FormOperationMixin, FilterMixin, GraphMixin,
+    FormOperationMixin, VmDescFilterMixin, GraphMixin,
     TransferOwnershipConfirmView, TransferOwnershipView,
 )
 from ..forms import (
@@ -921,7 +921,7 @@ vm_mass_ops = OrderedDict([
 ])
 
 
-class VmList(LoginRequiredMixin, FilterMixin, ListView):
+class VmList(LoginRequiredMixin, VmDescFilterMixin, ListView):
     template_name = "dashboard/vm-list.html"
     allowed_filters = {
         'name': "name__icontains",