ForeignKeyに、複数のモデルを設定したい。

Question

DjangoでWebアプリを作成しています。

データベース上に顧客情報を登録するにあたり、どのような構成にしたら良いかと悩んでいます。

顧客は、法人の場合もあり、個人の場合もあります。
そのため、案件毎に顧客を登録するにあたり、下記コードの、customer = ForeignKey()に、Company（法人）とPerson（個人）のどちらでも持って来れるようにしたいです。

# models.py

# 法人情報
class Company(models.Model):
    name = models.CharField(max_length=30) # 法人名
    represetative = models.ForeignKey("Person", blank=True, null=True) # 法人の代表者
   …

# 個人情報
class Person(models.Model):
    name = models.CharField(max_length=30) # 個人名
   …

# 案件情報
class Work(models.Model):
    name = models.CharField(max_length=50) #案件名
    customer = models.ForeignKey() # ここの書き方で困っています。
   …

上記のようなニーズがある場合はよくあるのではないかと思うのですが、一般的にどのように解決されることが多いのでしょうか。

お分かりの方、ご教示頂ければ幸いです。
よろしくお願いいたします。

Answer 1

最終的に、下記のコードで実現するようにしました。

CUSTOMER_CHOICES = (
(0, u'法人'),
(1, u'個人')
)

class Company(models.Model):
"""法人情報"""
name = models.CharField(max_length=30) # 法人名

class Person(models.Model):
"""個人情報"""
name = models.CharField(max_length=30) # 個人名

class Customer(models.Model):
"""顧客"""
customer_type = models.IntegerField(choices=CUSTOMER_CHOICES)
company = models.OneToOneField(Company, null=True)
person = models.ForeignKey(Person)

class Work(models.Model):
"""案件情報"""
name = models.CharField(max_length=50) # 案件名
customer = models.ForeignKey(Customer)

テーブルは増えていますが、法人・個人の違いをcustomerが吸収してくれるため、分かりやすいコードで実現できたと思います。

Comments

[PolymetisOutis7]

私もForeignKeyで複数のモデルを候補にしたく取り組んでいます。
質問者様のやり方だと、実装してみたところ、Customerの項目で個人を選んでも会社名が必須項目になってしまいます。
それでも良いのでしょうか？

Answer 2

一番単純な方法は generic foreign keyを使うことです。ただし色々制限があるのでその辺りは調べてから使ってください。 https://docs.djangoproject.com/ja/1.9/ref/contrib/contenttypes/

Comments

[退会済みユーザー]

ご回答頂き、ありがとうございます。
ご教示頂いた、generic foreign keyでの方法も検討したのですが、最終的に、自己解決の方に書いたコードで実現することにしました。