diff --git a/tv_program.py b/tv_program.py index e627d30..0d53eab 100644 --- a/tv_program.py +++ b/tv_program.py @@ -27,22 +27,22 @@ myday = f"{date.today():%d}" print("Day:", myday) # Define the time range (modify as needed) -default_start_time = 2015 # 5:30 AM -default_end_time = 2030 # 4:00 PM +default_start_time = "20:15" # 8:15 PM +default_end_time = "20:30" # 8:30 PM # Create an argument parser parser = argparse.ArgumentParser(description="TV Program Viewer") parser.add_argument( "--start-time", - type=int, + type=str, default=default_start_time, - help="Start time in HHMM format (e.g., 0530 for 5:30 AM)", + help="Start time in HH:MM format (e.g., 05:30 for 5:30 AM)", ) parser.add_argument( "--end-time", - type=int, + type=str, default=default_end_time, - help="End time in HHMM format (e.g., 2030 for 8:30 PM)", + help="End time in HH:MM format (e.g., 20:30 for 8:30 PM)", ) args = parser.parse_args() @@ -59,15 +59,16 @@ def getValues(id, start_time, end_time): st = a.get('startTime') d = a.get('duration') start = time.strftime("%-H:%M", time.localtime(st)) - start_time_int = int(start.replace(":", "")) + # Split the start time into hours and minutes + start_hours, start_minutes = map(int, start.split(":")) day = time.strftime("%d", time.localtime(int(st))) - if day == myday and start_time <= start_time_int < end_time: + if day == myday and start_time <= f"{start_hours:02}:{start_minutes:02}" <= end_time: data.append([start, title]) return data def makeList(start_time, end_time): - print("Channel:", len(dictList)) - table.append(["Channel", f"TV Shows from {start_time // 100}:{start_time % 100:02d} to {end_time // 100}:{end_time % 100:02d}"]) + print("Channels:", len(dictList)) + table.append(["Channel", f"TV Shows from {start_time} to {end_time}"]) for ch in dictList: data = [ch.upper()] channel_data = getValues(dictList.get(ch), start_time, end_time)