Exemplo n.º 1
0
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
Exemplo n.º 2
0
    def test_activity_instance_creation(self):
        scheduling_manager = ActivityInstanceManager()
        vendor1 = create_vendor('vendor1')
        activity1 = create_activity('activity1', vendor1)
        activity_instance1 = scheduling_manager.create_activity_instance(activity1.activity_id,
                                                                         util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Australia/Sydney')).replace(hour=3, minute=0, month=3).isoformat('T')),
                                                                         util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Australia/Sydney')).replace(hour=3, minute=45, month=3).isoformat('T')),
                                                                         12.12, 10, 0)
        self.assertIsNotNone(activity_instance1)
        activity_instance2 = scheduling_manager.create_activity_instance(activity1.activity_id,
                                                                         util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('US/Eastern')).replace(hour=6, year=2025).isoformat('T')),
                                                                         util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('US/Eastern')).replace(hour=7, year=2025).isoformat('T')),
                                                                         12.12, 1)
        self.assertIsNotNone(activity_instance2)

        self.assertIsNotNone(activity_instance1)
        activity_instance3 = scheduling_manager.create_activity_instance(activity1.activity_id,
                                                                         util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Europe/Warsaw')).replace(hour=4, month=12).isoformat('T')),
                                                                         util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Europe/Warsaw')).replace(hour=10, month=12).isoformat('T')),
                                                                         12.12, 10, 5)
        self.assertIsNotNone(activity_instance3)

        self.assertIsNotNone(ActivityInstance.objects.get(activity_instance_id=activity_instance1.activity_instance_id))
        self.assertIsNotNone(ActivityInstance.objects.get(activity_instance_id=activity_instance2.activity_instance_id))

        with self.assertRaises(InvalidActivityException):
            activity_instance4 = scheduling_manager.create_activity_instance(None,
                                                                             util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('GMT')).replace(hour=1, year=2030).isoformat('T')),
                                                                             util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('GMT')).replace(hour=15, year=2030).isoformat('T')),
                                                                             12.12, 10, 0)

        with self.assertRaises(InvalidActivityException):
            activity_instance5 = scheduling_manager.create_activity_instance("", util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Jamaica')).replace(hour=1, minute=15, month=5).isoformat('T')),
                                                                             util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Jamaica')).replace(hour=6, month=5).isoformat('T')),
                                                                             12.12, 10, 0)

        with self.assertRaises(InvalidActivityException):
            activity_instance6 = scheduling_manager.create_activity_instance("0", util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Asia/Tokyo')).replace(hour=4).isoformat('T')),
                                                                             util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Asia/Tokyo')).replace(hour=9).isoformat('T')),
                                                                             12.12, 10, 0)

        with self.assertRaises(BookingsCountInvalidException):
            activity_instance7 = scheduling_manager.create_activity_instance(activity1.activity_id,
                                                                             util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Poland')).replace(hour=13, month=7).isoformat('T')),
                                                                             util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Poland')).replace(hour=15, month=7).isoformat('T')),
                                                                             12.12, 10, 11)

        with self.assertRaises(InvalidCapacityException):
            activity_instance8 = scheduling_manager.create_activity_instance(activity1.activity_id,
                                                                             util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Pacific/Fiji')).replace(hour=17).isoformat('T')),
                                                                             util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Pacific/Fiji')).replace(hour=19).isoformat('T')),
                                                                             12.12, 0)