读取当前迭代器值而不在Python中递增

时间:2015-10-05 21:18:48

标签: python iterator next itertools

我正在编写一个具有两种不同状态的程序(脚和仪表之间的可逆单位转换器),程序中的几个函数取决于当前状态(交替迭代的循环(循环)迭代器)。用户可以调用反向函数来切换当前状态并反转计算转换函数。

目前,我使用next(currentstate)返回迭代器的下一个值,如下所示:

self.currentstate = cycle(range(2))    

def reverse(self):
    if next(self.currentstate) == 0:
        self.feet_label.grid(column=2, row=2, sticky=tk.W)
    if next(self.currentstate) == 1:
        self.feet_label.grid(column=2, row=1, sticky=tk.W)

def calculate(self, *args):
    if next(self.currentstate) == 0:
        # convert feet to meters
    if next(self.currentstate) == 1:
        # convert meters to feet

不幸的是,无论何时调用函数并计算if语句,循环迭代器都会被下一个运算符递增,下一次调用将产生不同的结果。计算函数可能会在同一个状态下多次调用,因此我想要一些检索迭代器当前值的方法,而无需修改或增加运算符。

def calculate(self, *args):
    if currentvalue(self.currentstate) == 0:
        # convert feet to meters
    if currentvalue(self.currentstate) == 1:
        # convert meters to feet

我发现了一个非常丑陋的解决方法,涉及在每个if语句中调用next(currentvalue)两次来重置二进制值。这可能是编写像这样的两个状态程序的非常糟糕的方式,但似乎应该有这样做的方法。我是Python的新手,也可能没有完全理解迭代器的基础理论。

由于

1 个答案:

答案 0 :(得分:3)

听起来你不应该在这里使用迭代器。你应该使用一些必须明显改变状态的东西。将这一切包装在自己的类中可能会更好。

class StateMachine(object):

    STATE_ON = 1
    STATE_OFF = 0  # this could be an enum maybe?

    def __init__(self, starting_state=0):
        self.state = starting_state

    def self.change_state(self):
        if self.state = self.STATE_ON:
            self.state = self.STATE_OFF
        else:
            self.state = self.STATE_ON

现在,只要您使用状态机,就必须明确更改状态。

statemachine = StateMachine()

def calculate(*args):
    if statemachine.state == statemachine.STATE_ON:
        do_something
    if statemachine.state == statemachine.STATE_OFF:
        do_something_else

def switch_state(*args):
    do_something  # and...
    statemachine.change_state()