2017-08-16 41 views
0

我已經通過使用抽象用戶定製了內置用戶模型。我製作了另一款與用戶模型具有一對一關係的模型車輛,類似於每個用戶的車輛。但它存儲每個user_id的單個車輛。我需要它爲每個用戶存儲多輛車。我的代碼是下面給出 -需要在django API中存儲每個user_id更多的車輛

Models.py

class User(AbstractUser): 
    phone_number = models.IntegerField(null=True) 
    otp = models.IntegerField(null=True) 


class Vehicle(models.Model): 
    user = models.OneToOneField(User) 
    maker = models.CharField(max_length=500, blank=True) 
    car_model = models.CharField(max_length=30, blank=True) 

Serializer.py

class VehicleSerializer(serializers.ModelSerializer): 
user_id = serializers.IntegerField() 
class Meta: 
    model = Vehicle 
    fields = ('maker', 'car_model', 'user_id')                  

def create(self, validated_data): 
    maker = validated_data.pop('make', None) 
    car_model = validated_data.get('car_model', None) 
    user_id = validated_data.get('user_id', None) 
    instance = self.Meta.model(**validated_data) 

    if user_id is not None: 
     instance.user_id = user_id 
     instance.save() 
    if car_model is not None: 
     instance.car_model = car_model 
     instance.save() 
    return instance 

Views.py

@api_view(['POST']) 
def vehicle_add(request): 
car_model = request.data.get('car_model') 
user_id = request.user.id 
serializer = VehicleSerializer(data={'user_id': user_id, 'car_model': car_model}) 
if serializer.is_valid(): 
    serializer.save() 
    message = "i suceeded" 
    return Response(message, status=status.HTTP_201_CREATED) 
else: 
    return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) 

我只需要知道如何存儲多部車輛在單個user_id下。謝謝

回答