コード例 #1
0
    def post(self, request, *args, **kwargs):
        user_url = request.data.get("user", None)
        if user_url:
            try:
                path = urlparse.urlparse(user_url).path
                match = resolve(path)
                user_pk = match.kwargs.get("pk")
            except (Resolver404, User.DoesNotExist) as e:
                raise CustomAPIException(
                    "You need to send the url of a User to create its UserPreference.",
                    status=status.HTTP_400_BAD_REQUEST,
                )
        else:
            raise CustomAPIException(
                "You need to identify a User to create its UserPreference.",
                status=status.HTTP_400_BAD_REQUEST,
            )

        userpreference_obj = UserPreference(user_id=user_pk,
                                            send_email=request.data.get(
                                                "send_email", False))
        userpreference_obj.save()
        d = UserPreferenceSerializer(userpreference_obj,
                                     context={
                                         "request": request
                                     }).data
        return Response(d, status=status.HTTP_201_CREATED)
コード例 #2
0
ファイル: test_userpreference.py プロジェクト: DDMAL/Rodan
    def test_delete(self):
        userpreference = UserPreference(**self.test_user_preference_data)
        userpreference.save()

        retr_userpreference = UserPreference.objects.get(user_id=self.test_user.id)
        retr_userpreference.delete()

        retr_userpreference = UserPreference.objects.filter(user_id=self.test_user.id)
        self.assertFalse(retr_userpreference.exists())
コード例 #3
0
    def test_delete(self):
        userpreference = UserPreference(**self.test_user_preference_data)
        userpreference.save()

        retr_userpreference = UserPreference.objects.get(
            user_id=self.test_user.id)
        retr_userpreference.delete()

        retr_userpreference = UserPreference.objects.filter(
            user_id=self.test_user.id)
        self.assertFalse(retr_userpreference.exists())
コード例 #4
0
ファイル: userpreference.py プロジェクト: DDMAL/Rodan
    def post(self, request, *args, **kwargs):
        user_url = request.data.get('user', None)
        if user_url:
            try:
                path = urlparse.urlparse(user_url).path
                match = resolve(path)
                user_pk = match.kwargs.get('pk')
            except (Resolver404, User.DoesNotExist) as e:
                raise CustomAPIException("You need to send the url of a User to create its UserPreference.", status=status.HTTP_400_BAD_REQUEST)
        else:
            raise CustomAPIException("You need to identify a User to create its UserPreference.", status=status.HTTP_400_BAD_REQUEST)

        userpreference_obj = UserPreference(user_id=user_pk, send_email=request.data.get('send_email', False))
        userpreference_obj.save()
        d = UserPreferenceSerializer(userpreference_obj, context={'request': request}).data
        return Response(d, status=status.HTTP_201_CREATED)