StreamCat_GUI.py

from database import DatabaseConnection
import customtkinter as ctk
from CTkListbox import *
ctk.set_appearance_mode("dark")
ctk.set_default_color_theme("blue")

class CTkAutocompleteCombobox(ctk.CTkComboBox):
    """:class:`ctk.CTkComboBox` widget that features autocompletion."""
    def __init__(self, master=None, completevalues=None, **kwargs):
        """Create a CTkAutocompleteCombobox.

        Args:
            master (widget): Parent frame of widget
            completevalues (list): autocomplettion values
            kwargs: keyword arguments passed to the :class:`ctk.CTkComboBox` initializer
        """

        ctk.CTkComboBox.__init__(self, master, values=completevalues, **kwargs)
        self._completion_list = completevalues
        if isinstance(completevalues, list):
            self.set_completion_list(completevalues)
        self._hits = []
        
        self._hit_index = 0
        self.position = 0
        self.tk.eval
        # navigate on keypress in the dropdown:
        # code taken from https://wiki.tcl-lang.org/page/ttk%3A%3Acombobox by Pawel Salawa, copyright 2011
#         self.tk.eval("""
# proc ComboListKeyPressed {w key} {
#         if {[string length $key] > 1 && [string tolower $key] != $key} {
#                 return
#         }

#         set cb [winfo parent [winfo toplevel $w]]
#         set text [string map [list {[} {\[} {]} {\]}] $key]
#         if {[string equal $text ""]} {
#                 return
#         }

#         set values [$cb cget -values]
#         set x [lsearch -glob -nocase $values $text*]
#         if {$x < 0} {
#                 return
#         }

#         set current [$w curselection]
#         if {$current == $x && [string match -nocase $text* [lindex $values [expr {$x+1}]]]} {
#                 incr x
#         }

#         $w selection clear 0 end
#         $w selection set $x
#         $w activate $x
#         $w see $x
# }

# set popdown [ttk::combobox::PopdownWindow %s]
# bind $popdown.f.l <KeyPress> [list ComboListKeyPressed %%W %%K]
# """ % (self))

    def set_completion_list(self, completion_list):
        """
        Use the completion list as drop down selection menu, arrows move through menu.

        :param completion_list: completion values
        :type completion_list: list
        """
        self._completion_list = sorted(completion_list, key=str.lower)  # Work with a sorted list
        self.configure(values=completion_list)
        self._hits = []
        self._hit_index = 0
        self.position = 0
        self.bind('<KeyRelease>', self.handle_keyrelease)
        self['values'] = self._completion_list  # Setup our popup menu

    def autocomplete(self, delta=0):
        """
        Autocomplete the Combobox.

        :param delta: 0, 1 or -1: how to cycle through possible hits
        :type delta: int
        """
        if delta:  # need to delete selection otherwise we would fix the current position
            self._entry.delete(self.position, ctk.END)
            self
        else:  # set position to end so selection starts where textentry ended
            self.position = len(self.get())
        # collect hits
        _hits = []
        for element in self._completion_list:
            if element.lower().startswith(self.get().lower()):  # Match case insensitively
                _hits.append(element)
        # if we have a new hit list, keep this in mind
        if _hits != self._hits:
            self._hit_index = 0
            self._hits = _hits
        # only allow cycling if we are in a known hit list
        if _hits == self._hits and self._hits:
            self._hit_index = (self._hit_index + delta) % len(self._hits)
        # now finally perform the auto completion
        if self._hits:
            self._entry.delete(0, ctk.END)
            self._entry.insert(0, self._hits[self._hit_index])
            self._entry.select_range(self.position, ctk.END)

    def handle_keyrelease(self, event):
        """
        Event handler for the keyrelease event on this widget.

        :param event: Tkinter event
        """
        if event.keysym == "BackSpace":
            self._entry.delete(self._entry.index(ctk.INSERT), ctk.END)
            self.position = self._entry.index(ctk.END)
        if event.keysym == "Left":
            if self.position < self._entry.index(ctk.END):  # delete the selection
                self._entry.delete(self.position, ctk.END)
            else:
                self.position -= 1  # delete one character
                self._entry.delete(self.position, ctk.END)
        if event.keysym == "Right":
            self.position = self._entry.index(ctk.END)  # go to end (no selection)
        if event.keysym == "Return":
            self.handle_return(None)
            return
        if len(event.keysym) == 1:
            self.autocomplete()
            # No need for up/down, we'll jump to the popup
            # list at the position of the autocompletion

    def handle_return(self, event):
        """
        Function to bind to the Enter/Return key so if Enter is pressed the selection is cleared

        :param event: Tkinter event
        """
        self._entry.icursor(ctk.END)
        self.selection_clear()

    def config(self, **kwargs):
        """Alias for configure"""
        self.configure(**kwargs)

    def configure(self, **kwargs):
        """Configure widget specific keyword arguments in addition to :class:`ctk.CTkComboBox` keyword arguments."""
        if "completevalues" in kwargs:
            self.set_completion_list(kwargs.pop("completevalues"))
        return ctk.CTkComboBox.configure(self, **kwargs)

    def cget(self, key):
        """Return value for widget specific keyword arguments"""
        if key == "completevalues":
            return self._completion_list
        return ctk.CTkComboBox.cget(self, key)

    def keys(self):
        """Return a list of all resource names of this widget."""
        keys = ctk.CTkComboBox.keys(self)
        keys.append("completevalues")
        return keys

    def __setitem__(self, key, value):
        self.configure(**{key: value})

    def __getitem__(self, item):
        return self.cget(item)


