Tôi không coi đây là một trường hợp sử dụng thường xuyên, xem xét việc này như các máy phát điện ban đầu:
def original_gen():
for x in range(10):
should_break = yield x
if should_break:
break
Nếu giá trị của should_break luôn tính toán dựa trên một số gọi hàm với x thì tại sao không chỉ viết máy phát như thế này:
def processing_gen(check_f):
for x in range(10):
yield x
should_break = check_f(x)
if should_break:
break
Tuy nhiên, tôi thường nghĩ về mã xử lý các giá trị được tạo ra như được viết bên trong vòng lặp (nếu không điểm của việc có một vòng ở tất cả là bao nhiêu?)
gì nó thực sự có vẻ như bạn muốn làm là tạo ra một máy phát điện ở đâu gọi phương thức __next__ thực sự ngụ ý send(process(LAST_VALUE)) mà có thể được thực hiện với một lớp:
class Followup_generator(): #feel free to use a better name
def __init__(self,generator,following_function):
self.gen = generator
self.process_f = following_function
def __iter__(self):
return self
def __next__(self):
if hasattr(self,"last_value"):
return self.send(self.process_f(self.last_value))
else:
self.last_value = next(self.gen)
return self.last_value
def send(self,arg):
self.last_value = self.gen.send(arg)
return self.last_value
def __getattr__(self,attr):
"forward other lookups to the generator (.throw etc.)"
return getattr(self.gen, attr)
# call signature is the exact same as @tobias_k's checking_generator
traversal = Followup_generator(bfs(g, start_node), process)
for n in traversal:
print(n)
n = traversal.send(DATA) #you'd be able to send extra values to it
tuy nhiên điều này vẫn không coi đây là thường xuyên sử dụng, tôi muốn được hoàn toàn tốt đẹp với một vòng lặp while, mặc dù tôi muốn đặt .send gọi ở đầu trang:
traversal = bfs(g, start_node)
send_value = None
while True:
n = traversal.send(send_value)
#code for loop, ending in calculating the next send_value
send_value = process(n)
Và bạn có thể quấn rằng trong một try: ... except StopIteration:pass mặc dù tôi thấy rằng chỉ đơn giản là chờ đợi một lỗi để nâng cao được thể hiện tốt hơn với một người quản lý bối cảnh:
class Catch:
def __init__(self,exc_type):
if issubclass(exc_type,BaseException):
self.catch_type = exc_type
else:
raise TypeError("can only catch Exceptions")
def __enter__(self):
return self
def __exit__(self,exc_type,err, tb):
if issubclass(exc_type, self.catch_type):
self.err = err
return True
with Catch(StopIteration):
traversal = bfs(g, start_node)
send_value = None
while True:
n = traversal.send(send_value)
#code for loop, ending in calculating the next send_value
send_value = process(n)
Vâng, bạn có thể làm cho nó ngắn hơn nhiều bằng cách đặt thử/ngoại trừ bên ngoài vòng lặp; điều này sẽ giúp bạn tiết kiệm một bộ thử/trừ và điều kiện if. –
@tobias_k đã được khắc phục, cảm ơn. – max