day75作业

# settings.py配置
TATIC_URL = '/static/'

MEDIA_URL = '/media/'

MEDIA_ROOT = os.path.join(BASE_DIR, 'media')

REST_FRAMEWORK = {
    'EXCEPTION_HANDLER': 'api.exception.exception_handler'
}

BASE_URL = 'http://127.0.0.1:8000'

# 主路由
from django.conf.urls import url, include
from django.contrib import admin
from django.views.static import serve
from django.conf import settings


urlpatterns = [
    url(r'^admin/', admin.site.urls),
    url(r'^api/', include('api.urls')),

    url(r'^media/(?P<path>.*)', serve, {'document_root': settings.MEDIA_URL})
]

# 分路由
from django.conf.urls import url, include
from . import views

from rest_framework.routers import SimpleRouter
router = SimpleRouter()
router.register('v8/cars', views.CarModelViewSet, basename='car')

urlpatterns = [
    url(r'', include(router.urls))
]

# models模型
from django.db import models
from django.conf import settings

class BaseModel(models.Model):
    is_delete = models.BooleanField(default=False)
    created_time = models.DateTimeField(auto_now_add=True)

    class Meta:
        abstract = True


class Car(BaseModel):
    name = models.CharField(max_length=64)
    price = models.DecimalField(max_digits=10, decimal_places=2)
    img = models.ImageField(upload_to='img', default='img/default.jpg')

    brand = models.ForeignKey(to='Brand', on_delete=models.SET_NULL, null=True, db_constraint=False,
                              related_name='cars')
    sponsors = models.ManyToManyField(to='Sponsor', db_constraint=False, related_name='cars')

    def __str__(self):
        return self.name

    @property
    def icon(self):
        return '%s%s%s' % (settings.BASE_URL, settings.MEDIA_URL, self.img)

    @property
    def brand_name(self):
        return self.brand.name

    @property
    def sponsor_list(self):
        sponsor_list_tmp = []
        for sponsor in self.sponsors.all():
            sponsor_dic = {
                'name': sponsor.name
            }
            try:
                sponsor_dic['phone'] = sponsor.detail.phone
            except:
                sponsor_dic['phone'] = ''
            sponsor_list_tmp.append(sponsor_dic)

        return sponsor_list_tmp


class Brand(models.Model):
    name = models.CharField(max_length=64)

    def __str__(self):
        return self.name


class Sponsor(models.Model):
    name = models.CharField(max_length=64)

    def __str__(self):
        return self.name


class SponsorDetail(models.Model):
    phone = models.CharField(max_length=64)
    sponsor = models.OneToOneField(to=Sponsor, related_name='detail', db_constraint=False, on_delete=models.CASCADE,
                                   null=True)

    def __str__(self):
        try:  # 连表可能会出现问题，所以要异常处理
            return self.sponsor.name + '的详情'
        except:
            return super().__str__()

# 序列化类
from rest_framework import serializers
from . import models


class CarModelSerializer(serializers.ModelSerializer):
    class Meta:
        model = models.Car
        fields = ['name', 'price', 'brand', 'icon', 'sponsors', 'brand_name', 'sponsor_list']
        extra_kwargs = {
            'brand': {'write_only': True},
            'sponsors': {'write_only': True}
        }

# 视图类

class CarModelViewSet(ModelViewSet):
    queryset = models.Car.objects.filter(is_delete=False).all()
    serializer_class = serializer.CarModelSerializer

    # 群整体改，群局部改，全删三个接口可以独立成三个方法
    def many_update(self, request, *args, **kwargs):
        request_data = request.data
        try:
            pks = []
            for dic in request_data:
                pk = dic.pop('pk')
                pks.append(pk)
            car_query = models.Car.objects.filter(is_delete=False, pk__in=pks).all()
            if len(pks) != len(car_query):
                raise Exception('pk对应的数据不存在')

        except Exception as e:
            return Response({'detail': '%s' % e}, status=400)
        car_ser = serializer.CarModelSerializer(instance=car_query, data=request_data, many=True)
        car_ser.is_valid(raise_exception=True)
        car_list = car_ser.save()
        return APIResponse(results=serializer.CarModelSerializer(instance=car_list, many=True).data)

    def many_partial_update(self, request, *args, **kwargs):
        request_data = request.data
        try:
            pks = []
            for dic in request_data:
                pk = dic.pop('pk')
                pks.append(pk)
            car_query = models.Car.objects.filter(is_delete=False, pk__in=pks).all()
            if len(pks) != len(car_query):
                raise Exception('pk对应的数据不存在')

        except Exception as e:
            return Response({'detail': '%s' % e}, status=400)
        car_ser = serializer.CarModelSerializer(instance=car_query, data=request_data, partial=True, many=True)
        car_ser.is_valid(raise_exception=True)
        car_list = car_ser.save()
        return APIResponse(results=serializer.CarModelSerializer(instance=car_list, many=True).data)

    def many_destroy(self, request, *args, **kwargs):
        pks = request.data
        try:
            rows = models.Car.objects.filter(is_delete=False, pk__in=pks).update(is_delete=True)
        except:
            return APIResponse(1, '数据有误')

        if rows:
            return APIResponse(0, '删除成功')
        return APIResponse(1, '删除失败')

    # 群增与单增必须公用一个接口，都要走create方法 - 重写create方法，用逻辑进行拆分
    def create(self, request, *args, **kwargs):
        if isinstance(request.data, list):
            car_ser = serializer.CarModelSerializer(data=request.data, many=True)
            car_ser.is_valid(raise_exception=True)
            car_obj = car_ser.save()
            return APIResponse(results=serializer.CarModelSerializer(car_obj, many=True).data)

        return super().create(request, *args, **kwargs)

    # destroy方法是完成is_delete字段值修改 - 重写destroy方法，自定义实现体
    def destroy(self, request, *args, **kwargs):
        car_obj = self.get_object()
        car_obj.is_delete = True
        car_obj.save()
        return APIResponse(msg='删除成功')

    # 让群查有状态码和状态信息 - 重写list方法
    def list(self, request, *args, **kwargs):
        response = super().list(request, *args, **kwargs)
        return APIResponse(results=response.data)

    # 重写retrieve方法
    def retrieve(self, request, *args, **kwargs):
        response = super().retrieve(request, *args, **kwargs)
        return APIResponse(results=response.data)