main.py

from ExportTvTime.tv_time import ExportTvTimeData


def main():
    with ExportTvTimeData() as my_tvtime:
        shows = my_tvtime.get_list_followed_shows()
        print(type(shows))

        sorted_shows = my_tvtime.change_columns_order(shows)
        print(type(sorted_shows))

        keep_columns = my_tvtime.delete_columns(
            sorted_shows,
            [
                "followed",
                "last_seen",
                "next_aired",
                "diffusion",
                "folder",
                "notification_type",
                "is_web_serie",
                "hashtag",
                "number_of_seasons",
                "favorite",
                "last_aired",
                "all_images",
            ],
            True
        )

        my_tvtime.create_csv_file(keep_columns)


if __name__ == "__main__":
    main()