class ProgressbarFrame(ctk.CTkToplevel):
    def __init__(self, parent):
        super().__init__(parent)
        self.parent = parent

class DbResultsFrame(ctk.CTkToplevel):
    def __init__(self, parent, response):
        super().__init__(parent)
        self.parent = parent
        self.geometry("450x350")

        self.label = ctk.CTkLabel(self, text="Database Response")
        self.label.pack(side=ctk.TOP, padx=20, pady=20)
        self.response = response
        self.response_str = ''
        if isinstance(self.response, (tuple, list)):
            for i in self.response:
                if not isinstance(i, str):
                    flattend_i = [item for sub in i for item in sub]
                    self.response_str += '\n'.join(map(str, flattend_i))
                else:
                    self.response_str += f'\n{i}'
        else:
            self.response_str = str(self.response)
        self.textbox = ctk.CTkTextbox(self)
        self.textbox.insert("0.0", text=self.response_str)
        self.textbox.configure(state="disabled")
        self.textbox.pack(expand=True, fill=ctk.BOTH)
    
class DbUpdatesFrame(ctk.CTkToplevel):
    def __init__(self, parent):
        super().__init__(parent)
        self.parent = parent
        self.geometry("450x350")

        self.label = ctk.CTkLabel(self, text="Database Update Form")
        self.label.pack(side=ctk.TOP, padx=20, pady=20)

        self.partition_label = ctk.CTkLabel(self, text="Was this update for streamcat or lakecat?")
        self.partition_label.pack(padx=10, pady=5)
        self.partition_var = ctk.StringVar()
        self.partition_var.set('streamcat')
        self.partition_radio_streamcat = ctk.CTkRadioButton(self, text="StreamCat", variable=self.partition_var, value='streamcat')
        self.partition_radio_streamcat.pack(padx=10, pady=5)
        self.partition_radio_lakecat = ctk.CTkRadioButton(self, text="LakeCat", variable=self.partition_var, value='lakecat')
        self.partition_radio_lakecat.pack(padx=10, pady=5)

        self.internal_update_label = ctk.CTkLabel(self, text="Internal database update (what changes did you make?): ")
        self.internal_update_label.pack(padx=10, pady=5)

        self.internal_update_entry = ctk.CTkEntry(self, width=280) # maybe change to textbox
        self.internal_update_entry.pack(padx=10, pady=5)

        self.public_update_label = ctk.CTkLabel(self, text="Public database update, this will be displayed on the changelog page, be specific: ")
        self.public_update_label.pack(padx=10, pady=5)

        self.public_update_entry = ctk.CTkEntry(self, width=280) # maybe change to textbox
        self.public_update_entry.pack(padx=10, pady=5)

        self.submit_button = ctk.CTkButton(self, text="Submit", command=self.add_to_changelog)
        self.submit_button.pack(fill=ctk.X, padx=10, pady=5)

    def add_to_changelog(self):
        change_desc = self.internal_update_entry.get()
        public_desc = self.public_update_entry.get()
        partition = self.partition_var.get()
        result = db_conn.newChangelogRow(partition, public_desc, change_desc)
        
    
