Пример #1
0
    def _create_or_update(self, request, year_year=None, playa_event_id=None):
        user = request.user
        if not user.get_profile().api_allowed:
            return rc_response(request, rc.BAD_REQUEST, 'User not permitted to use the API')

        method = request.method

        if method == "PUT":
            data = request.PUT.copy()
        elif method == "POST":
            data = request.POST.copy()
        else:
            return rc_response(request, rc.BAD_REQUEST, 'Bad request method: %s' % method)

        if year_year and 'year' not in data:
            data['year'] = year_year

        if not (data and user):
            log.debug('Bad request: data=%s, user=%s', data, user)
            return rc_response(request, rc.BAD_REQUEST, 'Missing critical information')

        log.debug('data for create_or_update: %s', data)

        if playa_event_id:
            try:
                obj = PlayaEvent.objects.get(pk=playa_event_id)
                log.debug('got playaevent #%i for update', obj.id)

            except PlayaEvent.DoesNotExist:
                return rc_response(request, rc.NOT_HERE, 'Event not found #%s' % playa_event_id)

            # no updating the year!
            if 'year' in data:
                del data['year']

        else:
            log.debug('creating new event')
            obj = PlayaEvent()
            obj.creator = user
            if not 'event_type' in data:
                event_type = EventType.objects.get(pk=1)
            else:
                try:
                    event_type = EventType.objects.get(abbr=data['event_type'])
                except EventType.DoesNotExist:
                    return rc_response(request, rc.NOT_HERE, 'No such EventType: %s' % data['event_type'])

            obj.event_type = event_type

        # get rid of illegal-to-update attributes
        for key in ('id','pk'):
            if key in data:
                del data[key]

        # now loop through the data, updating as needed
        if 'year' in data:
            try:
                year = Year.objects.get(year=data['year'])
            except Year.DoesNotExist:
                return rc_response(request, rc.NOT_HERE, 'No such year: %s' % data['year'])

            obj.year = year

        # text fields
        for key in ('print_description', 'url', 'contact_email', 'other_location', 'slug'):
            if key in data:
                val = data[key]
                log.debug('setting %s=%s', key, val)
                setattr(obj, key, val)

        # moderation
        if 'moderation' in data:
            modkey = data['moderation'].upper()
            if modkey in ('U','A','R'):
                log.debug('setting moderation=%s', modkey)
                obj.moderation = modkey

        if 'hosted_by_camp' in data:
            key = data['hosted_by_camp']
            try:
                camp = ThemeCamp.objects.get(pk=key)
                log.debug('located at camp: %s', camp)
                obj.hosted_by_camp = camp
            except ThemeCamp.DoesNotExist:
                return rc_response(request, rc.NOT_HERE, 'No such camp: %s' % key)

        if 'located_at_art' in data:
            key = data['located_at_art']
            try:
                art = ArtInstallation.objects.get(pk=key)
                log.debug('located at art: %s', art)
                obj.located_at_art = art
            except ArtInstallation.DoesNotExist:
                return rc_response(request, rc.NOT_HERE, 'No such art: %s' % key)

        # booleans
        for key in ('check_location', 'all_day', 'list_online', 'list_contact_online'):
            if key in data:
                val = data[key].upper()
                val = val in ('1', 'T', 'Y', 'YES', 'TRUE', 'ON')
                log.debug('setting %s=%s', key, val)
                setattr(obj, key, val)

        obj.save()

        if method == 'PUT':
            response = rc.ALL_OK

        else:
            response = rc.CREATED

        response.content = json.dumps({'pk' : obj.id})
        return response