meerschaum 2.9.0rc3__py3-none-any.whl → 2.9.2__py3-none-any.whl

meerschaum/_internal/shell/Shell.py

@@ -10,6 +10,7 @@ import os
 from copy import deepcopy
 from itertools import chain
 import shlex
+import shutil
 
 from meerschaum.utils.typing import Union, SuccessTuple, Any, Callable, Optional, List, Dict
 from meerschaum.utils.packages import attempt_import
@@ -309,7 +310,7 @@ class Shell(cmd.Cmd):
         try:
             for c in hidden_commands:
                 self.hidden_commands.append(c)
-        except Exception as e:
+        except Exception:
             pass
 
         ### Finally, spawn the version update thread.
@@ -387,8 +388,9 @@ class Shell(cmd.Cmd):
         username: Optional[str] = None,
         executor_keys: Optional[str] = None,
     ):
-        from meerschaum.utils.formatting import ANSI, colored
+        from meerschaum.utils.formatting import ANSI, colored, UNICODE, get_console
         from meerschaum._internal.entry import _shell, get_shell
+        from meerschaum.utils.misc import truncate_text_for_display, remove_ansi
 
         cmd.__builtins__['input'] = input_with_sigint(
             _old_input,
@@ -414,7 +416,6 @@ class Shell(cmd.Cmd):
 
         if '{username}' in shell_attrs['_prompt']:
             if username is None:
-                from meerschaum.utils.misc import remove_ansi
                 from meerschaum.connectors.parse import parse_instance_keys
                 from meerschaum.connectors.sql import SQLConnector
                 try:
@@ -422,15 +423,19 @@ class Shell(cmd.Cmd):
                         remove_ansi(shell_attrs['instance_keys']),
                         construct=False,
                     )
-                    if 'username' not in conn_attrs:
-                        if 'uri' in conn_attrs:
+                    if conn_attrs and 'username' in conn_attrs:
+                        username = conn_attrs.get('username', '(no username)')
+                    elif conn_attrs and 'uri' in conn_attrs:
+                        try:
                             username = SQLConnector.parse_uri(conn_attrs['uri'])['username']
+                        except Exception:
+                            username = '(no username)'
                     else:
-                        username = conn_attrs['username']
+                        username = '(no username)'
                 except KeyError:
                     username = '(no username)'