class CreateDatasetFrame(ctk.CTkScrollableFrame):
    def __init__(self, parent):
        super().__init__(parent)
        self.parent = parent

        # Create widgets for creating a dataset
        self.partition_label = ctk.CTkLabel(self, text="What partition would you like to use?")
        self.partition_label.pack(padx=10, pady=5)

        self.partition_var = ctk.StringVar()
        self.partition_var.set('streamcat')
        self.partition_radio_streamcat = ctk.CTkRadioButton(self, text="StreamCat", variable=self.partition_var, value='streamcat')
        self.partition_radio_streamcat.pack(padx=10, pady=5)
        self.partition_radio_lakecat = ctk.CTkRadioButton(self, text="LakeCat", variable=self.partition_var, value='lakecat')
        self.partition_radio_lakecat.pack(padx=10, pady=5)

        self.dataset_label = ctk.CTkLabel(self, text="What is the name of this dataset? ") # sc_datasets.dsname and sc_metrics_tg.final_table
        self.dataset_label.pack(padx=10, pady=5)

        self.dataset_entry = ctk.CTkEntry(self, width=280)
        self.dataset_entry.pack(padx=10, pady=5)

        self.source_name_label = ctk.CTkLabel(self, text="Enter metric source name:")
        self.source_name_label.pack(padx=10, pady=5)
        self.source_name_entry = ctk.CTkEntry(self, width=280)
        self.source_name_entry.pack(padx=10, pady=5)

        self.source_url_label = ctk.CTkLabel(self, text="Enter metric source url:")
        self.source_url_label.pack(padx=10, pady=5)
        self.source_url_entry = ctk.CTkEntry(self, width=280)
        self.source_url_entry.pack(padx=10, pady=5)

        self.date_label = ctk.CTkLabel(self, text="Enter the date the dataset was downloaded with the format dd-MM-YY:")
        self.date_label.pack(padx=10, pady=5)
        self.date_entry = ctk.CTkEntry(self, width=280)
        self.date_entry.pack(padx=10, pady=5)

        self.visible_var = ctk.StringVar()
        self.visible_var.set('invisible')
        self.visible_radio = ctk.CTkRadioButton(self, text="Visible?", variable=self.partition_var, value='invisible')
        self.visible_radio.pack(padx=10, pady=5)
        self.invisible_radio = ctk.CTkRadioButton(self, text="Invisible?", variable=self.partition_var, value='visible')
        self.invisible_radio.pack(padx=10, pady=5)


        self.files_label = ctk.CTkLabel(self, text="Choose files:")
        self.files_label.pack(fill=ctk.X, padx=10, pady=5)

        self.files_button = ctk.CTkButton(self, width=280, text="Browse", command=self.browse_files)
        self.files_button.pack(padx=10, pady=5) # fill=ctk.X,

        self.files_entry = ctk.CTkEntry(self)
        self.files_entry.pack(fill=ctk.X, padx=10, pady=5)

        self.submit_button = ctk.CTkButton(self, width=280, text="Submit", command=self.create_dataset)
        self.submit_button.pack(side=ctk.BOTTOM, padx=10, pady=5)

        self.results_window = None
        self.updates_window = None

    def browse_files(self):
        files = ctk.filedialog.askopenfilenames()
        self.files_entry.delete(0, ctk.END)
        self.files_entry.insert(0, ', '.join(files))
    
    def create_dataset(self):
        #progressbar = ProgressbarFrame(self)
        progressbar_frame = ctk.CTkToplevel(self)
        # progressbar_frame.pack(side=ctk.BOTTOM)
        # progressbar_frame.grid_columnconfigure(0, weight=1)
        # progressbar_frame.grid_rowconfigure(1, weight=1)
        progressbar = ctk.CTkProgressBar(progressbar_frame, orientation='horizontal', mode='determinate')
        progressbar.configure(mode='determinate')
        
        partition = self.partition_var.get().lower()
        files = self.files_entry.get()
        dsname = self.dataset_entry.get()
        visible = self.visible_var.get()
        active = 1 if visible == 'visible' else 0
        progressbar.start()
        ds_result, metric_result, display_result = db_conn.CreateDatasetFromFiles(partition, dsname, files, active)
        progressbar.stop()
        # print(ds_result)
        # print(metric_result)
        # print(display_result)
        results = (ds_result, metric_result, display_result)
        self.results_window = DbResultsFrame(self, results)
        self.results_window.focus()
        #after this move to the create metric info frame.
        # prepopulate final table (dataset name) and dsid from this functions results
        known_info = {}
        known_info['final_table'] = self.dataset_entry.get()
        known_info['source_name'] = self.source_name_entry.get()
        known_info['source_url'] = self.source_url_entry.get()
        known_info['date_downloaded'] = self.date_entry.get()
        known_info['active'] = self.visible_var.get()

        self.metric_info_frame = CreateMetricInfoFrame(self, known_info)
        self.metric_info_frame.pack(fill=ctk.BOTH, expand=True, padx=10, pady=5)

        self.updates_window = DbUpdatesFrame(self)
        self.updates_window.focus()


class CreateTableFrame(ctk.CTkScrollableFrame):
    def __init__(self, parent):
        super().__init__(parent)
        self.parent = parent

        self.table_name_label = ctk.CTkLabel(self, text="Enter table name:")
        self.table_name_label.pack(fill=ctk.X, padx=10, pady=10)

        self.table_name_entry = ctk.CTkEntry(self)
        self.table_name_entry.pack(fill=ctk.X, padx=10, pady=5)

        files_btn = ctk.CTkButton(self, text="Choose Files", command=self.browse_files)
        files_btn.pack(fill=ctk.X, padx=10, pady=5)

        self.files_entry = ctk.CTkEntry(self)
        self.files_entry.pack(fill=ctk.X, padx=10, pady=5)

        self.submit_button = ctk.CTkButton(self, text="Submit", command=self.create_table)
        self.submit_button.pack(fill=ctk.X, padx=10, pady=5)

        self.results_window = None
        self.updates_window = None

    def browse_files(self):
        files = ctk.filedialog.askopenfilenames()
        self.files_entry.delete(0, ctk.END)
        self.files_entry.insert(0, ', '.join(files))
    
    def create_table(self):
        table_name = self.table_name_entry.get()
        files = self.files_entry.get()
        results = db_conn.CreateTableFromFile(table_name, files)

        self.results_window = DbResultsFrame(self, results)
        self.results_window.focus()

        self.updates_window = DbUpdatesFrame(self)
        self.updates_window.focus()

