Django представление на основе классов, сохранение в другой модели после CreateView

У меня есть представление создания (Loan_assetCreateView(generic.CreateView)), где я сохраняю, если актив будет предоставлен во временное пользование и когда он будет возвращен в модели под названием Loan_asset(models.Model). Затем у меня есть актив в другой модели Asset(model.Model). Я хочу, чтобы после сохранения данных в модели Loan_assetCreateView(generic.CreateView) значение в Asset.is_loaned было установлено на True. Как я могу это сделать?

Мой models.py:

class Asset(models.Model):

    # Relationships
    room = models.ForeignKey("asset_app.Room", on_delete=models.SET_NULL, blank=True, null=True)
    model_hardware = models.ForeignKey("asset_app.Model_hardware", on_delete=models.SET_NULL, blank=True, null=True)

    # Fields
    name = models.CharField(max_length=30)
    serial = models.CharField(max_length=30, unique=True, blank=True, null=True, default=None)
    mac_address = models.CharField(max_length=30, null=True, blank=True)
    purchased_date = models.DateField(null=True, blank=True)
    may_be_loaned = models.BooleanField(default=False, blank=True, null=True)
    is_loaned = models.BooleanField(default=False, blank=True, null=True)
    missing = models.BooleanField(default=False, blank=True, null=True)
    notes = HTMLField(default="")
    ip = models.CharField(max_length=90, null=True, blank=True)
    created = models.DateTimeField(auto_now_add=True, editable=False)
    last_updated = models.DateTimeField(auto_now=True, editable=False)

class Loan_asset(models.Model):

    # Relationships
    asset = models.ForeignKey("asset_app.Asset", on_delete=models.SET_NULL, blank=True, null=True)
    loaner_type = models.ForeignKey("asset_app.Loaner_type", on_delete=models.SET_NULL, blank=True, null=True)
    location = models.ForeignKey("asset_app.Locations", on_delete=models.SET_NULL, blank=True, null=True)

    # Fields
    loaner_name = models.CharField(max_length=60)
    loaner_address = models.TextField(max_length=100, null=True, blank=True)
    loaner_telephone_number = models.CharField(max_length=30)
    loaner_email = models.EmailField()
    loaner_quicklink = models.URLField(null=True, blank=True)
    loan_date = models.DateField()
    return_date = models.DateField()
    notes = HTMLField(default="")
    returned = models.BooleanField(default=False, blank=True, null=True)
    created = models.DateTimeField(auto_now_add=True, editable=False)
    last_updated = models.DateTimeField(auto_now=True, editable=False)
    class Meta:
        pass

    def __str__(self):
        return str(self.loaner_name)

    def get_absolute_url(self):
        return reverse("asset_app_loan_asset_detail", args=(self.pk,))

    def get_update_url(self):
        return reverse("asset_app_loan_asset_update", args=(self.pk,))

my urls.py

`path("asset_app/loan_asset/create/", views.Loan_assetCreateView.as_view(), name="asset_app_loan_asset_create")`,

my views.py

class Loan_assetCreateView(generic.CreateView):
    model = models.Loan_asset
    form_class = forms.Loan_assetForm

Вы можете переопределить метод post в своем Loan_assetCreateView.

class Loan_assetCreateView(generic.CreateView):
    model = models.Loan_asset
    form_class = forms.Loan_assetForm
    
    def post(request, *args, **kwargs):
        response = super().post(request, *args. **kwargs)
        # Do your thing
        return response
        

Вот некоторые варианты:

  • переопределите метод form_valid, который вызывается в реализации метода post, чтобы после проверки формы (сохранения экземпляра модели) вы могли установить флаг через внешний ключ/создавая экземпляр Asset:
...

def form_valid(self, form):
    self.object = form.save()

    if self.object.asset:
        self.object.asset.is_loaned = True
    else:
        self.object.asset = Asset.objects.create(is_loaned=True) 
    return HttpResponseRedirect(self.get_success_url())
@receiver(post_save, sender=Loan_asset)
def create_transaction(sender, instance, created, **kwargs):
    if created:
        Asset.objects.create(is_loaned=True)
Вернуться на верх