select_time_from_csv.py

from datetime import timedelta

import pandas


def read_csv(file_name):
    file_name_with_extension = f"{file_name}.csv"
    df = pandas.read_csv(file_name_with_extension, parse_dates=["Date"])
    df = df[["Zone", "Date", "Status Code", "Response Time"]]
    return df


def select_time(df, hour: int, minute: int, mins_duration: int):
    date_of_csv = df["Date"][0]
    start_time = date_of_csv.replace(hour=hour, minute=minute)
    end_time = start_time + timedelta(minutes=mins_duration)

    selected_time = df[(df["Date"] > start_time) & (df["Date"] <= end_time)]
    return selected_time