django异构查询集代理模型

时间:2010-10-12 00:33:55

标签: python django django-models

我试图弄清楚如何在Django中使用代理类。我想收到一个查询集,其中每个对象都属于一个公共超类的代理类,这样我就可以运行具有相同名称的自定义子类方法,而我的控制器逻辑不需要知道或关心哪种代理它正在使用的模型。我不想做的一件事是将信息存储在多个表中,因为我希望有统一的标识符以便于参考/管理。

我是django / python的新手,所以我很乐意听到其他方法来完成我想要做的事情。

这就是我所拥有的:

TYPES = (
    ('aol','AOL'),
    ('yhoo','Yahoo'),
)

class SuperConnect(models.Model):
  name = models.CharField(max_length=90)
  type = models.CharField(max_length=45, choices = TYPES)
  connection_string = models.TextField(null=True)

class ConnectAOL(SuperConnect):
  class Meta:
    proxy = True

  def connect(self):
     conn_options = self.deconstruct_constring()
     # do special stuff to connect to AOL

  def deconstruct_constring(self):
     return pickle.loads(self.connection_string)

class ConnectYahoo(SuperConnect):
  class Meta:
    proxy = True

  def connect(self):
     conn_options = self.deconstruct_constring()
     # do special stuff to connect to Yahoo

  def deconstruct_constring(self):
     return pickle.loads(self.connection_string)

现在我要做的是:

connections = SuperConnect.objects.all()

for connection in connections:
  connection.connect()
  connection.dostuff

我环顾四周,发现了一些黑客,但看起来有问题,可能要我​​去每个项目的数据库,以便检索我可能已经拥有的数据......

有人请救我:)或者我会选择这个黑客:

class MixedQuerySet(QuerySet):
    def __getitem__(self, k):
        item = super(MixedQuerySet, self).__getitem__(k)
        if item.atype == 'aol':
            yield(ConnectAOL.objects.get(id=item.id))
        elif item.atype == 'yhoo':
            yield(ConnectYahoo.objects.get(id=item.id))
        else:
            raise NotImplementedError

    def __iter__(self):
        for item in super(MixedQuerySet, self).__iter__():
            if item.atype == 'aol':
                yield(ConnectAOL.objects.get(id=item.id))
            elif item.atype == 'yhoo':
                yield(ConnectYahoo.objects.get(id=item.id))
            else:
                raise NotImplementedError

class MixManager(models.Manager):
    def get_query_set(self):
        return MixedQuerySet(self.model)

TYPES = (
    ('aol','AOL'),
    ('yhoo','Yahoo'),
)

class SuperConnect(models.Model):
  name = models.CharField(max_length=90)
  atype = models.CharField(max_length=45, choices = TYPES)
  connection_string = models.TextField(null=True)
  objects = MixManager()

class ConnectAOL(SuperConnect):
  class Meta:
    proxy = True

  def connect(self):
     conn_options = self.deconstruct_constring()
     # do special stuff to connect to AOL

  def deconstruct_constring(self):
     return pickle.loads(self.connection_string)

class ConnectYahoo(SuperConnect):
  class Meta:
    proxy = True

  def connect(self):
     conn_options = self.deconstruct_constring()
     # do special stuff to connect to Yahoo

  def deconstruct_constring(self):
     return pickle.loads(self.connection_string)

2 个答案:

答案 0 :(得分:1)

正如您在问题中提到的,解决方案的问题在于它为每个对象生成SQL查询,而不是使用一个SQL in = (id1, id2)查询。代理模型不能包含其他数据库字段,因此不需要额外的SQL查询。

相反,您可以使用SuperConnect.__init__属性将SuperConnect对象转换为__class__中的相应类型:

class SuperConnect(models.Model):
    name = models.CharField(max_length=90)
    type = models.CharField(max_length=45, choices = TYPES)
    connection_string = models.TextField(null=True)

    def __init__(self, *args, **kwargs):
        super(SuperConnect, self).__init__(*args, **kwargs)
        if self.type == 'aol':
            self.__class__ = ConnectAOL
        elif self.type == 'yahoo':
            self.__class__ = ConnectYahoo

不需要自定义管理器或查询集,初始化SuperConnect对象时会设置正确的类型。

答案 1 :(得分:0)

如何将所有逻辑放在一个类中。像这样:

def connect(self):
    return getattr(self, "connect_%s" % self.type)()

def connect_aol(self):
    pass # AOL stuff

def connect_yahoo(self):
    pass # Yahoo! stuff

最后你有type字段,你应该可以做大部分(如果不是全部)你可以用单独的代理类做的事情。

如果此方法无法解决您的具体用例,请澄清。