-                except Exception as e:
-                    username = str(e)
+                except Exception:
+                    username = None
                 if username is None:
                    username = '(no username)'
             shell_attrs['username'] = (
@@ -463,12 +468,32 @@ class Shell(cmd.Cmd):
                     _c = colored(_c, **get_config('shell', 'ansi', 'prompt', 'rich'))
                 remainder_prompt[i] = _c
 
+        truncation_suffix = (
+            '…'
+            if UNICODE
+            else '...'
+        )
+        console = get_console()
+        truncation_kwargs = {
+            'suffix': truncation_suffix,
+            'max_length': int(console.size.width / 4),
+        }
+
         self.prompt = ''.join(remainder_prompt).replace(
-            '{username}', shell_attrs['username']
+            '{username}', truncate_text_for_display(
+                shell_attrs['username'],
+                **truncation_kwargs
+            )
         ).replace(
-            '{instance}', shell_attrs['instance']
+            '{instance}', truncate_text_for_display(
+                shell_attrs['instance'],
+                **truncation_kwargs
+            )
         ).replace(
-            '{executor_keys}', shell_attrs['executor_keys']
+            '{executor_keys}', truncate_text_for_display(
+                shell_attrs['executor_keys'],
+                **truncation_kwargs
+            )
         )
         shell_attrs['prompt'] = self.prompt
         ### flush stdout
@@ -1018,15 +1043,14 @@ def input_with_sigint(_input, session, shell: Optional[Shell] = None):
     """
     Replace built-in `input()` with prompt_toolkit.prompt.
     """
-    from meerschaum.utils.formatting import CHARSET, ANSI, colored
+    from meerschaum.utils.formatting import CHARSET, ANSI, colored, UNICODE
     from meerschaum.connectors import is_connected, connectors
-    from meerschaum.utils.misc import remove_ansi
+    from meerschaum.utils.misc import remove_ansi, truncate_text_for_display
     from meerschaum.config import get_config
     import platform
     if shell is None:
         from meerschaum.actions import get_shell
         shell = get_shell()
-
     style = prompt_toolkit_styles.Style.from_dict({
         'bottom-toolbar': 'black',
     })
@@ -1039,46 +1063,71 @@ def input_with_sigint(_input, session, shell: Optional[Shell] = None):
             return shell_attrs['_old_bottom_toolbar']
         size = os.get_terminal_size()
         num_cols, num_lines = size.columns, size.lines
-
+        truncation_suffix = (
+            '…'
+            if UNICODE
+            else '...'
+        )
+        truncation_kwargs = {
+            'suffix': truncation_suffix,
+            'max_length': int(num_cols / 4),
+        }
+        instance_text = truncate_text_for_display(
+            remove_ansi(shell_attrs['instance_keys']),
+            **truncation_kwargs
+        )
         instance_colored = (
             colored(
-                remove_ansi(shell_attrs['instance_keys']),
+                instance_text,
                 'on ' + get_config('shell', 'ansi', 'instance', 'rich', 'style')
             )
             if ANSI
-            else colored(shell_attrs['instance_keys'], 'on white')
+            else colored(instance_text, 'on white')
+        )
+        repo_text = truncate_text_for_display(
+            remove_ansi(shell_attrs['repo_keys']),
+            **truncation_kwargs
         )
         repo_colored = (
             colored(
-                remove_ansi(shell_attrs['repo_keys']),
+                repo_text,
                 'on ' + get_config('shell', 'ansi', 'repo', 'rich', 'style')
             )
             if ANSI
             else colored(shell_attrs['repo_keys'], 'on white')
         )
+        executor_text = truncate_text_for_display(
+            remove_ansi(shell_attrs['executor_keys']),
+            **truncation_kwargs
+        )
         executor_colored = (
             colored(
-                remove_ansi(shell_attrs['executor_keys']),
+                executor_text,
                 'on ' + get_config('shell', 'ansi', 'executor', 'rich', 'style')
             )
             if ANSI
-            else colored(remove_ansi(shell_attrs['executor_keys']), 'on white')
+            else colored(executor_text, 'on white')
         )
 
         try:
             typ, label = shell_attrs['instance_keys'].split(':', maxsplit=1)
             connected = typ in connectors and label in connectors[typ]
-        except Exception as e:
+        except Exception:
             connected = False
         last_connected = connected
-        connected_str = (('dis' if not connected else '') + 'connected')
+        connected_str = truncate_text_for_display(
+            ('dis' if not connected else '') + 'connected',
+            **truncation_kwargs
+        )
+        connected_icon = get_config(
+            'formatting', connected_str, CHARSET, 'icon', warn=False,
+        ) or ''
         connection_text = (
-            get_config(
-                'formatting', connected_str, CHARSET, 'icon'
-            ) + ' ' + (
-                colored(connected_str.capitalize(), 'on ' + get_config(
-                    'formatting', connected_str, 'ansi', 'rich', 'style'
-                ) + '  ') if ANSI else (colored(connected_str.capitalize(), 'on white') + ' ')
+            connected_icon + ' ' + (
+                colored(connected_str.capitalize(), 'on ' + (get_config(
+                    'formatting', connected_str, 'ansi', 'rich', 'style',
+                    warn=False,
+                ) or '') + '  ') if ANSI else (colored(connected_str.capitalize(), 'on white') + ' ')
             )
         )
 
@@ -1104,8 +1153,8 @@ def input_with_sigint(_input, session, shell: Optional[Shell] = None):
         _args = []
         for a in args:
             try:
-                _a = prompt_toolkit_formatted_text.ANSI(a)
-            except Exception as e:
+                _a = prompt_toolkit_formatted_text.ANSI(a) if isinstance(a, str) else str(a)
+            except Exception:
                 _a = a
             _args.append(_a)
         try:

meerschaum/api/dash/callbacks/__init__.py

@@ -6,13 +6,16 @@
 Define callbacks for pages.
 """
 
+from meerschaum.api import debug as _debug
 import meerschaum.api.dash.callbacks.dashboard
 import meerschaum.api.dash.callbacks.login
 import meerschaum.api.dash.callbacks.plugins
 import meerschaum.api.dash.callbacks.jobs
 import meerschaum.api.dash.callbacks.register
 import meerschaum.api.dash.callbacks.pipes
+import meerschaum.api.dash.callbacks.settings
 from meerschaum.api.dash.callbacks.custom import init_dash_plugins, add_plugin_pages
 
-init_dash_plugins()
-add_plugin_pages()
+init_dash_plugins(debug=_debug)
+add_plugin_pages(debug=_debug)
+

meerschaum/api/dash/callbacks/custom.py

@@ -11,12 +11,12 @@ from typing import Any, Dict
 
 from meerschaum.api.dash import dash_app
 from meerschaum.plugins import _dash_plugins, _plugin_endpoints_to_pages
-from meerschaum.utils.warnings import warn
+from meerschaum.utils.warnings import warn, dprint
 from meerschaum.api.dash.callbacks.dashboard import _paths, _required_login, _pages
 from meerschaum.api.dash.components import pages_navbar
 
 
-def init_dash_plugins():
+def init_dash_plugins(debug: bool = False):
     """
     Fire the initial callbacks for Dash plugins.
     """
@@ -32,29 +32,21 @@ def init_dash_plugins():
                 )
 
 
-def add_plugin_pages():
+def add_plugin_pages(debug: bool = False):
     """
     Allow users to add pages via the `@web_page` decorator.
     """
-    for _endpoint, _page_dict in _plugin_endpoints_to_pages.items():
-        page_layout = _page_dict['function']()
-        if not _page_dict['skip_navbar']:
-            if isinstance(page_layout, list):
-                page_layout = [pages_navbar] + page_layout
-            else:
-                page_layout = [pages_navbar, page_layout]
-        page_key = (
-            ' '.join(
-                [
-                    word.capitalize()
-                    for word in (
-                        _endpoint.replace('/dash', '').lstrip('/').rstrip('/').strip()
-                        .replace('-', ' ').replace('_', ' ').split(' ')
-                    )
-                ]
-            )
-        )
-        _pages[page_key] = _endpoint
-        _paths[_endpoint] = page_layout
-        if _page_dict['login_required']:
-            _required_login.add(_endpoint)
+    for plugin_name, pages_dicts in _plugin_endpoints_to_pages.items():
+        if debug:
+            dprint(f"Adding pages from plugin '{plugin_name}'...")
+        for _endpoint, _page_dict in pages_dicts.items():
+            page_layout = _page_dict['function']()
+            if not _page_dict['skip_navbar']:
+                if isinstance(page_layout, list):
+                    page_layout = [pages_navbar] + page_layout
+                else:
+                    page_layout = [pages_navbar, page_layout]
+            _pages[_page_dict['page_key']] = _endpoint
+            _paths[_endpoint] = page_layout
+            if _page_dict['login_required']:
+                _required_login.add(_endpoint)

meerschaum/api/dash/callbacks/dashboard.py

@@ -33,7 +33,10 @@ from meerschaum.api.dash.users import get_users_cards
 from meerschaum.api.dash.graphs import get_graphs_cards
 from meerschaum.api.dash.webterm import get_webterm
 from meerschaum.api.dash.components import (
-    alert_from_success_tuple, console_div, build_cards_grid,
+    alert_from_success_tuple,
+    console_div,
+    build_cards_grid,
+    build_pages_offcanvas_children,
 )
 from meerschaum.api.dash import pages
 from meerschaum.utils.typing import Dict
@@ -1093,26 +1096,7 @@ def toggle_pages_offcanvas(n_clicks: Optional[int], is_open: bool):
     """
     Toggle the pages sidebar.
     """
-    pages_children = dbc.Card(
-        dbc.ListGroup(
-            [
-                dbc.ListGroupItem(
-                    dbc.Button(
-                        html.P(
-                            ' '.join([word.capitalize() for word in page_key.split(' ')]),
-                            style={'text-decoration': 'none', 'fontSize': '18px'},
-                        ),
-                        style={'width': '100%', 'text-align': 'left'},
-                        href=page_href,
-                        color='dark',
-                    )
-                )
-                for page_key, page_href in _pages.items()
-            ],
-            flush=True,
-        ),
-        outline=True,
-    )
+    pages_children = build_pages_offcanvas_children()
     if n_clicks:
         return not is_open, pages_children
     return is_open, pages_children

meerschaum/api/dash/callbacks/settings/__init__.py

@@ -0,0 +1,8 @@
+#! /usr/bin/env python3
+# vim:fenc=utf-8
+
+"""
+Define the callbacks for the settings pages.
+"""
+
+import meerschaum.api.dash.callbacks.settings.password_reset

meerschaum/api/dash/callbacks/settings/password_reset.py

@@ -0,0 +1,76 @@
+#! /usr/bin/env python3
+# vim:fenc=utf-8
+
+"""
+Define the callbacks for the password reset page.
+"""
+
+import dash
+from dash.exceptions import PreventUpdate
+from dash.dependencies import Input, Output, State
+import dash_bootstrap_components as dbc
+
+from meerschaum.core.User import User
+from meerschaum.config.static import STATIC_CONFIG
+from meerschaum.api import get_api_connector, debug
+from meerschaum.api.dash import dash_app
+from meerschaum.api.dash.components import alert_from_success_tuple
+from meerschaum.api.dash.sessions import get_username_from_session
+
+
+@dash_app.callback(
+    Output('password-reset-alert-div', 'children'),
+    Input('password-reset-button', 'n_clicks'),
+    State('password-reset-input', 'value',),
+    State('session-store', 'data'),
+)
+def password_reset_button_click(n_clicks, new_password_value, session_store_data):
+    """
+    Attempt the password reset with the form data.
+    """
+    if not n_clicks:
+        raise PreventUpdate
+
+    session_id = session_store_data.get('session-id', None)
+    username = get_username_from_session(session_id)
+    if not username:
+        return alert_from_success_tuple(
+            (False, "Invalid session. Are you logged in correctly?")
+        )
+
+    instance_connector = get_api_connector()
+    user = User(username, new_password_value)
+    success, msg = instance_connector.edit_user(user, debug=debug)
+    return alert_from_success_tuple((success, msg))
+
+
+@dash_app.callback(
+    Output('password-reset-input', 'valid'),
+    Output('password-reset-input', 'invalid'),
+    Input('password-reset-input', 'value'),
+)
+def validate_new_password(new_password_value):
+    if not new_password_value:
+        raise PreventUpdate
+
+    valid = (len(new_password_value) >= STATIC_CONFIG['users']['min_password_length'])
+    return valid, not valid
+ 
+
+@dash_app.callback(
+    Output("password-reset-confirm-input", "valid"),
+    Output("password-reset-confirm-input", "invalid"),
+    Output("password-reset-button", 'disabled'),
+    Input("password-reset-confirm-input", "value"),
+    State("password-reset-input", "value"),
+)
+def validate_new_passwords_match(confirm_password_value, new_password_value):
+    if not confirm_password_value:
+        raise PreventUpdate
+
+    new_password_is_valid, _ = validate_new_password(new_password_value)
+    if not new_password_is_valid:
+        raise PreventUpdate
+
+    valid = confirm_password_value == new_password_value
+    return valid, not valid, not valid

meerschaum/api/dash/components.py

@@ -110,6 +110,13 @@ instance_select = dbc.Select(
     class_name='dbc_dark custom-select custom-select-sm',
 )
 
+sign_out_button = dbc.Button(
+    "Sign out",
+    color='link',
+    style={'margin-left': '30px'},
+    id='sign-out-button',
+)
+
 logo_row = dbc.Row(
     [
         dbc.Col(
@@ -128,7 +135,30 @@ logo_row = dbc.Row(
 pages_navbar = html.Div(
     [
         pages_offcanvas,
-        dbc.Navbar(dbc.Container(logo_row), dark=True, color='dark'),
+        dbc.Navbar(
+            dbc.Container(
+                [
+                    logo_row,
+                    dbc.NavbarToggler(id="navbar-toggler", n_clicks=0),
+                    dbc.Collapse(
+                        dbc.Row(
+                            [
+                                dbc.Col(
+                                    sign_out_button,
+                                    className="ms-auto",
+                                ),
+                            ],
+                            className="g-0 ms-auto flex-nowrap mt-3 mt-md-0",
+                        ),
+                        id='navbar-collapse',
+                        is_open=False,
+                        navbar=True,
+                    ),
+                ]
+            ),
+            dark=True,
+            color='dark'
+        ),
     ],
     id='pages-navbar-div',
 )
@@ -143,12 +173,7 @@ navbar = dbc.Navbar(
                     [
                         dbc.Col(instance_select),
                         dbc.Col(
-                            dbc.Button(
-                                "Sign out",
-                                color='link',
-                                style={'margin-left': '30px'},
-                                id='sign-out-button',
-                            ),
+                            sign_out_button,
                             className="ms-auto",
                         ),
                     ],
@@ -207,3 +232,81 @@ def build_cards_grid(cards: List[dbc.Card], num_columns: int = 3) -> html.Div:
         rows.append(r)
         rows.append(html.Br())
     return html.Div(rows)
+
+
+def build_pages_offcanvas_children():
+    """
+    Return the contents of the pages offcanvas.
+    """
+    from meerschaum.api.dash.callbacks.dashboard import _pages
+    from meerschaum.api.dash.callbacks.custom import _plugin_endpoints_to_pages
+    pages_listgroup_items = []
+    custom_pages = []
+    for pages_dicts in _plugin_endpoints_to_pages.values():
+        for page_dict in pages_dicts.values():
+            custom_pages.append(page_dict['page_key'])
+
+    for page_key, page_href in _pages.items():
+        if page_key in custom_pages:
+            continue
+        pages_listgroup_items.append(
+            dbc.ListGroupItem(
+                dbc.Button(
+                    ' '.join([word.capitalize() for word in page_key.split(' ')]),
+                    style={
+                        'width': '100%',
+                        'text-align': 'left',
+                        'text-decoration': 'none',
+                        'padding-left': '0',
+                    },
+                    href=page_href,
+                    color='dark',
+                )
+            )
+        )
+
+    plugins_accordion_items = []
+    for page_group, pages_dicts in _plugin_endpoints_to_pages.items():
+        plugin_listgroup_items = [
+            dbc.ListGroupItem(
+                dbc.Button(
+                    ' '.join([word.capitalize() for word in page_dict['page_key'].split(' ')]),
+                    style={
+                        'width': '100%',
+                        'text-align': 'left',
+                        'text-decoration': 'none',
+                        'padding-left': '0',
+                    },
+                    href=page_href,
+                    color='dark',
+                )
+            )
+            for page_href, page_dict in pages_dicts.items()
+        ]
+        plugin_listgroup = dbc.ListGroup(plugin_listgroup_items, flush=True)
+        plugin_accordion_item = dbc.AccordionItem(
+            plugin_listgroup,
+            title=(
+                page_group.capitalize()
+                if page_group and not page_group[0].isupper()
+                else page_group
+            ),
+            class_name='pages-offcanvas-accordion',
+        )
+        plugins_accordion_items.append(plugin_accordion_item)
+    plugins_accordion = dbc.Accordion(
+        plugins_accordion_items,
+        start_collapsed=True,
+        flush=True,
+        always_open=True,
+    )
+    pages_listgroup_items.append(plugins_accordion)
+
+    pages_children = dbc.Card(
+        dbc.ListGroup(
+            pages_listgroup_items,
+            flush=True,
+        ),
+        outline=True,
+    )
+    return pages_children

meerschaum/api/dash/pages/__init__.py

@@ -12,3 +12,4 @@ import meerschaum.api.dash.pages.plugins
 import meerschaum.api.dash.pages.register
 import meerschaum.api.dash.pages.pipes
 import meerschaum.api.dash.pages.job
+import meerschaum.api.dash.pages.settings

meerschaum/api/dash/pages/pipes.py

@@ -7,13 +7,16 @@ Display pipes via a shareable URL.
 
 from meerschaum.api import CHECK_UPDATE
 from meerschaum.utils.packages import import_html, import_dcc
-from meerschaum.api.dash.components import download_dataframe
+from meerschaum.api.dash.components import download_dataframe, pages_navbar
 
 html, dcc = import_html(check_update=CHECK_UPDATE), import_dcc(check_update=CHECK_UPDATE)
 import dash_bootstrap_components as dbc
 
-layout = dbc.Container([
-    dcc.Location('pipes-location'),
-    download_dataframe,
-    html.Div(id='pipe-output-div'),
-])
+layout = [
+    pages_navbar,
+    dbc.Container([
+        dcc.Location('pipes-location'),
+        download_dataframe,
+        html.Div(id='pipe-output-div'),
+    ])
+]

meerschaum/api/dash/pages/settings/__init__.py

@@ -0,0 +1,8 @@
+#! /usr/bin/env python3
+# vim:fenc=utf-8
+
+"""
+Define the layouts for the `settings` pages.
+"""
+
+import meerschaum.api.dash.pages.settings.password_reset

meerschaum/api/dash/pages/settings/password_reset.py

@@ -0,0 +1,63 @@
+#! /usr/bin/env python3
+# vim:fenc=utf-8
+
+"""
+Define the password reset page layout.
+"""
+
+import dash_bootstrap_components as dbc
+import dash.html as html
+import dash.dcc as dcc
+from meerschaum.plugins import web_page
+
+
+@web_page('password-reset', login_required=True, page_group='Settings')
+def page_layout():
+    """
+    Return the layout for this page.
+    """
+    return dbc.Container([
+        html.Br(),
+        html.H3('Password Reset'),
+        html.Br(),
+        html.Div(id="password-reset-alert-div"),
+        dbc.Form([
+            dbc.Row(
+                [
+                    dbc.Label("New Password", html_for="password-reset-input", width=2),
+                    dbc.Col(
+                        dbc.Input(
+                            type="password",
+                            id="password-reset-input",
+                            placeholder="Enter new password",
+                        ),
+                        width=10,
+                    ),
+                ],
+                className="mb-3",
+            ),
+            dbc.Row(
+                [
+                    dbc.Label("Confirm Password", html_for="password-reset-confirm-input", width=2),
+                    dbc.Col(
+                        dbc.Input(
+                            type="password",
+                            id="password-reset-confirm-input",
+                            placeholder="Confirm new password",
+                        ),
+                        width=10,
+                    ),
+                ],
+                className="mb-3",
+            ),
+            dbc.Row(
+                [
+                    dbc.Col(
+                        dbc.Button("Submit", id="password-reset-button", disabled=True),
+                    ),
+                ],
+                justify="end",
+                className="mb-3",
+            ),
+        ]),
+    ])

meerschaum/api/resources/static/css/dash.css

@@ -73,3 +73,10 @@ a {
 .input-text {
   font-family: Consolas, Menlo, Monaco, Lucida Console, Liberation Mono, DejaVu Sans Mono, Bitstream Vera Sans Mono, Courier New, monospace, serif !important;
 }
+.pages-offcanvas-accordion button {
+  background-color: var(--bs-dark) !important;
+  color: white !important;
+}
+.pages-offcanvas-accordion div {
+  padding: 0 !important;
+}