class AddGroundtruthRecordView(BaseAddGroundtruthRecordView):
content_type_model = get_lot_model()
form_class = GroundtruthRecordForm
def get_form_kwargs(self):
kwargs = super(AddGroundtruthRecordView, self).get_form_kwargs()
kwargs.update({ 'user': self.request.user })
return kwargs
class AddStewardNotificationView(BaseAddStewardNotificationView):
content_type_model = get_lot_model()
form_class = StewardNotificationForm
def get_success_url(self):
return reverse('steward:add_stewardnotification_success',
kwargs={
'pk': self.object.content_object.pk,
'stewardnotification_pk': self.object.pk,
})
def get_lot(self, to_address):
"""
Get the lot using the address the message was sent to, which should
include an ID.
"""
try:
lot_id = self.lot_id_pattern.match(to_address).group(1)
return get_lot_model().objects.get(pk=lot_id)
except Exception:
return None
class AddGroundtruthRecordView(BaseAddGroundtruthRecordView):
content_type_model = get_lot_model()
form_class = GroundtruthRecordForm
def get_form_kwargs(self):
kwargs = super(AddGroundtruthRecordView, self).get_form_kwargs()
kwargs.update({'user': self.request.user})
return kwargs
def get_success_url(self):
return reverse('groundtruth:add_groundtruthrecord_success',
kwargs={
'pk': self.object.content_object.pk,
'groundtruthrecord_pk': self.object.pk,
})
def fetch_owners(self, owner_type, grouped=False, visible_only=True):
"""Get owners from database"""
owners = get_owner_model().objects.filter(
owner_type=owner_type
).order_by('name')
if visible_only:
owners = [o for o in owners if get_lot_model().visible.filter(owner=o).exists()]
if grouped:
# Get all ownergroups for type
groups = get_owner_group_model().objects.filter(owner_type=owner_type)
# Remove owners in groups
grouped_owners = groups.values_list('owners', flat=True)
owners = [o for o in owners if o.pk not in grouped_owners]
# Insert into owners and sort again
owners += groups
owners = sorted(owners, key=lambda o: o.name)
return owners
class AddStewardNotificationView(BaseAddStewardNotificationView):
content_type_model = get_lot_model()
form_class = StewardNotificationForm