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