class ActivityInstanceView(View):
def __init__(self):
self.activity_instance_manager = ActivityInstanceManager()
def get(self, request, activity_instance_id):
activity_instance = self.activity_instance_manager.retrieve_activity_instance(activity_instance_id)
if activity_instance is None:
raise Http404
else:
activity_instance = convert_activity_instance_timestamp_to_iso(activity_instance)
return HttpResponse(util.obj_to_json(activity_instance), content_type='application/json')
def put(self, request, activity_instance_id):
attributes_dictionary = ast.literal_eval(request.read().decode('utf-8'))
if attributes_dictionary.get('start_time') is not None:
attributes_dictionary['start_time'] = util.iso8601_to_utc_timestamp(attributes_dictionary['start_time'])
if attributes_dictionary.get('end_time') is not None:
attributes_dictionary['end_time'] = util.iso8601_to_utc_timestamp(attributes_dictionary['end_time'])
activity_instance = self.activity_instance_manager.update_activity_instance(activity_instance_id,
**attributes_dictionary)
if activity_instance is None:
raise Http404
else:
activity_instance = convert_activity_instance_timestamp_to_iso(activity_instance)
return HttpResponse(util.obj_to_json(activity_instance), content_type='application/json')
def post(self, request, activity_instance_id=None):
activity_id = request.POST.get('activity_id')
start_time = util.iso8601_to_utc_timestamp(request.POST.get('start_time'))
end_time = util.iso8601_to_utc_timestamp(request.POST.get('end_time'))
price = float(request.POST.get('price'))
capacity = int(request.POST.get('capacity'))
bookings = request.POST.get('bookings')
if bookings is None:
activity_instance = self.activity_instance_manager.create_activity_instance(activity_id=activity_id,
start_time=start_time,
end_time=end_time,
price=price,
capacity=capacity)
else:
activity_instance = self.activity_instance_manager.create_activity_instance(activity_id=activity_id,
start_time=start_time,
end_time=end_time,
price=price,
capacity=capacity,
bookings=int(bookings))
activity_instance = convert_activity_instance_timestamp_to_iso(activity_instance)
return HttpResponse(util.obj_to_json(activity_instance), content_type='application/json')
def delete(self, request, activity_instance_id):
if self.activity_instance_manager.delete_activity_instance(activity_instance_id):
return HttpResponse()
else:
raise Http404
def test_activity_instance_deletion(self):
scheduling_manager = ActivityInstanceManager()
vendor1 = create_vendor('vendor1')
activity1 = create_activity('activity1', vendor1)
activity2 = create_activity('activity2', vendor1)
activity3 = create_activity('activity3', vendor1)
activity1_instance1 = ActivityInstance(activity=activity1,
start_time=util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Australia/Sydney')).replace(hour=12).isoformat('T')),
end_time=util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Australia/Sydney')).replace(hour=14).isoformat('T')),
price=12.12, capacity=10, bookings=0)
activity1_instance1.save()
activity1_instance2 = ActivityInstance(activity=activity1,
start_time=util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('US/Eastern')).replace(hour=13, month=11).isoformat('T')),
end_time=util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('US/Eastern')).replace(hour=17, month=11).isoformat('T')),
price=12.12, capacity=10, bookings=0)
activity1_instance2.save()
activity2_instance1 = ActivityInstance(activity=activity2,
start_time=util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Europe/Warsaw')).replace(hour=11).isoformat('T')),
end_time=util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Europe/Warsaw')).replace(hour=14).isoformat('T')),
price=12.12, capacity=10, bookings=0)
activity2_instance1.save()
activity2_instance2 = ActivityInstance(activity=activity2,
start_time=util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Canada/Mountain')).replace(hour=1, month=4).isoformat('T')),
end_time=util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Canada/Mountain')).replace(hour=2, month=4).isoformat('T')),
price=12.12, capacity=10, bookings=0)
activity2_instance2.save()
activity3_instance1 = ActivityInstance(activity=activity3,
start_time=util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Europe/Berlin')).replace(hour=1, minute=0).isoformat('T')),
end_time=util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Europe/Berlin')).replace(hour=1, minute=30).isoformat('T')),
price=12.12, capacity=10, bookings=0)
activity3_instance1.save()
activity3_instance2 = ActivityInstance(activity=activity3,
start_time=util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Brazil/West')).replace(hour=15, minute=0, month=2).isoformat('T')),
end_time=util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Brazil/West')).replace(hour=15, minute=45, month=2).isoformat('T')),
price=12.12, capacity=10, bookings=0)
activity3_instance2.save()
self.assertTrue(scheduling_manager.delete_activity_instance(activity1_instance1.activity_instance_id))
self.assertEqual(ActivityInstance.objects.filter(
activity_instance_id=activity1_instance1.activity_instance_id).count(), 0)
self.assertTrue(scheduling_manager.delete_activity_instance(activity1_instance2.activity_instance_id))
self.assertEqual(ActivityInstance.objects.filter(
activity_instance_id=activity1_instance2.activity_instance_id).count(), 0)
delete_activity(activity2)
self.assertEqual(ActivityInstance.objects.filter(
activity_instance_id=activity2_instance1.activity_instance_id).count(), 0)
self.assertEqual(ActivityInstance.objects.filter(
activity_instance_id=activity2_instance2.activity_instance_id).count(), 0)
delete_vendor(vendor1)
self.assertEqual(Activity.objects.filter(activity_id=activity1.activity_id).count(), 0)
self.assertEqual(Activity.objects.filter(activity_id=activity2.activity_id).count(), 0)
self.assertEqual(Activity.objects.filter(activity_id=activity3.activity_id).count(), 0)
self.assertEqual(ActivityInstance.objects.filter(
activity_instance_id=activity3_instance1.activity_instance_id).count(), 0)
self.assertEqual(ActivityInstance.objects.filter(
activity_instance_id=activity3_instance2.activity_instance_id).count(), 0)
self.assertFalse(scheduling_manager.delete_activity_instance(activity1_instance1.activity_instance_id))
self.assertFalse(scheduling_manager.delete_activity_instance(activity1_instance2.activity_instance_id))
