from rest_framework import viewsets
from . import serializers


class BaseViewSet(viewsets.ModelViewSet):
    permission_classes = []  # 允许所有访问


class ProductViewSet(BaseViewSet):
    queryset = serializers.basic_models.Product.objects
    serializer_class = serializers.ProductSerializer


class WareHouseViewSet(BaseViewSet):
    queryset = serializers.basic_models.WareHouse.objects
    serializer_class = serializers.WareHouseSerializer


class ProductCategoryViewSet(BaseViewSet):
    queryset = serializers.basic_models.ProductCategory.objects
    serializer_class = serializers.ProductCategorySerializer


class SupplierViewSet(BaseViewSet):
    queryset = serializers.basic_models.Supplier.objects
    serializer_class = serializers.SupplierSerializer


class EmployeeViewSet(BaseViewSet):
    queryset = serializers.basic_models.Employee.objects
    serializer_class = serializers.EmployeeSerializer


class CustomerViewSet(BaseViewSet):
    queryset = serializers.basic_models.Customer.objects
    serializer_class = serializers.CustomerSerializer


class VehicleTypeViewSet(BaseViewSet):
    queryset = serializers.basic_models.VehicleType.objects
    serializer_class = serializers.VehicleTypeSerializer


class BankAccountViewSet(BaseViewSet):
    queryset = serializers.basic_models.BankAccount.objects
    serializer_class = serializers.BankAccountSerializer


class DeviceInfoViewSet(BaseViewSet):
    queryset = serializers.basic_models.DeviceInfo.objects
    serializer_class = serializers.DeviceInfoSerializer


class VehicleTransportRecordViewSet(BaseViewSet):
    queryset = serializers.basic_models.VehicleTransportRecord.objects
    serializer_class = serializers.VehicleTransportRecordSerializer