将选择限制为用户属性

问题描述

我有一系列通过外键连接的模型。

通用设置应用

class Company(HistoryBaseModel):
    code = models.CharField(max_length=3)
    name = models.CharField(max_length=50)
    user_company = models.ManyToManyField(User,through="UserCompany",through_fields=("company","user"))
    

    class Meta(BaseModel.Meta):
        verbose_name = "Company"
        verbose_name_plural = "Companies"
        ordering = ('code',)

    def __str__(self):
        return f'{self.name} ({self.code})'


class UserCompany(HistoryBaseModel):

    company = models.ForeignKey(Company,on_delete=models.PROTECT,verbose_name="Company",related_name="%(app_label)s_%(class)s_company")
    user = models.ForeignKey(User,verbose_name="User",related_name="%(app_label)s_%(class)s_user")
    

    class Meta(BaseModel.Meta):
        verbose_name = "User's Company"
        verbose_name_plural = "User's Companies"
        constraints.UniqueConstraint(fields=['company','user'],name='unique_user_company')
        

    def __str__(self):
        return f'{self.user} ({self.company})'


class MachineTypes(HistoryBaseModel):
    manufacturer = models.CharField(max_length=50)
    family = models.CharField(max_length=20)
    type = models.CharField(max_length=10)
    series = models.CharField(max_length=10)
    company_machine_types = models.ManyToManyField(Company,through = 'CompanyMachineTypes',through_fields=('type_series','company'))
    

    class Meta(BaseModel.Meta):
        verbose_name = "machine Type"
        verbose_name_plural = "machine Types"
        constraints.UniqueConstraint(fields=['family','type','series'],name='unique_type_series')
        ordering = ('manufacturer','series')

    def __str__(self):
        return f'{self.manufacturer} {self.type}-{self.series}'

    # Unique: Type,Series  

class CompanyMachineTypes(HistoryBaseModel):
    company = models.ForeignKey(Company,related_name="%(app_label)s_%(class)s_company")
    type_series = models.ForeignKey(MachineTypes,related_name="%(app_label)s_%(class)s_type_series",verbose_name='Type/Series')
    

    class Meta(BaseModel.Meta):
        verbose_name = "Company machine Types"
        verbose_name_plural = "Company machine Types"
        constraints.UniqueConstraint(fields=['company','machineTypes'],name='unique_company_typeSeries')

    def __str__(self):
        return f'{self.company} ({self.type_series})'
    

class Machine(HistoryBaseModel):
    type = models.ForeignKey(MachineTypes,related_name="%(app_label)s_%(class)s_type")
    company = models.ForeignKey(Company,related_name="%(app_label)s_%(class)s_company")
    registration = models.CharField(max_length=10,unique=True)
    sn= models.CharField(max_length=10,unique=True)
    
    

    class Meta(BaseModel.Meta):
        verbose_name = "machine"
        verbose_name_plural = "machine"
        ordering = ('company','sn')

    def __str__(self):
        return f'{self.company} {self.registration} ({self.sn})'

电源APP

class PowerSources(HistoryBaseModel):
    
    Machine_type = models.ForeignKey("genetics.MachineTypes",on_delete=models.PROTECT)
    code = models.CharField(max_length=10,unique=True)
    description = models.CharField(max_length=200,blank=True)
    type = models.ForeignKey("genetics.SystemVariables",verbose_name= ("Power Source Type"),limit_choices_to={"variable_type": 2})
    current_type = models.BooleanField(default=True,choices= ((True,"AC"),(False,"DC")))
    voltage = models.DecimalField(max_digits=3,decimal_places=2)
    limit = models.DecimalField(max_digits=7,decimal_places=2)
    phases = models.PositiveSmallIntegerField(help_text="If Current Type is set as DC,this field will default to 0")

    def save(self):
        if self.type == False:
            self.phases = 0
        super().save()

class MachinePowerSources(PowerSources):

    Machine_type = None
    Machine = models.ForeignKey("genetics.Machine",verbose_name= ("Machine"),limit_choices_to={?WHAT SHOULD I ENTER HERE?})

如何将 MachinePowerSources.Machine 中的_choices_to 限制为属于 Company.user_company 的机器类型。 I.E.：我想将返回的机器选项限制为扩展用户配置文件的公司和用户公司模块为用户授权的机器选项。如果机器 X 属于公司 1，机器 Y 属于公司 2，并且用户在 UserCompany 中只有公司 2，则模块应该只接受/返回机器 y。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）