我必须创建一个在内部调用中进行一些艰苦工作的函数。该函数必须是生成器,因为我正在使用服务器发送的事件。因此,我希望此函数通过使用“收益”来通知计算进度。之后,该函数必须将结果传递给父函数,以便继续进行其他计算。
我想要这样的东西:
def hardWork():
for i in range(N):
# hard work
yield 'Work done: ' + str(i)
# Here is the problem: I can't return a result if I use a yield
return result
def generator():
# do some calculations
result = hardWork()
# do other calculations with this result
yield finalResult
我发现一个基于的解决方案产生了一个字典,该字典可以告诉函数是否已完成,但是执行此操作的代码很脏。
还有其他解决方法吗?
谢谢!
我想到了类似的东西
def innerFunction(gen):
calc = 1
for iteration in range(10):
for i in range(50000):
calc *= random.randint(0, 10)
gen.send(iteration)
yield calc
def calcFunction(gen):
gen2 = innerFunction(gen)
r = next(gen2)
gen.send("END: " + str(r + 1))
gen.send(None)
def notifier():
while True:
x = yield
if x is None:
return
yield "Iteration " + x
def generator():
noti = notifier()
calcFunction(noti)
yield from noti
for g in generator():
print(g)
但是我收到此错误:
TypeError: can't send non-None value to a just-started generator
答案 0 :(得分:0)
此解决方案也适用于最新的Python版本,尽管Python3.5中的async def
似乎更适合您的用例。 请参阅下一部分。
通过迭代或使用next
获得生成器产生的值。在结尾处返回的值存储在value
异常的StopIteration
属性中,该属性指示生成器的结尾。幸运的是,恢复起来并不难。
def hardWork():
output = []
for i in range(10):
# hard work
yield 'Doing ' + str(i)
output.append(i ** 2)
return output
def generator():
# do some calculations
work = hardWork()
while True:
try:
print(next(work))
except StopIteration as e:
result = e.value
break
yield result
foo = generator()
print(next(foo))
Doing 0
Doing 1
Doing 2
Doing 3
Doing 4
Doing 5
Doing 6
Doing 7
Doing 8
Doing 9
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
async def
如果您正在运行Python3.5 +,则您尝试的操作似乎完全适合使用等待函数的事件循环。
import asyncio
async def hardWork():
output = []
for i in range(10):
# do hard work
print('Doing ', i)
output.append(i**2)
# Break point to allow the event loop to do other stuff on the side
await asyncio.sleep(0)
return output
async def main():
result = await asyncio.wait_for(hardWork(), timeout=None)
print(result)
loop = asyncio.get_event_loop()
loop.run_until_complete(main())
Doing 0
Doing 1
Doing 2
Doing 3
Doing 4
Doing 5
Doing 6
Doing 7
Doing 8
Doing 9
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
答案 1 :(得分:0)
我的建议是将您的功能嵌入类中。
def Worker:
def hardWork(self):
self.Finished = False
for i in range(10):
yield 'Work done: ' + str(i)
self.Finished = True
self.Result = 'result'
def generator(self):
while (not self.Finished):
print(next(self.hardWork()))
return self.Result
这将具有您想要的功能,而不必担心围绕异常抛出捕获逻辑对逻辑进行编程。
答案 2 :(得分:0)
Python 3.3引入了yield from
for generator delegation,几乎完全符合您的要求。 yield from
允许主生成器将功能委派给另一个包含yield
语句的函数,如下所示:
def hardWork():
for i in range(N):
# hard work
yield 'Work done so far: ' + str(i)
# With this construction, you can still return a result after the `yield`s
return result
def generator():
# here's the new construction that makes it all work:
result = yield from hardWork()
# do other calculations with this result
yield finalResult