from django.contrib.auth.models import User
from django import forms
+from django.db.models import Q
from django.db.utils import ProgrammingError
from patchwork.models import Patch, State, Bundle, UserProfile
class DelegateField(forms.ModelChoiceField):
- def __init__(self, *args, **kwargs):
- queryset = User.objects
+ def __init__(self, project, instance=None, *args, **kwargs):
+ q = Q(profile__in=UserProfile.objects
+ .filter(maintainer_projects=project)
+ .values('pk').query)
+ if instance and instance.delegate:
+ q = q | Q(username=instance.delegate)
+ queryset = User.objects.complex_filter(q)
super(DelegateField, self).__init__(queryset, *args, **kwargs)
if not project:
raise Exception("meep")
super(PatchForm, self).__init__(instance=instance, *args, **kwargs)
- self.fields['delegate'] = DelegateField(required=False)
+ self.fields['delegate'] = DelegateField(project, instance,
+ required=False)
class Meta:
model = Patch
def __init__(self, project, *args, **kwargs):
super(MultiplePatchForm, self).__init__(*args, **kwargs)
- self.fields['delegate'] = OptionalDelegateField(required=False)
+ self.fields['delegate'] = OptionalDelegateField(project=project,
+ required=False)
def save(self, instance, commit=True):
opts = instance.__class__._meta