import logging from django.core.exceptions import ValidationError logger = logging.getLogger("bornhack.%s" % __name__) def check_speaker_event_camp_consistency(sender, instance, **kwargs): if kwargs["action"] == "pre_add": from program.models import Speaker, Event if isinstance(instance, Event): # loop over speakers being added to this event for pk in kwargs["pk_set"]: # check if this speaker belongs to a different Camp than the event does speaker = Speaker.objects.get(id=pk) if speaker.camp != instance.camp: raise ValidationError( { "speakers": "The speaker (%s) belongs to a different camp (%s) than the event does (%s)" % (speaker, speaker.camp, instance.camp) } ) elif isinstance(instance, Speaker): # loop over events being added to this speaker for pk in kwargs["pk_set"]: # check if this event belongs to a different Camp than the speaker does event = Event.objects.get(id=pk) if event.camp != instance.camp: raise ValidationError( { "events": "The event (%s) belongs to a different camp (%s) than the event does (%s)" % (event, event.camp, instance.camp) } ) def check_speaker_camp_change(sender, instance, **kwargs): if instance.pk: for event in instance.events.all(): if event.camp != instance.camp: raise ValidationError( { "camp": "You cannot change the camp a speaker belongs to if the speaker is associated with one or more events." } )