Làm thế nào tôi có thể xác định nếu trường hợp của lớp từ mô hình Django là phân lớp của một mô hình khác?

Question

Tôi có một lớp được gọi là BankAccount làm lớp cơ sở. Tôi cũng có các lớp học CheckingAccount và SavingsAccount được kế thừa từ BankAccount.

BankAccount không phải là lớp trừu tượng nhưng tôi không tạo đối tượng từ lớp đó, chỉ có các lớp kế thừa.

Sau đó, tôi thực hiện một truy vấn như thế này:

account = BankAccount.objects.get(id=10) 

Làm thế nào để tôi biết nếu tài khoản là CheckingAccount hoặc SavingsAccount?

Cách tôi làm điều này bây giờ là theo cách này:

checking_account = CheckingAccount.objects.get(id=account.id) 

Nếu nó tồn tại, nó là một CheckingAccount, nếu không, nó là một SavingsAccount.

Answer 1

Hãy thử sử dụng thuộc tính checkingaccount và savingsaccount. Cái nó sẽ không nổ tung.

Answer 2

Một janky chút, nhưng điều này sẽ làm việc:

>>> class BankAccount(object): pass 
... 
>>> class SavingsAccount(BankAccount): pass 
... 
>>> class CheckingAccount(BankAccount): pass 
... 
>>> x = SavingsAccount() 
>>> type(x) == type(SavingsAccount()) 
True 
>>> type(x) == type(CheckingAccount()) 
False 

Answer 3

Bạn có thể sử dụng isinstance (account, SavingsAccount), nhưng là generally preferred to avoid it và sử dụng duck type inference bằng cách nhìn vào các thuộc tính của đối tượng, và xem nếu nó quacks như một lớp con.

Để xem nếu object has an attribute, bạn sử dụng tên thích hợp hasattr built-infunction hoặc sử dụng getattr và kiểm tra việc tăng ngoại lệ AttributeError.

Answer 4

Thêm phương thức GetAccountType() vào tài khoản Kiểm tra và tiết kiệm của bạn, khi bạn lấy đối tượng trở lại từ BankAccount.objects.get() rồi gọi, nếu mọi thứ có nguồn gốc từ BankAccount có phương pháp đó thì bạn sẽ ổn .

Answer 5

Sau khi một số tìm kiếm hơn tôi tìm thấy giải pháp tương tự như này: Django multi-table inheritance, how to know which is the child class of a model?

Về cơ bản, không có giải pháp thanh lịch cho việc này. Bạn phải thực hiện một loạt các câu lệnh try-except và django để sử dụng lớp bạn muốn.