task03 completed

task-03/get_top_names.py

@@ -4,22 +4,31 @@
 https://github.com/astropgh/learning-by-doing/tree/master/task-03
 """
 
-def extract_data_lines(filename, start_text, end_text):
+def extract_data_lines(filename, start_text, end_text, include_start, include_end):
     """
     open `filename`, and yield the lines between
     the line that contains `start_text` and the line that contains `end_text`
     """
     # fill in code as needed
     with open(filename) as fh:
+        on = False
         for line in fh:
-            # fill in code as needed
-            # use `yield line` to return desired lines but keep the function going
+            if start_text in line:
+                on = True
+                if not include_start:
+                    continue
+            if end_text in line and not include_end:
+                    break
+            if on:
+                yield line
+            if end_text in line and include_end:
+                break
 
 
 if __name__ == '__main__':
     filename = 'top5names.html'
     start_text = '<tr><td align="center">2017</td>'
     end_text = '</table></center></div><!-- end #content -->'
 
-    for line in extract_data_lines(filename, start_text, end_text):
+    for line in extract_data_lines(filename, start_text, end_text, include_start=True, include_end=True):
         print(line)