from datetime import datetime
from dsst_gtk3.gtk_ui import DSSTGtkUi
from dsst_sql import sql
from dsst_gtk3 import dialogs, util


class LeftColumnHandlers:
    def __init__(self, app: DSSTGtkUi):
        self.app = app

    def do_add_season(self, *_):
        name = dialogs.enter_string_dialog(self.app.ui, 'Name for the new Season')
        if name:
            sql.Season.create(game_name=name, number=1)
            self.app.reload_seasons()

    def do_season_selected(self, *_):
        self.app.reload_for_season(self.app.get_selected_season_id())

    def do_add_episode(self, *_):
        season_id = self.app.get_selected_season_id()
        if not season_id:
            return
        episode = dialogs.show_episode_dialog(self.app.ui, 'Create new Episode', season_id)
        self.app.reload_for_season(season_id)