Python, SimPy: использование yield внутри функций

Question

Привет, я строю относительно сложную модель имитации дискретных событий в SimPy.

Когда я пытаюсь поместить свои операторы yield в функции, моя программа, похоже, не работает. Ниже приведен пример.

import SimPy.SimulationTrace as Sim
import random

## Model components ##
class Customer(Sim.Process):
    def visit(self):
        yield Sim.hold, self, 2.0
        if random.random()<0.5:
            self.holdLong()
        else:
            self.holdShort()

    def holdLong(self):
        yield Sim.hold, self, 1.0
        # more yeild statements to follow

    def holdShort(self):
        yield Sim.hold, self, 0.5
        # more yeild statements to follow

## Experiment data ##
maxTime = 10.0 #minutes

## Model/Experiment ##
#random.seed(12345)
Sim.initialize()
c = Customer(name = "Klaus") #customer object
Sim.activate(c, c.visit(), at = 1.0)
Sim.simulate(until=maxTime)

Вывод, который я получаю при запуске:

0 activate <Klaus > at time: 1.0 prior: False
1.0 hold  < Klaus >  delay: 2.0
3.0 <Klaus > terminated

Методы holdLong () и holdShort, похоже, не работают вообще. Как я могу это исправить? Заранее спасибо.

Answer 1

Вызов функции генератора возвращает объект генератора, который может быть повторен.Вы просто игнорируете это возвращаемое значение, поэтому ничего не происходит.Вместо этого вы должны перебрать генератор и заново выдать все значения:

class Customer(Sim.Process):
    def visit(self):
        yield Sim.hold, self, 2.0
        if random.random()<0.5:
            for x in self.holdLong():
                yield x
        else:
            for x in self.holdShort():
                yield x

Answer 2

В Python yield не может распространяться вверх через вызов функции. Измените visit на что-то вроде этого:

def visit(self):
    yield Sim.hold, self, 2.0
    if random.random()<0.5:
        for x in self.holdLong():
            yield x
    else:
        for x in self.holdShort():
            yield x