У меня проблема с наложением нескольких таблиц в django.
Давайте сделаем пример с банковскими счетами.
class account(models.Model):
name = models……
class accounttypeA(account):
balance = models.float…..
def addToBalance(self, value):
self.balance += value
class accounttypeB(account):
balance = models.int…. # NOTE this
def addToBalance(self, value):
value = do_some_thing_with_value(value) # NOTE this
self.balance += value
Теперь я хочу добавить значение к типу accounttype, но все, что у меня есть, - это объект учетной записи, например acc = account.object.get(pk = 29). Итак, кто ребенок ребенка?
Django автоматически создает поле account_ptr_id в accounttypeA и accounttypeB. Итак, мое решение было:
child_class_list = ['accounttypeA', 'accounttypeB']
for cl in child_class_list:
try:
exec("child = " + str(cl) + ".objects.select_for_update().get(account_ptr_id=" + str(acc.id) + ")")
logger.debug("Child found and ready to use.")
return child
except ObjectDoesNotExist:
logger.debug("Object does not exist, moving on…")
Возможно, это проблема чертежной доски на этом этапе!:)
Надеюсь, в моем примере я был ясен. Благодаря
Насколько мне известно, встроенный способ Django для этого не существует.
Однако, учитывая acc=account.object.get(pk=29), вы можете использовать:
try:
typeA = acc.accounttypeA
# acc is typeA
except accounttypeA.DoesNotExist:
# acc should be typeB if account only has typeA and typeB subclasses
try:
typeB = acc.accounttypeB
# acc is typeB
except accounttypeB.DoesNotExist:
# acc should be typeA if account only has typeA and typeB subclasses
мое решение было основано на этом
class account(models.Model):
name = models……
def cast(self):
"""
This method is quite handy, it converts "self" into its correct child class. For example:
.. code-block:: python
class Fruit(models.Model):
name = models.CharField()
class Apple(Fruit):
pass
fruit = Fruit.objects.get(name='Granny Smith')
apple = fruit.cast()
:return self: A casted child class of self
"""
for name in dir(self):
try:
attr = getattr(self, name)
if isinstance(attr, self.__class__) and type(attr) != type(self):
return attr
except:
pass
@staticmethod
def allPossibleAccountTypes():
#this returns a list of all the subclasses of account (i.e. accounttypeA, accounttypeB etc)
return [str(subClass).split('.')[-1][:-2] for subClass in account.__subclasses__()]
def accountType(self):
try:
if type(self.cast()) == NoneType:
#it is a child
return self.__class__.__name__
else:
#it is a parent, i.e. an account
return str(type(self.cast())).split('.')[-1][:-2]
except:
logger.exception()
accountType.short_description = "Account type"
class accounttypeA(account):
balance = models.float…..
def addToBalance(self, value):
self.balance += value
class accounttypeB(account):
balance = models.int…. # NOTE this
Django добавляет к классу account два поля: accounttypea и accounttypeb. Если у вас есть объект accounttypeb с pk = 42, вы можете получить доступ к родителям следующим образом:
account.objects.get(pk=42).accounttypeb
>>> <accounttypeB instance>
Вы можете добавить CharField к родительской модели с фактическим типом child для каждого дочернего элемента, а затем использовать getattr, если есть много дочерних моделей (это может быть лучше, чем много блоков try .. except xxx.DoesNotExist).
class account(models.Model):
name = models……
cls = CharField(...)
def ext(self):
return getattr(self, self.cls.lower())
class accounttypeA(account):
balance = models.float…..
def addToBalance(self, value):
self.balance += value
class accounttypeB(account):
balance = models.int…. # NOTE this
def addToBalance(self, value):
value = do_some_thing_with_value(value) # NOTE this
self.balance += value
# example
accounttypeB.objects.create(balance=10, name='Vincent Law', cls="accounttypeB")
accounttypeA.objects.create(balance=9.5, name='Re-l Mayer', cls="accounttypeA")
for obj in account.objects.all():
obj.ext.addToBalance(1.0)
print(obj.name, obj.cls)
но вы ДОЛЖНЫ создавать модели с помощью accounttypeA.objects.create(...) и accounttypeB.objects.create(...) - иначе этот трюк не будет работать.
(https://docs.djangoproject.com/en/1.5/topics/db/models/#multi-table-inheritance)