オーバーライドを共通化したいです

Question

実現したいこと

大目的
いくつかのクラスで下記のようなsaveメソッドのオーバーライドが複数あるので、
それらを一つのコードにまとめて共通化したいです。

小目的
共通化するために、下記のコードに記載してあるFruitクラスのインスタンスを
HogeMixinのクラスに渡したいです。

class Fruit(models.Model):
      quantity = models.ForeignKey("Quantity", on_delete=models.CASCADE, null=True)
      kind = = models.IntegerField(default=0)

　　　　　　　　　　　＃　共通化したい箇所
      def save(self, *args, **kwargs):
        if self._state.adding:
            if self.quantity:
                self.kind = self.quantity.kind
                self.kind = self.quantity.kind
                
        return super().save(*args, **kwargs)

該当のソースコード

# 共通化するために、mixinを作成
class HogeMixin(models.Model):
    def save(self, *args, **kwargs):
        if self._state.adding:
            if self.quantity:　# selfにFruitのインスタンスを渡したいです。
                self.kind = self.quantity.kind
                self.kind = self.quantity.kind
                
        return super().save(*args, **kwargs)

class Fruit(HogeMixin): 
　　　　　　　　　　　# models.ModelはHogeMixinに継承させました
      quantity = models.ForeignKey("Quantity", on_delete=models.CASCADE, null=True)
      kind = = models.IntegerField(default=0)

　　     a = HogeMixin()
      a.save()

発生している問題・エラーメッセージ

上記のコードを作成して、正しく動作するか単体テストを実行したところ
下記のエラーが表示されました。
（コードを共通化する前は、テストをクリアしていたのでテスト自体に問題はないと思います。）

AttributeError: 'HogeMixin' object has no attribute 'quantity'

試したこと

AttributeError: 'HogeMixin' object has no attribute 'quantity'

のエラーが出るということは、 if self.quantity:のselfはHogeMixinを指していると考えています。

下記の記事を参考に、継承元（HogeMixin）で継承先（Fruit）のインスタンスを参照するようにコードを書き換えました。
https://naruport.com/blog/2019/8/24/python-tutorial-class-extends/

selfの型がCatになっているのがわかります。
このように、継承先（Cat）で継承元（Animal）のメソッドを呼び出すと、メソッドのselfは継承先（Cat）のインスタンスになります。

何か解決の糸口になるヒントを頂けますと幸いです。
よろしくお願いいたします。

Answer 1

継承とか一切関係なく↓のコードはHogeMixinをインスタンス化してsaveメソッドを呼んでいるので、記載されているエラーが発生します。

    a = HogeMixin()
    a.save()

多分やりたい事は↑の2行を消すことで解決されると思いますが如何でしょうか。
つまり、↓のような感じです。

class Fruit(HogeMixin): 
      quantity = models.ForeignKey("Quantity", on_delete=models.CASCADE, null=True)
      kind = = models.IntegerField(default=0)

これによりFruitのインスタンスのsaveが呼ばれた際に、HogeMixinのsaveが呼ばれます。（この際にselfはFruitのインスタンスになります。）

---

以下は直接の回答ではないですが参考になれば。
なんとなくですが、saveメソッド内でquantityを呼ぶのであれば、HogeMixinにquantityを持たせるべきな気がします。
そうでないと、HogeMixinを継承したモデル全てにquantityを設定する必要があります。
（継承モデルでquantityの設定が異なるのであればHogeMixinで共通化するべきではないですし）

class HogeMixin(models.Model):
    def save(self, *args, **kwargs):
        if self._state.adding:
            if self.quantity:
                self.kind = self.quantity.kind
                self.kind = self.quantity.kind                
        return super().save(*args, **kwargs)

class Fruit(HogeMixin): 
    huga = models.... # Fruit独自の属性

Comments

[Taka2401]

ご回答ありがとうございます。

ご記載いただいたコードを試したところ、該当のエラーを解決することができました。
別途、django.core.exceptions.FieldError: Local field 'id' in class 'Fruit' clashes with field of the same name from base class 'HogeMixin'.のエラーが出ましたが、まずは自分で調査して解決したいと思います。

＞以下は直接の回答ではないですが参考になれば。
＞HogeMixinを継承したモデル全てにquantityを設定する必要があります。
また、ご厚意ありがとうございます。
実は今回のケースだと、quantityは仮のカラム名で投稿させていただきまして、
継承するモデル全てに共通のカラムが存在しております。

ご丁寧に説明していただき、大変理解しやすかったです。
ありがとうございました。