update generator

michaelliao · michaelliao · commit e9d9ef6c1873 · 2015-04-03T10:25:46.000+08:00
diff --git a/py3/advance/do_generator.py b/py3/advance/do_generator.py
@@ -1,9 +1,9 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 
-g = (x * x for x in range(10))
-print(g)
-for x in g:
+s = (x * x for x in range(5))
+print(s)
+for x in s:
     print(x)
 
 def fib(max):
@@ -12,8 +12,29 @@ def fib(max):
         yield b
         a, b = b, a + b
         n = n + 1
+    return 'done'
 
 f = fib(10)
 print('fib(10):', f)
 for x in f:
     print(x)
+
+# call generator manually:
+g = fib(5)
+while 1:
+    try:
+        x = g.send(None)
+        print('g:', x)
+    except StopIteration as e:
+        print('Generator return value:', e.value)
+        break
+
+# call generator using iter:
+i = iter(fib(5))
+while 1:
+    try:
+        r = next(i)
+        print('i:', r)
+    except StopIteration as e:
+        print('Generator return value:', e.value)
+        break