class RenameStreamCatMetricFrame(ctk.CTkScrollableFrame):
    def __init__(self, parent):
        super().__init__(parent)
        self.parent = parent

        self.metric_name_options = [] #self.get_metric_names()

        self.partition_label = ctk.CTkLabel(self, text="Is this update for a streamcat or lakecat metric?")
        self.partition_var = ctk.StringVar()
        self.partition_var.set('streamcat')
        self.partition_radio_streamcat = ctk.CTkRadioButton(self, text="StreamCat", variable=self.partition_var, value='streamcat', command=self.get_metric_names)
        self.partition_radio_streamcat.grid(row=1, column=0, padx=10, pady=5)
        self.partition_radio_lakecat = ctk.CTkRadioButton(self, text="LakeCat", variable=self.partition_var, value='lakecat', command=self.get_metric_names)
        self.partition_radio_lakecat.grid(row=1, column=1, padx=10, pady=5)

        # select all metrics from sc_metrics
        self.metric_name_options = self.get_metric_names()
        self.metric_name_var = ctk.StringVar()
        self.metric_name_var.set(self.metric_name_options[0])

        self.old_metric_label = ctk.CTkLabel(self, text="Select the metric you want to rename")
        self.old_metric_label.grid(row=2, column=0, padx=10, pady=5)
        

        self.metric_name_dropdown = CTkAutocompleteCombobox(self, width=280, variable=self.metric_name_var, completevalues=self.metric_name_options)
        self.metric_name_dropdown.grid(row=2, column=1, padx=10, pady=5)

        self.new_metric_label = ctk.CTkLabel(self, text="Enter new metric name (case sensitive)")
        self.new_metric_label.grid(row=3, column=0, padx=10, pady=5)
        self.new_name = ctk.CTkEntry(self, width=280)
        self.new_name.grid(row=3, column=1, padx=10, pady=5)

        self.info_label = ctk.CTkLabel(self, text="We will apply the changes to all tables as well as all years and aoi's associated with this metric")
        self.info_label.grid(row=4, column=0, padx=10, pady=5)

        self.specific_info = ctk.CTkLabel(self, text="VERY IMPORTANT!\n The naming convention is to have the first letter of the metric be capitalized and then use camel casing (no underscores) in the rest of the name.\n If the metric applies to multiple years add [Year] to the end of the name.\n Afterwards almost all metrics should have the [AOI] tag as well. The only reason it wouldn't is if the area of interest is `Other`.")
        self.specific_info.grid(row=5, column=0, padx=10, pady=5)

        self.submit_button = ctk.CTkButton(self, text="Submit", command=self.rename_metric)
        self.submit_button.grid(row=6, column=0, padx=10, pady=5)

        self.results_window = None
        self.updates_window = None

    def get_metric_names(self):
        self.metric_name_options.clear()
        table_name = 'sc_metrics_tg' if self.partition_var.get() == 'streamcat' else 'lc_metrics_tg'
        results = db_conn.SelectColsFromTable(['metric_name'], table_name, {'orderby': 'metric_name'})
        for row in results:
            self.metric_name_options.append(row._t[0])
        return self.metric_name_options

    
    def rename_metric(self):
        partition = self.partition_var.get()
        old_name = self.metric_name_var.get()
        new_name = self.new_name.get()
        results = db_conn.UpdateMetricName(partition, old_name, new_name)

        self.results_window = DbResultsFrame(self, results)
        self.results_window.focus()

        self.updates_window = DbUpdatesFrame(self)
        self.updates_window.focus()

