CODEWITHSUNDEEP
djangomodel
Yuanfeng Lin
Yuanfeng Lin

Reputation: 43

Django design model - when a data of status field changes, remaining models status change with it

Its my first time trying django as my first programming project.

I have a hierarchy structure of Company -> Business -> Outlets using foreign key.

I would like to know is there anyway to structured it in a way where the Company status is saved as inactive status, the remaining business, outlets models that will be triggered as inactive status.

from django.db import models
from django.contrib.auth.models import User


# Create your models here.
class Common_Info(models.Model):
    """(Common description)"""
    name = models.CharField(blank=True, max_length=100)
    slug = models.SlugField(unique=True, max_length=120)
    address_1 = models.CharField(max_length=50, null=True)
    address_2 = models.CharField(max_length=50, null=True)
    address_3 = models.CharField(max_length=50, null=True)
    post_code = models.CharField(max_length=6, null=False)
    registration_no. = models.CharField(max_length=15,null=False)
    gst_no. = models.CharField(max_length=15,null=True)
    telphone_no. = models.CharField(max_legth=15, null=False)
    fax_no. = models.CharField(max_legth=15, null=True)
    email_address = models.EmailField(max_length=254,null=False)
    """(Status choice)"""
    Active_Status = 1
    Inactive_Status = 0
    STATUS_CHOICES = (
        (Active_Status, 'Active'),
        (Inactive_Status, 'Inactive'),
    )
    status = models.IntegerField(choices=STATUS_CHOICES, default=Active_Status)
    create_date = models.DateTimeField(auto_now_add=True)
    create_user = models.ForeignKey(settings.AUTH_USER_MODEL)
    modified_date = models.DateTimeField(auto_now_add=True)
    modified_user = models.ForeignKey(settings.AUTH_USER_MODEL)

    class Meta:
        abstract = True

class Company(Common_Info):
    """(Company additional description)"""
    gst_no. = models.CharField(max_length=15,null=True)

    class Meta:
        verbose_name ='Company'
        verbose_name_plural = "Companies"

    def __unicode__(self):
        return u"Company"

class Business(Common_Info):
    """(Business description)"""
    parent=models.ForeignKey(Company, on_delete=models.CASCADE)
    gst_no. = models.CharField(max_length=15,null=True)

    class Meta:
        verbose_name ='Business'
        verbose_name_plural = "Businesses"

    def __unicode__(self):
        return u"Business"

class Outlet(Common_Info):
    outlet_code = models.CharField(max_length=3, unique=True)
    business_name = models.ForeignKey(Business, on_delete=models.CASCADE)

    def __unicode__(self):
        return u"Outlet"

Is there something similar to cascade on delete or other more elegant way of extending such a function across other apps in the project.

Upvotes: 2

Views: 970

Answers (1)

ruddra
ruddra

Reputation: 52018

I don't think there is any direct support in ORM. But you can override the save() method to update the related the outlets and business. You can use related objects to fetch business and outlets. Here is an example:

class Company(Common_Info):
    """(Company additional description)"""
    gst_no. = models.CharField(max_length=15,null=True)

    class Meta:
        verbose_name ='Company'
        verbose_name_plural = "Companies"

    def __unicode__(self):
        return u"Company"

    def save(self, *args, **kwargs):
        super(Company, self).save(*args, **kwargs)
        if self.status == 0:
             self.business_set.update(status=self.status)
             self.outlet_set.update(status=self.status)

Upvotes: 1

Related Questions