Skip to main content
 首页 » 编程设计

python之如何确定多个外键中SQL插入的顺序

2025年05月04日9del

我最近开始使用 Django,并且还没有停止享受 python/Django,但我目前正在为一个逻辑问题而苦苦挣扎。

情况(简化):

class A(models.Model): 
    foo = models.CharField(max_length=255) 
 
class B(models.Model): 
    bar = models.CharField(max_length=255) 
    foo =  models.ForeignKey(A) 
 
class C(models.Model): 
    title = models.CharField(max_length=255) 
    bar =  models.ForeignKey(B) 
 
class D(models.Model): 
    name = models.CharField(max_length=255) 
    title =  models.ForeignKey(C) 
    bar =  models.ForeignKey(B) 

(实际用例由数百个此类组成,是的,一团糟,它清楚地证明了糟糕的数据库设计,但我对此无能为力)

我已经在每个类上创建了动态 ModelForms。一般目的是检索一个 excel 文件并将它们插入到字段验证等中的正确 ModelForms 中。每个 excel 文件都有多个映射到类的工作表,第一行(标题)描述模型字段,所有其他行代表数据。

数据完全未排序,因此通常不破坏外键序列的插入顺序为 A => B => C => D。但在这种情况下,整个序列可能像 D => B => C => A. 当我验证第一个工作表 D 时出现问题,因为尚未定义相关的外键而未验证。

问题是,如何添加所有数据并在之后验证参照完整性?

提前致谢!


感谢您的帮助!

实际上所有的主键都是从根模型派生出来的,根模型保存着所有子表的映射表。我没有在第一篇文章中提及它,因为我想让情况简单化。话虽如此,我不能改变它(困惑!),也不能重新设计类,因为它们映射到任何现有的(困惑!)数据库。为了让这个困惑变得完整,每个字段都设置为“not Null”。

我的第二个想法是首先填充一个映射表(还不知道如何做),并以此对传入的数据进行排序。听起来像是胡闹,很脏,我自己也不喜欢这个想法,我希望有更聪明的方法。

对于这个问题的任何数学解决方案,您有任何提示吗?这就像在任意数据上生成一棵树。

更新:

我做了两个函数来解决这个问题,还没有测试错误处理。

validate_tables:查找与给定应用程序相关的所有表并将嵌套列表 (self.found_fields) 保存在字典中 (child: [parent, parent, (...)])。

gen_sequence:将正确的序列映射到 object_names 写入列表 (self.sequence)。

欢迎批准!

这是我目前的解决方案(获得想法的片段)

    def validate_tables(self): 
        app = get_app("testdata") 
        self.sequence = [] 
        self.found_fields = {} 
        for model in get_models(app): 
            hits = [] 
            for local_field in model._meta.local_fields: 
                if isinstance(local_field, models.ForeignKey): 
                    hits.append(local_field.related.parent_model._meta.object_name) 
            self.found_fields.update({model._meta.object_name: hits}) 
        if self.gen_sequence(): 
            return True 
        else: 
            raise self.sequence_errors 
 
 
    def gen_sequence(self, unresolved=None): 
 
        if unresolved: 
            self.found_fields = unresolved 
            unresolved = {} 
        else: 
            unresolved = {} 
 
        for model in self.found_fields: 
            if ((all(parent in self.sequence for parent in self.found_fields[model])  
                 and self.sequence) 
                or not self.found_fields[model]): 
                self.sequence.append(model) 
            else: 
                unresolved.update({model: self.found_fields[model]}) 
 
        if unresolved == self.found_fields: 
            self.sequence_errors = unresolved 
            return False 
        elif not unresolved: 
            return self.gen_sequence 
        else: 
            return self.gen_sequence(unresolved) 

请您参考如下方法:

您将需要定义自己的主键,我假设您有一个合适的字段,否则不会出现此问题,并且还允许 ForeignKey 为 null。困难的部分将在稍后建立参照完整性,which is difficult but seemingly not impossible to do在 Django 中。

相反,我会有两个字段,一个是你的虚拟主键,另一个是让你当前的外键可以为空:

class A(models.Model): 
    foo = models.CharField(max_length=255) 
 
class B(models.Model): 
    bar = models.CharField(max_length=255) 
    foo =  models.ForeignKey(A, null=True) 
    foo_key =  models.CharField() 

然后,数据导入后,找到所有具有foo_key的'B'对象,建立关系并将foo_key设置为null。

这是我在将大量数据从以前的 GAE 项目导入到 PostgreSQL 数据库时使用的机制。