class ActivateDatasetFrame(ctk.CTkScrollableFrame):
    def __init__(self, parent):
        super().__init__(parent)
        self.parent = parent

        self.partition_var = ctk.StringVar()
        self.partition_var.set('streamcat')
        self.partition_radio_streamcat = ctk.CTkRadioButton(self, text="StreamCat", variable=self.partition_var, value='streamcat', command=self.get_table_options)
        self.partition_radio_streamcat.pack(padx=10, pady=5)
        self.partition_radio_lakecat = ctk.CTkRadioButton(self, text="LakeCat", variable=self.partition_var, value='lakecat', command=self.get_table_options)
        self.partition_radio_lakecat.pack(padx=10, pady=5)
        self.partition_radio_both = ctk.CTkRadioButton(self, text="Both", variable=self.partition_var, value='both', command=self.get_table_options)
        self.partition_radio_both.pack(padx=10, pady=5)

        self.dsname_var = ctk.StringVar()
        self.dsname_options = self.get_table_options() # db_conn.GetAllDatasetNames()
        self.dsname_var.set(self.dsname_options[0])
        self.dsname_dropdown = ctk.CTkComboBox(self, variable=self.dsname_var, values=self.dsname_options, command=self.get_current_active_value)
        self.dsname_dropdown.pack(fill=ctk.X, expand=True, padx=10, pady=5)

        self.curr_active_label = ctk.CTkLabel(self, text="This dataset is currently: ")
        self.curr_active_label.pack(padx=10, pady=5)
        self.curr_active_val = ctk.CTkEntry(self, width=280)
        self.curr_active_val.pack(padx=10, pady=5)

        self.submit_button = ctk.CTkButton(self, text="Submit", command=self.update_active_dataset)
        self.submit_button.pack(side=ctk.BOTTOM, fill=ctk.X, padx=10, pady=5)

        self.results_window = None
        self.updates_window = None

    def get_table_options(self):
        options = []
        if self.partition_var.get() == 'both':
            options = db_conn.GetAllDatasetNames()
            
            
        else:
            table = 'sc_datasets' if self.partition_var.get() == 'streamcat' else 'lc_datasets'
            dsnames = db_conn.SelectColsFromTable(['dsname'], table)
            for row in dsnames:
                options.append(row._t[0])
        # full_list = list(db_conn.metadata.tables.keys())
        # if self.partition_var.get() == 'both':
        #     return full_list
        
        # prefix = 'sc_ds' if self.partition_var.get() == 'streamcat' else 'lc_ds'
        # options = [x for x in full_list if x.startswith(prefix)]
        return options

    def get_current_active_value(self, choice):
        table_name = 'sc_datasets' if self.partition_var.get() == 'streamcat' else 'lc_datasets'
        col = 'active'
        where = {'dsname': choice}
        curr_val = db_conn.SelectColWhere(table_name, col, where)
        for row in curr_val:
            active_val = 'Visible' if row._t[0] == 1 else 'Invisible'
            self.curr_active_val.insert(0, active_val)

    def update_active_dataset(self):
        dsname = self.dsname_var.get()
        partition = self.partition_var.get()
        results = db_conn.UpdateActiveDataset(dsname, partition)

        self.results_window = DbResultsFrame(self, results)
        #self.results_window.focus()

        self.updates_window = DbUpdatesFrame(self)
        self.updates_window.focus()

class UpdateTableFrame(ctk.CTkScrollableFrame):
    def __init__(self, parent):
        super().__init__(parent)
        self.parent = parent

        self.partition_var = ctk.StringVar()
        self.partition_var.set('streamcat')
        self.partition_radio_streamcat = ctk.CTkRadioButton(self, text="StreamCat", variable=self.partition_var, value='streamcat', command=self.get_table_options)
        self.partition_radio_streamcat.pack(padx=10, pady=5)
        self.partition_radio_lakecat = ctk.CTkRadioButton(self, text="LakeCat", variable=self.partition_var, value='lakecat', command=self.get_table_options)
        self.partition_radio_lakecat.pack(padx=10, pady=5)

        self.table_dropdown_label = ctk.CTkLabel(self, text="Select table to update:")
        self.table_dropdown_label.pack(padx=10, pady=5)

        self.table_var = ctk.StringVar()
        self.table_options = self.get_table_options() #list(db_conn.metadata.tables.keys())
        self.table_var.set(self.table_options[0])
        self.table_dropdown = ctk.CTkComboBox(self, variable=self.table_var, values=self.table_options)
        self.table_dropdown.pack(padx=10, pady=5)

        self.files_label = ctk.CTkLabel(self, text="Choose files:")
        self.files_label.pack(padx=10, pady=10)

        self.files_button = ctk.CTkButton(self, text="Browse", command=self.browse_files)
        self.files_button.pack(padx=10, pady=5)

        self.files_entry = ctk.CTkEntry(self)
        self.files_entry.pack(padx=10, pady=5)

        self.submit_button = ctk.CTkButton(self, text="Submit", command=self.update_table)
        self.submit_button.pack(side=ctk.BOTTOM, fill=ctk.X, padx=10, pady=5)

        self.results_window = None
        self.updates_window = None
    
    def get_table_options(self):
        prefix = 'sc' if self.partition_var.get() == 'streamcat' else 'lc'
        full_list = list(db_conn.metadata.tables.keys())
        options = [x for x in full_list if x.startswith(prefix)]
        return options

    def browse_files(self):
        files = ctk.filedialog.askopenfilenames()
        self.files_entry.delete(0, ctk.END)
        self.files_entry.insert(0, ', '.join(files))

    def update_table(self):
        table_name = self.table_var.get()
        file = self.files_entry.get()
        results = db_conn.BulkInsertFromFile(table_name, file)

        self.results_window = DbResultsFrame(self, results)
        self.results_window.focus()

        self.updates_window = DbUpdatesFrame(self)
        self.updates_window.focus()

