def create(self, validated_data): guests = [EventGuest(**item) for item in validated_data] self.validate_same_invitation(guests) inviteNumber = EventGuest.nextFreeInvitation(guests[0].event if len(guests) > 0 else None) for guest in guests: guest.invitation = inviteNumber return EventGuest.objects.bulk_create(guests)
def create(self, validated_data):
guests = [EventGuest(**item) for item in validated_data]
self.validate_same_invitation(guests)
inviteNumber = EventGuest.nextFreeInvitation(
guests[0].event if len(guests) > 0 else None)
for guest in guests:
guest.invitation = inviteNumber
return EventGuest.objects.bulk_create(guests)
def test_nextFreeInvite_incriments(self):
ev = Event(name='Test Event', event_date=datetime.date.today())
ev.save()
defaults = {'event': ev, 'first': 'FirstName', 'last': 'Last Name'}
latest = 0
for x in range(10):
guest = EventGuest(invitation=EventGuest.nextFreeInvitation(ev), **defaults)
guest.save()
self.assertTrue(guest.invitation > latest)
latest = guest.invitation
def test_nextFreeInvite_incriments(self):
ev = Event(name='Test Event', event_date=datetime.date.today())
ev.save()
defaults = {'event': ev, 'first': 'FirstName', 'last': 'Last Name'}
latest = 0
for x in range(10):
guest = EventGuest(invitation=EventGuest.nextFreeInvitation(ev),
**defaults)
guest.save()
self.assertTrue(guest.invitation > latest)
latest = guest.invitation
def many_init(cls, *args, **kwargs):
"""
Here's where we'll validate that all the invitations are the same,
or assign them.
When passing many=True with a data kwarg, we need to make sure the representations
are not orphaned from their event. Representations must either each have a dict key
"event" which corresponds to the integer primary key, or you must pass the event as
a keyword argument to the serializer along with many=True.
"""
if 'data' in kwargs:
#validate data is a list.
data = kwargs.pop('data')
assert type(data) == list, (
'''Invitation Serializer with many=True and non-empty data
expects a list in the data representation, not a dict or other
type.''')
eventNumber = kwargs.pop('event', None)
if eventNumber is None:
try:
eventNumber = data[0]['event']
except KeyError:
raise NoEventError(
'''No event information associated with this list of guests.
Try passing `event` keyword arg to the serializer or guest representations
with event attributes''')
#build invitations list to analyze
invitations = [obj.pop('invitation', None) for obj in data]
for inx, val in enumerate(invitations):
if type(val) == str: #some data cleaning for types.
val = val.strip()
if val == '':
invitations[inx] = None
else:
try:
intval = int(val)
invitations[inx] = intval
except ValueError:
pass
integerInvitations = [x for x in invitations if type(x) == int]
if integerInvitations: #isn't epmty
allsame = map(lambda x: x == integerInvitations[0],
integerInvitations)
if not all(allsame):
raise MixedInvitationError(
'All Numbered invitations must be same.')
numberToAssign = integerInvitations[0]
else: #usually means invitations == [None,]*len(data):
numberToAssign = EventGuest.nextFreeInvitation(eventNumber)
for representation in data:
representation['invitation'] = numberToAssign
# put our work back in the keyword args dict, which we'll use below.
kwargs['data'] = data
else:
instance_queryset = [
args[0],
] if isinstance(args[0],
EventGuest) else args[0] #per the serializer API.
allsame = map(
lambda x: x.invitation == instance_queryset[0].invitation,
instance_queryset)
if not all(allsame):
raise MixedInvitationError
# finally make the serializer, knowing the above is done.
child = cls(*args, **kwargs)
list_kwargs = {
'child': child,
}
list_kwargs.update({
key: value
for key, value in kwargs.items()
if key in serializers.LIST_SERIALIZER_KWARGS
})
return InvitationListSerializer(*args, **list_kwargs)
def test_nextFreeInvitation_method_raises_exception_on_noEvent(self): with self.assertRaises(NoEventError): EventGuest.nextFreeInvitation(None)
def many_init(cls, *args, **kwargs):
"""
Here's where we'll validate that all the invitations are the same,
or assign them.
When passing many=True with a data kwarg, we need to make sure the representations
are not orphaned from their event. Representations must either each have a dict key
"event" which corresponds to the integer primary key, or you must pass the event as
a keyword argument to the serializer along with many=True.
"""
if 'data' in kwargs:
#validate data is a list.
data = kwargs.pop('data')
assert type(data) == list, (
'''Invitation Serializer with many=True and non-empty data
expects a list in the data representation, not a dict or other
type.'''
)
eventNumber = kwargs.pop('event', None)
if eventNumber is None:
try:
eventNumber = data[0]['event']
except KeyError:
raise NoEventError(
'''No event information associated with this list of guests.
Try passing `event` keyword arg to the serializer or guest representations
with event attributes'''
)
#build invitations list to analyze
invitations = [obj.pop('invitation', None) for obj in data]
for inx, val in enumerate(invitations):
if type(val) == str: #some data cleaning for types.
val = val.strip()
if val == '':
invitations[inx] = None
else:
try:
intval = int(val)
invitations[inx] = intval
except ValueError:
pass
integerInvitations = [x for x in invitations if type(x)==int]
if integerInvitations: #isn't epmty
allsame = map(lambda x: x==integerInvitations[0], integerInvitations)
if not all(allsame):
raise MixedInvitationError('All Numbered invitations must be same.')
numberToAssign = integerInvitations[0]
else: #usually means invitations == [None,]*len(data):
numberToAssign = EventGuest.nextFreeInvitation(eventNumber)
for representation in data:
representation['invitation'] = numberToAssign
# put our work back in the keyword args dict, which we'll use below.
kwargs['data'] = data
else:
instance_queryset = [args[0],] if isinstance(args[0], EventGuest) else args[0] #per the serializer API.
allsame = map(lambda x: x.invitation==instance_queryset[0].invitation, instance_queryset)
if not all(allsame):
raise MixedInvitationError
# finally make the serializer, knowing the above is done.
child = cls(*args, **kwargs)
list_kwargs = {'child': child,}
list_kwargs.update({
key: value for key, value in kwargs.items()
if key in serializers.LIST_SERIALIZER_KWARGS
})
return InvitationListSerializer(*args, **list_kwargs)
def test_nextFreeInvitation_method_raises_exception_on_noEvent(self):
with self.assertRaises(NoEventError):
EventGuest.nextFreeInvitation(None)