get program for pixel flut - wrap, line count arg

1 files changed, 19 insertions, 15 deletions

diff --git a/src/get_program_for_pixel_flut.py b/src/get_program_for_pixel_flut.py
index 5320c9c..e86489c 100644
--- a/src/get_program_for_pixel_flut.py
+++ b/src/get_program_for_pixel_flut.py
@@ -1,19 +1,23 @@
-from program_parser import program_parser_factory, Event
-
+import os
 
-program_parser = program_parser_factory()
+from program_parser import program_parser_factory, Event
 
-events_to_show: list[Event] = program_parser.get_events_to_show(program_parser.get_program(), programs_to_show_count = 1000)
-#{event.title}  {day}  {start_time}-{end_time} {icon}
-programs_formatted_list = []
-last_event = None
-for event in events_to_show:
-    programs_formatted_list.append(f"{event.title} {event.time}  {event.icon} {event.progress if event.progress > 0 else ''}")
-    if last_event and last_event.day != event.day:
-        # Day change
-        programs_formatted_list.append(f'\n\n{event.day}\n')
-    last_event = event
+def main():
+    program_parser = program_parser_factory()
 
-program_formatted = '\n'.join(programs_formatted_list)
-print(program_formatted)
+    lines: int = int(os.environ["LINES"]) if "LINES" in os.environ else 100
+    events_to_show: list[Event] = program_parser.get_events_to_show(program_parser.get_program(), lines=lines)
+    #{event.title}  {day}  {start_time}-{end_time} {icon}
+    programs_formatted_list = []
+    last_event = None
+    for event in events_to_show:
+        programs_formatted_list.append(f"{event.title} {event.time}  {event.icon} {event.progress if event.progress > 0 else ''}")
+        if last_event and last_event.day != event.day:
+            # Day change
+            programs_formatted_list += [f'', '', event.day,'']
+        last_event = event
+    program_formatted = '\n'.join(programs_formatted_list[:lines])
+    print(program_formatted)
 
+if __name__ == '__main__':
+    main()