class CreateMetricInfoFrame(ctk.CTkScrollableFrame):
    def __init__(self, parent, known_info=None):
        super().__init__(parent)
        self.parent = parent

        self.metric_name_label = ctk.CTkLabel(self, text="Enter new metric name:")
        self.metric_name_label.grid(row=2, column=0, padx=10, pady=5)

        self.metric_name_entry = ctk.CTkEntry(self, width=280)
        self.metric_name_entry.grid(row=2, column=1, columnspan=2, padx=10, pady=5)
    
        self.category_label = ctk.CTkLabel(self, text="Select metric category:")
        self.category_label.grid(row=3, column=0, padx=10, pady=5)

        self.category_dropdown_var = ctk.StringVar()
        self.category_dropdown = ctk.CTkComboBox(self, width=280, variable=self.category_dropdown_var, values=["Base", "Natural", "Anthropogenic"])
        self.category_dropdown.grid(row=3, column=1, columnspan=2, padx=10, pady=5)
        

        self.aoi_label = ctk.CTkLabel(self, text="Select all AOIs for the metric:")
        self.aoi_label.grid(row=4, column=0, padx=10, pady=5)
        
        self.aoi_values = ["Cat", "Ws", "CatRp100", "WsRp100", "Other"]
        self.aoi_listbox = CTkListbox(self, multiple_selection=True)
        for i, aoi in enumerate(self.aoi_values):
            self.aoi_listbox.insert(i, aoi)
        self.aoi_listbox.grid(row=4, column=1, columnspan=2, rowspan=5, padx=10, pady=5)
        
        
        self.year_label = ctk.CTkLabel(self, text="Enter comma seperated list of years (if available):")
        self.year_label.grid(row=9, column=0, padx=10, pady=5)
        
        self.year_entry = ctk.CTkEntry(self, width=280)
        self.year_entry.grid(row=9, column=1, columnspan=2, padx=10, pady=5)

        self.webtool_label = ctk.CTkLabel(self, text="Enter Webtool Name:")
        self.webtool_label.grid(row=10, column=0, padx=10, pady=5)
        self.webtool_entry = ctk.CTkEntry(self, width=280)
        self.webtool_entry.grid(row=10, column=1, columnspan=2, padx=10, pady=5)

        self.description_label = ctk.CTkLabel(self, text="Enter metric description:")
        self.description_label.grid(row=11, column=0, padx=10, pady=5)
        self.description_entry = ctk.CTkEntry(self, width=280)
        self.description_entry.grid(row=11, column=1, columnspan=2, padx=10, pady=5)

        self.units_label = ctk.CTkLabel(self, text="Enter metric units:")
        self.units_label.grid(row=12, column=0, padx=10, pady=5)
        
        self.units_options = []
        self.units_results = db_conn.SelectColsFromTable(['metric_units'], 'sc_metrics_tg', {'distinct': 'metric_units'})
        for unit in self.units_results:
            self.units_options.append(unit._t[0])
        self.units_var = ctk.StringVar()
        self.units_entry = CTkAutocompleteCombobox(self, width=280, completevalues=self.units_options, variable=self.units_var) 
        self.units_entry.grid(row=12, column=1, columnspan=2, padx=10, pady=5)

        self.uuid_label = ctk.CTkLabel(self, text="Enter metric uuid:")
        self.uuid_label.grid(row=13, column=0, padx=10, pady=5)
        self.uuid_entry = ctk.CTkEntry(self, width=280)
        self.uuid_entry.grid(row=13, column=1, columnspan=2, padx=10, pady=5)

        self.metadata_label = ctk.CTkLabel(self, text="Enter metric metadata:")
        self.metadata_label.grid(row=14, column=0, padx=10, pady=5)
        self.metadata_entry = ctk.CTkEntry(self, width=280)
        self.metadata_entry.grid(row=14, column=1, columnspan=2, padx=10, pady=5)

        self.source_name_label = ctk.CTkLabel(self, text="Enter metric source name:")
        self.source_name_label.grid(row=15, column=0, padx=10, pady=5)
        self.source_name_entry = ctk.CTkEntry(self, width=280)
        self.source_name_entry.grid(row=15, column=1, columnspan=2, padx=10, pady=5)

        self.source_url_label = ctk.CTkLabel(self, text="Enter metric source url:")
        self.source_url_label.grid(row=16, column=0, padx=10, pady=5)
        self.source_url_entry = ctk.CTkEntry(self, width=280)
        self.source_url_entry.grid(row=16, column=1, columnspan=2, padx=10, pady=5)

        self.date_label = ctk.CTkLabel(self, text="Enter the date the dataset was downloaded with the format dd-MM-YY:")
        self.date_label.grid(row=17, column=0, padx=10, pady=5)
        self.date_entry = ctk.CTkEntry(self, width=280)
        self.date_entry.grid(row=17, column=1, columnspan=2, padx=10, pady=5)

        self.dsid_label = ctk.CTkLabel(self, text="Enter metric dsid:")
        self.dsid_label.grid(row=18, column=0, padx=10, pady=5)
        self.dsid_entry = ctk.CTkEntry(self, width=280)
        self.dsid_entry.grid(row=18, column=1, columnspan=2, padx=10, pady=5)

        self.dataset_name_label = ctk.CTkLabel(self, text="Enter metric dataset name (aka final_table):")
        self.dataset_name_label.grid(row=19, column=0, padx=10, pady=5)
         
        self.dataset_name_entry = ctk.CTkEntry(self, width=280)
        self.dataset_name_entry.grid(row=19, column=1, columnspan=2, padx=10, pady=5)

        self.partition_label = ctk.CTkLabel(self, text="What partition is the metric in?")
        self.partition_label.grid(row=20, column=0, padx=10, pady=10)
        
        self.selected_partition = ctk.StringVar(value='streamcat')
        self.streamcat_radio = ctk.CTkRadioButton(self, text="StreamCat", variable=self.selected_partition, value='streamcat')
        self.streamcat_radio.grid(row=20, column=1, sticky='w', padx=10, pady=5)
        self.lakecat_radio = ctk.CTkRadioButton(self, text="LakeCat", variable=self.selected_partition, value='lakecat')
        self.lakecat_radio.grid(row=20, column=2, sticky='w', padx=10, pady=5)

        self.submit_button = ctk.CTkButton(self, text="Submit", command=self.create_metric_info)
        self.submit_button.grid(row=21, column=0, columnspan=2, padx=10, pady=5)

        self.results_window = None
        self.updates_window = None
        if known_info is not None:
            self.fill_known_values(known_info)

    def fill_known_values(self, known_vals: dict):
        keys = known_vals.keys()
        if 'final_table' in keys:
            self.dataset_name_entry.insert(0, known_vals['final_table'])

        if 'source_name' in keys:
            self.source_name_entry.insert(0, known_vals['source_name'])

        if 'source_url' in keys:
            self.source_url_entry.insert(0, known_vals['source_url'])

        if 'date_downloaded' in keys:
            self.date_entry.insert(0, known_vals['date_downloaded'])

    def create_metric_info(self):
        print("Creating metric info card")
        metric_data = {}
        #metric_data['partition'] = self.selected_partition.get()
        table_name = 'sc_metrics_tg' if self.selected_partition.get() == 'streamcat' else 'lc_metrics_tg'
        metric_data['metric_name'] = self.metric_name_entry.get()
        metric_data['indicator_category'] = self.category_dropdown_var.get()
        metric_data['aoi'] = self.aoi_listbox.get()
        metric_data['year'] = self.year_entry.get()
        metric_data['webtool_name'] = self.webtool_entry.get()
        metric_data['description'] = self.description_entry.get()
        metric_data['units'] = self.units_entry.get()
        metric_data['uuid'] = self.uuid_entry.get()
        metric_data['metadata'] = self.metadata_entry.get()
        metric_data['source_name'] = self.source_name_entry.get()
        metric_data['source_url'] = self.source_url_entry.get()
        metric_data['date_downloaded'] = self.date_entry.get()
        metric_data['dsid'] = self.dsid_entry.get()
        results = db_conn.InsertRow(table_name, metric_data)

        self.results_window = DbResultsFrame(self, results)
        self.results_window.focus()

        self.updates_window = DbUpdatesFrame(self)
        self.updates_window.focus()
        
