代码之家 › 专栏 › 技术社区 › Stephan

将原始sql映射到django orm

django-models orm django sql python

Stephan · 技术社区 · 17 年前

有没有办法简化这个工作代码? 这段代码为一个对象获取所有不同的投票类型,可能有20种,并对每种类型进行计数。我不喜欢编写原始sql,而是使用orm。这有点棘手,因为我在模型中使用了通用外键。

def get_object_votes(self, obj):
    """ 
    Get a dictionary mapping vote to votecount
    """
    ctype = ContentType.objects.get_for_model(obj)

    cursor = connection.cursor()
    cursor.execute("""
       SELECT v.vote , COUNT(*)
       FROM votes v
       WHERE %d = v.object_id AND %d = v.content_type_id
       GROUP BY 1
       ORDER BY 1 """ % ( obj.id, ctype.id )
    )
    votes = {}

    for row in cursor.fetchall():
       votes[row[0]] = row[1]

    return votes

class Vote(models.Model):
    user = models.ForeignKey(User)

    content_type = models.ForeignKey(ContentType)
    object_id = models.PositiveIntegerField()
    payload = generic.GenericForeignKey('content_type', 'object_id')

    vote = models.IntegerField(choices = possible_votes.items() )


class Issue(models.Model):
    title = models.CharField( blank=True, max_length=200)

2 回复 | 直到 17 年前

Stephan 17 年前

下面的代码帮我搞定了!

def get_object_votes(self, obj, all=False):
    """
    Get a dictionary mapping vote to votecount
    """
    object_id = obj._get_pk_val()
    ctype = ContentType.objects.get_for_model(obj)
    queryset = self.filter(content_type=ctype, object_id=object_id)

    if not all:
        queryset = queryset.filter(is_archived=False) # only pick active votes

    queryset = queryset.values('vote')
    queryset = queryset.annotate(vcount=Count("vote")).order_by()

    votes = {}

    for count in queryset:
        votes[count['vote']] = count['vcount']

    return votes

AlbertoPL 17 年前

是的,一定要用ORM。在您的模型中,您真正应该做的是:

class Obj(models.Model):
    #whatever the object has

class Vote(models.Model):
    obj = models.ForeignKey(Obj) #this ties a vote to its object

然后,要从一个对象获得所有投票,请将这些Django调用放在一个视图函数中:

obj = Obj.objects.get(id=#the id)
votes = obj.vote_set.all()

从那里可以很容易地看到如何计算它们(获得称为投票的列表的长度)。

http://www.djangoproject.com/documentation/models/many_to_one/

推荐文章

Johnny T · 基于当前值的SQL合并表[重复]

1 年前

Shukurullox Komiljonov · 从记录中获得相互和解。使用SQL

1 年前

John D · 需要为NULL或NOT NULL的WHERE子句

1 年前

ojek · 如何对SQL结果进行分组和编号?

1 年前

senek · 如何在PL/SQL中将选择结果(列)放入数组中

1 年前

Sax · 规范化Google表格(第一步)

1 年前

BarÄ±Å UÅaklÄ± · 如何在ON冲突更新中使用CTE中的值

1 年前

Jatin · 检索卷计数的动态sql抛出错误语法错误[关闭]

1 年前

Andrus · 如何在sql中查找第二个匹配项

1 年前

Nebula Tech · 在SQL Server中搜索字符串中单词的多次出现

1 年前