main_screen.py

Go to the documentation of this file.

from textual.screen import Screen
from textual.widgets import Footer
from textual.binding import Binding
 
# Textual OKS imports
from daqconf.cider.widgets.custom_rich_log import RichLogWError
from daqconf.cider.widgets.config_table import ConfigTable
from daqconf.cider.widgets.configuration_controller import ConfigurationController
from daqconf.cider.widgets.popups.file_io import SaveWithMessageScreen, OpenFileScreen
from daqconf.cider.widgets.popups.dropdown_selector import SelectSessionScreen
from daqconf.cider.app_structures.selection_panel import SelectionPanel
from daqconf.cider.widgets.popups.quit_screen import QuitScreen
from daqconf.cider.widgets.popups.config_object_modifier_screen import ConfigObjectModifierScreen
from daqconf.cider.widgets.popups.add_objects import AddNewObjectScreen
from daqconf.cider.widgets.popups.delete_object_screen import DeleteConfigObjectScreen
from daqconf.cider.widgets.popups.file_io import RenameConfigObjectScreen
 
from os import path
 
 
class MainScreen(Screen):
    """Main screen for navigating python DBE
    """
    
    # Key binds
    BINDINGS = [
                # Binding("ctrl+s", "save_configuration", "Save Configuration"),
                Binding("ctrl+s", "save_configuration_with_message", "Save Configuration"),
                Binding("o", "open_configuration", "Open Configuration"),
                Binding("ctrl+q", "request_quit", "Exit Cider"),
                Binding("e", "modify_relations", "Modify Relation to Object"),
                Binding("r", "rename_configuration", "Rename Conf Object"),
                Binding("d", "toggle_disable", "Toggle Disable"),
                Binding("a", "add_configuration", "Add Conf Object"),
                Binding("ctrl+d", "destroy_configuration", "Delete Conf Object"),
            ]
    
    _config_controller = None
    _init_input = None
    
    def __make_logger(self, splash: bool=False):
        self.logger = RichLogWError(id="main_log", highlight=True, markup=True)
        
        # Splash screen
        if splash:
            self.logger.write("[red]========================================================================")
            self.logger.write("    [bold yellow]Welcome to CIDER![/bold yellow]")
            self.logger.write("    [green]This is a work in progress, please use with[/green] [bold red]caution![/bold red]")
            self.logger.write("[red]========================================================================\n\n")
 
    async def on_mount(self):
        """Mount widgets with the logger appearing first"""
        self.__make_logger(splash=True)
        await self.mount(self.logger)  # Mount the logger first
 
        # Mount other widgets
        if self._config_controller is None:
            self._config_controller = ConfigurationController()
        await self.mount(self._config_controller)
        await self.mount(Footer())
        
        if self._init_input is not None:
            self.update_with_new_input(self._init_input)
 
    def set_initial_input_file(self, input_file: str):
        self._init_input = input_file
 
    def update_with_new_input(self, input_file_name: str):
        '''
        Update main screen to have a new input file.
        '''
        self._init_input = input_file_name
        
        try:
            self._config_controller.new_handler_from_str(input_file_name)
        except Exception as e:
            self.logger.write_error(e)
            return
 
        # Add interfaces
        self._config_controller.add_interface("class-selection")
        self._config_controller.add_interface("relation-selection")
        
        # Mount the selection panel
        try:
            self.mount(SelectionPanel())
        except Exception as e:
            raise e
 
        # Mount config table
        try:
            config_table = self.query_one(ConfigTable)
            config_table.update_table(self._config_controller.current_dal)
        except:
            config_table = ConfigTable(id="main_table")
            self.mount(config_table)
 
        # Refresh the screen for safety
        self.refresh()
        
        # Get logger (defined at the start)
        
        # Get the current database name
        current_database_path = self._config_controller.configuration.databases[0]
        data_base_name = path.basename(current_database_path)
        
        # Print everything!
        self.logger.write(f"[bold green]Opened new configuration file: [/bold green][bold red]{data_base_name}[/bold red][bold green].\nConnected databases are:[/bold green]\n" \
                     + "".join([f"   - [red]{db}[/red] \n" for db in self._config_controller.configuration.get_includes()]))
            
    def on_configuration_controller_changed(self, event):
        """Updates table based on global state of the configuration controller
        """        
        config_table = self.query_one(ConfigTable)
        if config_table is not None:
            config_table.update_table(event.dal)
        
    def action_save_configuration(self)->None:
        """Save current configuration
        """        
        config = self.query_one(ConfigurationController)
        config.commit_configuration("Update configuration")
 
    def action_save_configuration_with_message(self)->None:
        """Save current configuration with an update message
        """        
        self.app.push_screen(SaveWithMessageScreen())
        
    async def action_open_configuration(self) -> None:
        """Activate open file splash screen
        """        
        # Push the OpenFileScreen and wait for it to be closed
        await self.app.push_screen(OpenFileScreen())
 
    async def action_toggle_disable(self)->None:
        """Toggle disable on the selected configuration object
        """        
        if self._config_controller.can_be_disabled():
            await self.app.push_screen(SelectSessionScreen())
        
        else:        
            # except:
            self.query_one(RichLogWError).write_error("Could not toggle disable configuration object")
 
    
    def call_quit_handler(self):
        """Call the quit handler just like action_request_quit."""
        self.app.push_screen(QuitScreen())  # Show the quit confirmation screen
 
    async def action_request_quit(self)->None:
        """Quit TDBE
        """
        self.call_quit_handler()
 
    def handle_sigint(self, signum, frame):
        # In the event quit is done with ctrl+c
        self.call_quit_handler()
        
    async def action_modify_relations(self)->None:
        self.app.push_screen(ConfigObjectModifierScreen())     
 
    async def action_add_configuration(self)->None:
        try:
            self.app.push_screen(AddNewObjectScreen())
        except Exception as e:
            self.query_one(RichLogWError).write_error(e)
    
    async def action_destroy_configuration(self)->None:
        try:
            self.app.push_screen(DeleteConfigObjectScreen())
        except Exception as e:
            self.query_one(RichLogWError).write_error(e)
    
    async def action_rename_configuration(self)->None:
        self.app.push_screen(RenameConfigObjectScreen())