class EditMetricInfoFrame(ctk.CTkScrollableFrame):
    def __init__(self, parent):
        super().__init__(parent)
        self.parent = parent
        self.partition_label = ctk.CTkLabel(self, text="What partition would you like to use?")
        self.partition_label.grid(row=1, column=0, padx=10, pady=5)

        self.partition_var = ctk.StringVar()
        self.partition_var.set('streamcat')
        self.partition_radio_streamcat = ctk.CTkRadioButton(self, text="StreamCat", variable=self.partition_var, value='streamcat')
        self.partition_radio_streamcat.grid(row=1, column=1, padx=10, pady=5)
        self.partition_radio_lakecat = ctk.CTkRadioButton(self, text="LakeCat", variable=self.partition_var, value='lakecat')
        self.partition_radio_lakecat.grid(row=1, column=2, padx=10, pady=5)

        metric_name_options, tg_columns = self.get_edit_widget_data()

        self.metric_name_label = ctk.CTkLabel(self, text="What metric variable info do you want to edit?")
        self.metric_name_label.grid(row=2, column=0, padx=10, pady=5)

        self.metric_name_var = ctk.StringVar()
        self.metric_name_dropdown = ctk.CTkComboBox(self, width=280, variable=self.metric_name_var, values=metric_name_options)
        self.metric_name_dropdown.grid(row=2, column=1, padx=10, pady=5)

        self.tg_col_label = ctk.CTkLabel(self, text="Which value needs to be edited?")
        self.tg_col_label.grid(row=3, column=0, padx=10, pady=5)

        self.tg_col_var = ctk.StringVar()
        self.tg_col_dropdown = ctk.CTkComboBox(self, width=280, variable=self.tg_col_var, values=tg_columns, command=self.get_current_metric_value)
        self.tg_col_dropdown.grid(row=3, column=1, padx=10, pady=5)

        self.tg_curr_val_label = ctk.CTkLabel(self, text="Current value for selected metric attribute")
        self.tg_curr_val_label.grid(row=4, column=0, padx=10, pady=5)
        self.tg_col_curr_val = ctk.CTkEntry(self, width=280)
        self.tg_col_curr_val.grid(row=4, column=1, padx=10, pady=5)

        self.new_val_label = ctk.CTkLabel(self, text="Enter new value for selected metric value: ")
        self.new_val_label.grid(row=5, column=0, padx=10, pady=5)

        self.new_val_entry = ctk.CTkEntry(self, width=280)
        self.new_val_entry.grid(row=5, column=1, padx=10, pady=5)

        self.submit_button = ctk.CTkButton(self, text="Submit", command=self.edit_metric_info)
        self.submit_button.grid(row=6, column=0, padx=10, pady=5)
        
        self.results_window = None
        self.updates_window = None

    def get_edit_widget_data(self):
        table_name = 'sc_metrics_tg' if self.partition_var.get() == 'streamcat' else 'lc_metrics_tg'
        
        metric_name_results = db_conn.SelectColsFromTable(['metric_name'], table_name)
        metric_name_options = []
        for row in metric_name_results:
            metric_name_options.append(row._t[0])
        
        tg_columns = list(db_conn.metadata.tables[table_name].c.keys())

        return metric_name_options, tg_columns

    def get_current_metric_value(self, choice):
        table_name = 'sc_metrics_tg' if self.partition_var.get() == 'streamcat' else 'lc_metrics_tg'
        col = self.tg_col_dropdown.get()
        metric = self.metric_name_var.get()
        curr_val = db_conn.SelectColWhere(table_name, col, {'metric_name': metric})
        idx = 0
        for row in curr_val:
            self.tg_col_curr_val.insert(idx, str(row._t[0]))
            idx += len(str(row._t))

    def edit_metric_info(self):
        table_name = 'sc_metrics_tg' if self.partition_var.get() == 'streamcat' else 'lc_metrics_tg'
        col_name = self.tg_col_var.get()
        metric_name = self.metric_name_var.get()
        new_val = self.new_val_entry.get()
        results = db_conn.UpdateRow(table_name, col_name, 'metric_name', metric_name, new_val)

        self.results_window = DbResultsFrame(self, results)
        self.results_window.focus()

        self.updates_window = DbUpdatesFrame(self)
        self.updates_window.focus()


