0

我正在尝试自定义本教程以在 Django 中使用 Kombu 和 Celery 创建点击计数器。本教程解释了如何为 url 做这件事,但我想做的是创建一个 Click 模型,将 Posts 定义为ForeignKey字段,然后每次有人查看该帖子时,我都会为该帖子调用增量点击。

我有两个问题:第一个是我应该将函数调用放在视图中的什么位置,因为我使用的是通用视图(见下文)?第二个是我应该如何使用message?我是否将对象写入传递给函数?

模型.py

class ClickManager(models.Manager):

    def increment_clicks(self, for_url, increment_by=1):
        click, created = self.get_or_create(url=for_url, defaults={"click_count": increment_by})
        if not created:
            click.click_count += increment_by
            click.save()

        return click.click_count

class Click(models.Model):
    obj = models.ForeignKey(Post)
    click_count = models.PositiveIntegerField(_(u"click_count"), default=0)

    objects = ClickManager()

    def __unicode__(self):
        return self.obj

消息传递.py

def send_increment_clicks(obj):
    connection = establish_connection()
    publisher = Publisher(connection=connection, exchange="clicks", routing_key="increment_click", exchange_type="direct")

    publisher.send(obj)

    publisher.close()
    connection.close()


def process_clicks():
    connection = establish_connection()
    consumer = Consumer(connection=connection, queue="clicks", exchange="clicks", routing_key="increment_click", exchange_type="direct")

    clicks_for_url = {}
    message_for_url = {}

    for message in consumer.iterqueue():
        obj = message.body
        clicks_for_url[obj]= clicks_for_url.get(obj, 0) + 1
        if obj in messages_for_url:
            messages_for_url[obj].append(message)
        else:
            messages_for_url[obj] = [message]

    for obj, click_count in clicks_for_url.items():
        Click.objects.increment_click(obj, click_count)
        [message.ack() for message in messages_for_url[obj]]

    consumer.close()
    connection.close()

视图.py

class LinkDetailView(FormMixin, DetailView):
    models      = Post
    queryset    = Post.objects.all()

    """DON'T KNOW HOW TO PASS SELF.OBJECT TO THE FUNCTION """
    send_increment_clicks[self.object] 

    def get_success_url(self):
        ...

    def get_context_data(self, **kwargs):
        ...
4

0 回答 0