Python – 如何更简洁地使用生成器？

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了Python – 如何更简洁地使用生成器？，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含2427字，纯文字阅读大概需要4分钟。

内容图文

(Python 3)

我正在使用Python生成器从队列中读取消息.

在消费者读取队列消息之后,它需要能够告诉生成器在成功处理后删除队列消息.

为了将.send()发送到Python生成器,似乎我必须首先.send(None)到生成器.这使我的代码比我想象的要胖.

任何人都可以建议qconsumer.py用更少的代码行驱动生成器的方法吗？我已经确定了我希望消除的下面哪些行.

简而言之,我如何使下面的代码更紧凑,有关如何删除行的任何建议？

下面的代码是qconsumer.py：

from qserver import Qserver

myqserver = Qserver()

myproducer = myqserver.producer() # trying to eliminate this line
# first send to a generator must be None
myproducer.send(None) # trying to eliminate this line
for msg in myproducer:
    # do something with message
    print(msg)
    if messageprocessok:
        myproducer.send('delete')

下面的代码是qserver.py：

# -*- coding: utf-8 -*-
import boto
from boto.sqs.connection import SQSConnection
from boto.sqs.message import Message

QNAME = 'qinbound'
SQSREGION = 'us-west-1'

class Qserver():
    """A simple Q server."""

    def __init__(self, qname=None, sqsregion=None):
        self.qname = qname or QNAME
        self.sqsregion = sqsregion or SQSREGION
        self.sqsconn = boto.sqs.connect_to_region(self.sqsregion)
        self.q_in = self.sqsconn.get_queue(self.qname)

    def producer(self):
        while True:
            qmessage = self.q_in.read(wait_time_seconds=20)
            if qmessage is None:
                continue
            action = (yield qmessage.get_body())
            if action == 'delete':
                # if processing completed ok, clear message from this queue
                self.q_in.delete_message(qmessage)

解决方法:

理解了你想要做的事情后,我想我会避免将发送与迭代混合.将myqserver类作为迭代器本身似乎对我更有意义：

# -*- coding: utf-8 -*-
import boto
from boto.sqs.connection import SQSConnection
from boto.sqs.message import Message

QNAME = 'qinbound'
SQSREGION = 'us-west-1'

class Qserver():
    """A simple Q server."""
    _current_message = None

    def __init__(self, qname=None, sqsregion=None):
        self.qname = qname or QNAME
        self.sqsregion = sqsregion or SQSREGION
        self.sqsconn = boto.sqs.connect_to_region(self.sqsregion)
        self.q_in = self.sqsconn.get_queue(self.qname)

    def __iter__(self):
        return self

    def __next__(self):
        while True:
            qmessage = self.q_in.read(wait_time_seconds=20)
            if qmessage is not None:
                self._current_message = qmessage
                return qmessage

    next = __next__

    def delete_current(self):
        if self._current_message is not None:
            self.q_in.delete_message(self._current_message)

用法将是这样的：

from qserver import Qserver

myqserver = Qserver()
for msg in myqserver:
    # do something with message
    print(msg)
    if messageprocessok:
        myqserver.delete_current()

内容总结

以上是互联网集市为您收集整理的Python – 如何更简洁地使用生成器？全部内容，希望文章能够帮你解决Python – 如何更简洁地使用生成器？所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端