class DatabaseApp(ctk.CTk):
    def __init__(self):
        super().__init__()
        #self = root

        self.title("StreamCat GUI (modern)")
        # self.geometry("700x450")
        self.width = int(self.winfo_screenwidth()/2)
        self.height = int(self.winfo_screenheight()/1.5)
        self.geometry(f"{self.width}x{self.height}")
        self.minsize(500,500)
        self.action_frames = {}
        self.current_frame = None

        self.action_var = ctk.StringVar()
        # self.action_var.set('Create Dataset')

        self.actions = [
            'Create Dataset',
            'Create Table',
            'Rename Metric',
            'Activate/Deactivate Dataset',
            'Add File Data to Table',
            'Create Metric Info',
            'Edit Metric Info'
        ]
        self.action_var.set(self.actions[0]) # could add a default / info frame to be actions[0] called '--'
        self.action_dropdown = ctk.CTkComboBox(self, width=200, variable=self.action_var, values=self.actions)
        self.action_dropdown.pack(side=ctk.TOP, padx=10, pady=10)

        self.action_button = ctk.CTkButton(self, text="Go", command=self.show_frame)
        self.action_button.pack(side=ctk.TOP, padx=10, pady=5)

        self.execute_sql_var = ctk.BooleanVar()
        self.execute_sql_var.set(db_conn.execute)
        self.execute_sql_switch = ctk.CTkCheckBox(self, text="Execute SQL?", variable=self.execute_sql_var, command=lambda: setattr(db_conn, 'execute', self.execute_sql_var.get()))
        self.execute_sql_switch.pack(side=ctk.RIGHT, padx=10, pady=5)

        self.show_frame()

    def show_frame(self):
        action = self.action_var.get()
        if action not in self.action_frames:
            if action == 'Create Dataset':
                self.action_frames[action] = CreateDatasetFrame(self)
            elif action == 'Create Table':
                self.action_frames[action] = CreateTableFrame(self)
            elif action == 'Rename Metric':
                self.action_frames[action] = RenameStreamCatMetricFrame(self)
            elif action == 'Activate/Deactivate Dataset':
                self.action_frames[action] = ActivateDatasetFrame(self)
            elif action == 'Add File Data to Table':
                self.action_frames[action] = UpdateTableFrame(self)
            elif action == 'Create Metric Info':
                self.action_frames[action] = CreateMetricInfoFrame(self)
            elif action == 'Edit Metric Info':
                self.action_frames[action] = EditMetricInfoFrame(self)
            

        if self.current_frame:
            self.current_frame.pack_forget()

        self.current_frame = self.action_frames[action]
        self.current_frame.pack(fill='both', expand=True)

if __name__ == '__main__':
    db_conn = DatabaseConnection()
    db_conn.connect()
    app = DatabaseApp()
    app.mainloop()