ns2 0.2.6__py3-none-any.whl

ns2/ui/snmp_page.py

@@ -0,0 +1,353 @@
+from ns.lib.systemd import *
+
+from nicegui import ui, app
+from dataclasses import dataclass, asdict
+
+from ns.lib.commands import runCmd
+
+from typing import Optional
+
+from ns.dbus import dbus
+from ns.lib.snmp import GetSnmp, snmp_call,V3User, V2User, snmp_config_file, default_persistent_dir_path
+
+
+
+sourceValidation = {"Please enter a valid ip address, network or default": lambda value: len(value) > 0}
+passphraseValidation = {"Passphrase must be at least 8 characters": lambda value: len(value) >= 8,  
+                        "Passphrase must be 24 or less characaters": lambda value: 24 >= len(value)}
+usernameValidation = {"Username must be at least 5 characters": lambda value: len(value) >= 5, 
+                      "Username must be 24 or less characaters": lambda value: 24 >= len(value)}
+
+def validate_group(group: list):
+    return [x.validate() for x in group]
+    
+
+async def create_v3_user_dialog():
+
+    with ui.dialog() as createV3Dialog:
+        v3 = V3User()
+        with ui.card().classes("w-full"):
+            with ui.column().classes("w-full"):
+                version = ui.input(label="Version").classes("w-full").bind_value(v3, "Version")
+                version.disable()
+                username = ui.input(label="Username", validation=usernameValidation).classes("w-full").bind_value(v3, "UserName")
+                permissions = ui.select(label="Permissions", options=["roprivgroup", "rwprivgroup"]).classes("w-full").bind_value(v3, "Permissions")
+                auth_type = ui.select(label="Auth Alg", options=['SHA', 'MD5']).classes("w-full").bind_value(v3, "AuthType")
+                auth_pass = ui.input(label="Auth Passphrase", validation=passphraseValidation).classes("w-full").bind_value(v3, "AuthPassphrase")
+                priv_type = ui.select(label="Priv Alg", options=["AES", "DES"]).classes("w-full").bind_value(v3, "PrivType")
+                priv_pass = ui.input(label="Auth Passphrase", validation=passphraseValidation).classes("w-full").bind_value(v3, "PrivPassphrase")
+                with ui.row().classes("items-center justify-between gap-4 w-full"):
+                    
+                    async def on_save_cb():
+                        if all(validate_group([version, username, permissions, auth_type, auth_pass, priv_type, priv_pass])):
+                            print(asdict(v3))
+                            
+                            snmp = await GetSnmp(dbus.AppBus)
+                            rsp = await snmp.call_create_v3_user(asdict(v3))
+                            print(rsp)
+                            await v3table.refresh()
+                            createV3Dialog.close()
+                        else:
+                            ui.notify("Please correct the errors", type='negative')
+
+                    def on_cancel_cb():
+                        createV3Dialog.close()
+
+                    ui.button("save", on_click=on_save_cb).props("flat color=accent align=left") 
+                    ui.button(icon="cancel", on_click=on_cancel_cb).props("flat color=accent align=left")
+    
+    return createV3Dialog     
+
+
+@ui.refreshable
+async def v3table():
+    snmp = await GetSnmp(dbus.AppBus)
+    v3Users = await snmp.call_get_v3_users()
+    print(v3Users)
+    createV3Dialog = await create_v3_user_dialog()
+    
+    table = ui.table(
+            title="v3 Users",
+            rows=v3Users,
+            column_defaults={
+                "align": "left",
+                "headerClasses": "uppercase text-primary",
+            },
+        ).classes("w-full")
+    
+    table.add_slot(f'body-cell-UserName', f''' <q-td :props="props">
+                   <a :href="'/snmp/v3/'+ props.row.UserName" class="text-accent cursor-pointer hover:underline"> {{{{ props.value }}}} </a>
+                   </q-td> ''')
+
+    table.props(f'visible-columns={"UserName,Version,GroupName,AuthType,PrivType"}')  # Only show these
+    
+    with table.add_slot('top-right'):
+        ui.button(icon="add", on_click = createV3Dialog.open).props(
+            "flat color=accent align=left").classes("w-full").props("dense")
+    
+
+
+
+
+@ui.refreshable
+async def v2table():
+    snmp = await GetSnmp(dbus.AppBus)
+    v2Users = await snmp.call_get_v2_users()
+    
+    with ui.dialog() as createV2Dialog:
+        v2 = V2User()
+        v2.Version = "v2c"
+        v2.Permissions = "rwnoauthgroup"
+        v2.Source = "default"
+        with ui.card().classes("w-full"):
+            with ui.column().classes("w-full"):
+                version = ui.select(label="Version", options=["v2c", "v1"]).classes("w-full").bind_value(v2, "Version")
+                permissions = ui.select(label="Permissions", options=['rwnoauthgroup', 'ronoauthgroup']).classes("w-full").bind_value(v2, "Permissions")
+                community = ui.input("Community", 
+                                     validation={'Community required': lambda value: len(value) > 0}).classes("w-full").bind_value(v2, "Community")
+                source = ui.input("Source / IP Address", 
+                                  validation=sourceValidation).classes("w-full").bind_value(v2, "Source")
+                with ui.row().classes("items-center justify-between gap-4 w-full"):
+
+                    async def on_save_cb():
+                        if all(validate_group([version, permissions, community, source])):
+                            await snmp.call_create_v2_user(asdict(v2))
+                            await v2table.refresh()
+                            createV2Dialog.close()
+                        else:
+                            ui.notify("Please correct the errors", type='negative')
+                        
+                    def on_cancel_cb():
+                        createV2Dialog.close()
+                    ui.button("save", on_click=on_save_cb).props("flat color=accent align=left") 
+                    ui.button(icon="cancel", on_click=on_cancel_cb).props("flat color=accent align=left")
+
+    table = ui.table(
+            title="v2 Users",
+            rows=v2Users,
+            column_defaults={
+                "align": "left",
+                "headerClasses": "uppercase text-primary",
+            },
+        ).classes("w-full")
+    
+    table.add_slot(f'body-cell-Community', f'''
+            <q-td :props="props">
+                <a :href="'/snmp/v2/'+ props.row.Community" class="text-accent cursor-pointer hover:underline"> {{{{ props.value }}}} </a>
+            </q-td>
+        ''')
+
+    table.props(f'visible-columns={"Community,Version,Source,GroupName"}')  # Only show these
+    
+    with table.add_slot('top-right'):
+        ui.button(icon="add", on_click = createV2Dialog.open).props(
+            "flat color=accent align=left").classes("w-full").props("dense")
+        
+
+    
+async def snmp_status():
+    with ui.column() as status:
+        ui.label("SNMP").classes("text-h5")
+        
+        async def snmp_switch_cb(e):
+            action = "enable" if  e.sender.value else "disable"
+            with ui.dialog() as dialog, ui.card():
+                ui.label(f'Are you sure you want to {action} snmp?')
+                with ui.row():
+                    ui.button('Cancel', on_click=lambda: dialog.submit("Cancel")).props("flat color=accent align=left")
+                    ui.button(f'{action}', on_click=lambda: dialog.submit(action)).props("flat color=accent align=left")
+        
+            result = await dialog
+            active = await isActive(dbus.AppBus, 'snmpd.service')
+            
+            
+            if result == "enable" and not active:
+                await systemd_start(dbus.AppBus, 'snmpd.service')
+            
+            if result == "disable" and active:
+                await systemd_stop(dbus.AppBus, 'snmpd.service')
+
+
+            e.sender.value = await isActive(dbus.AppBus, 'snmpd.service')
+
+        async def snmp_reset_cb(e):
+            with ui.dialog() as dialog, ui.card():
+                ui.label(f'Are you sure you want to reset snmp?')
+                with ui.row():
+                    ui.button('Cancel', on_click=lambda: dialog.submit("Cancel")).props("flat color=accent align=left")
+                    ui.button('Reset', on_click=lambda: dialog.submit("reset")).props("flat color=accent align=left")    
+            if await dialog == "reset":
+                
+                snmp = await GetSnmp(dbus.AppBus)
+                await snmp.call_reset()
+                v2table.refresh()
+                v3table.refresh()
+            
+        
+        with ui.card().classes("w-full"):
+            snmp_service_switch = ui.switch("SNMPD Status").on('click', lambda e: snmp_switch_cb(e)).props("flat color=accent align=left dense")
+            snmp_service_switch.value = await isActive(dbus.AppBus, 'snmpd.service')
+            ui.button("Reset SNMPD Config", on_click=snmp_reset_cb).props("flat color=accent align=left dense")
+    
+    return status
+
+async def snmp_page():
+    
+    #def props_cb(interface_name, changed_properties, invalidated_properties):
+    #    print("props ch cb")
+    #    #print(changed_properties['ActiveState'])
+    #    status.update()
+    #    
+    #print("registered")
+    #snmpDaemon = await getUnitInterface(dbus.AppBus, "snmpd.service")
+    #snmpDaemon.on_properties_changed(props_cb)
+
+   
+    await snmp_status()
+    await v2table()  # Only show these
+    await v3table()
+        
+
+
+async def snmp_user_page(version :str, user: str):
+    with ui.row():
+        ui.link('SNMP', '/snmp')
+        ui.label('>')
+        ui.label(user)
+
+        if version == "v2":
+            await edit_delete_v2_user_card(user)
+        if version == "v3":
+            await edit_delete_v3_user_card(user)
+
+
+
+def enable_group(fields):
+    for f in fields:
+        f.enabled = True
+
+def disable_group(fields):
+    for f in fields:
+        f.enabled = False
+
+
+
+
+async def edit_delete_v2_user_card(community):
+    snmp = await GetSnmp(dbus.Bus)
+    user = await snmp.call_get_v2_user_by_community(community)
+    v2User = V2User(**user)
+    with ui.card().classes("w-full"):
+        with ui.column().classes("w-full"):
+            version = ui.select(label="Version", options=["v2c", "v1"]).classes("w-full").bind_value(v2User, "Version")
+            permissions = ui.select(label="Permissions", options=['rwnoauthgroup', 'ronoauthgroup']).classes("w-full").bind_value(v2User, "Permissions")
+            community = ui.input("Community", validation={'Community required': lambda value: len(value) > 0}).classes("w-full").bind_value(v2User, "Community")
+            source = ui.input("Source / IP Address", validation=sourceValidation).classes("w-full").bind_value(v2User, "Source")
+            with ui.row().classes("items-center justify-between gap-4 w-full"):
+
+                async def on_save_cb():
+                    disable_group(group)
+                    save_button.enabled = False
+                    edit_button.enabled = True
+                    await snmp.call_modify_v2_user(asdict(v2User))
+                    await v2table.refresh()
+                    ui.navigate.back()
+
+                def on_edit_cb():
+                    enable_group(group)
+                    edit_button.enabled = False
+                    save_button.enabled = True
+
+                async def on_delete_cb():
+                    with ui.dialog() as dialog, ui.card():
+                        ui.label(f'Are you sure you want to delete {v2User.Community}?')
+                        with ui.row():
+                            ui.button('Yes', on_click=lambda: dialog.submit(True)).props("flat color=accent align=left")
+                            ui.button('No', on_click=lambda: dialog.submit(False)).props("flat color=accent align=left")
+                    result = await dialog
+                    if result:
+                        await snmp.call_remove_v2_user(asdict(v2User))
+                        v2table.refresh()
+                        ui.navigate.back()
+                        ui.notify(f'User {v2User.Community} deleted...')
+                    else:
+                        dialog.close()
+
+                edit_button = ui.button("edit", on_click= on_edit_cb).props("flat color=accent align=left")
+                save_button = ui.button("save", on_click= on_save_cb).props("flat color=accent align=left") 
+                delete_button = ui.button(icon="delete", on_click=on_delete_cb).props("flat color=accent align=left")
+
+                group = [community, source, version, permissions]
+
+                disable_group(group)
+                edit_button.enabled = True
+                save_button.enabled = False
+    
+
+
+
+
+    
+async def edit_delete_v3_user_card(username):
+    snmp = await GetSnmp(dbus.Bus)
+    userData = await snmp.call_get_v3_user_by_username(username)
+    initUser = V3User(**userData)
+    finalUser = V3User(**userData)
+
+    with ui.card().classes("w-full"):
+        with ui.column().classes("w-full"):
+    
+            version = ui.input(label="Version").classes("w-full").bind_value(finalUser, "Version")
+            version.disable()
+            username = ui.input(label="Username", validation=usernameValidation).classes("w-full").bind_value(finalUser, "UserName")
+            permissions = ui.select(label="Permissions", options=["roprivgroup", "rwprivgroup"]).classes("w-full").bind_value(finalUser, "Permissions")
+            auth_type = ui.select(label="Auth Alg", options=['SHA', 'MD5']).classes("w-full").bind_value(finalUser, "AuthType")
+            auth_pass = ui.input(label="Auth Passphrase", validation=passphraseValidation).classes("w-full").bind_value(finalUser, "AuthPassphrase")
+            priv_type = ui.select(label="Priv Alg", options=["AES", "DES"]).classes("w-full").bind_value(finalUser, "PrivType")
+            priv_pass = ui.input(label="Auth Passphrase", validation=passphraseValidation).classes("w-full").bind_value(finalUser, "PrivPassphrase")
+
+            with ui.row().classes("items-center justify-between gap-4 w-full"):
+
+                async def on_save_cb():
+                        disable_group(group)
+                        save_button.enabled = False
+                        edit_button.enabled = True
+                        if all(validate_group([version, username, permissions, auth_type, auth_pass, priv_type, priv_pass])):
+                            await snmp.call_modify_v3_user(asdict(initUser), asdict(finalUser))
+                            #EditV3User(inituser, finaluser)
+                            ui.navigate.back()
+                        else:
+                            ui.notify("Please correct the errors", type='negative')
+
+
+                def on_edit_cb():
+                    enable_group(group)
+                    edit_button.enabled = False
+                    save_button.enabled = True
+
+
+                async def on_delete_cb():
+                    with ui.dialog() as dialog, ui.card():
+                        ui.label(f'Are you sure you want to delete {initUser.UserName}?')
+                        with ui.row():
+                            ui.button('Yes', on_click=lambda: dialog.submit(True)).props("flat color=accent align=left")
+                            ui.button('No', on_click=lambda: dialog.submit(False)).props("flat color=accent align=left")
+                    result = await dialog
+                    if result:
+                        await snmp.call_remove_v3_user(asdict(initUser))
+                        ui.navigate.back()
+                        ui.notify(f'User {initUser.UserName} deleted...')
+                    else:
+                        dialog.close()
+
+                edit_button = ui.button("edit", on_click= on_edit_cb).props("flat color=accent align=left")
+                save_button = ui.button("save", on_click= on_save_cb).props("flat color=accent align=left") 
+                delete_button = ui.button(icon="delete", on_click=on_delete_cb).props("flat color=accent align=left")
+
+                group = [permissions, username, auth_type,auth_pass,priv_type,priv_pass]
+
+                disable_group(group)
+                edit_button.enabled = True
+                save_button.enabled = False
+
+

ns2/ui/terminal.py

@@ -0,0 +1,65 @@
+#!/usr/bin/env python3
+"""Example for connecting a `Xterm` element with a `Bash` process running in a pty.
+
+WARNING: This example gives the clients full access to the server through Bash. Use with caution!
+"""
+import os
+import pty
+import signal
+from functools import partial
+import time
+from nicegui import core, events, ui
+
+pty_fd = None
+pty_pid = None
+
+isOpen = False
+
+# if terminal is open, dont do anything
+# if terminal is closed, open it
+
+def terminal_page():
+    global pty_fd, pty_pid, isOpen
+
+    if isOpen:
+        ui.label("terminal in use")
+        os.close(pty_fd)
+        os.kill(pty_pid, signal.SIGKILL)
+        print('Terminal closed')
+        isOpen = False
+
+    else:
+
+        pty_pid, pty_fd = pty.fork()  # create a new pseudo-terminal (pty) fork of the process
+        if pty_pid == pty.CHILD:
+            os.execv('/bin/bash', ('bash',))  # child process of the fork gets replaced with "bash"
+            isOpen = True
+            print('Terminal opened')
+
+        terminal = ui.xterm()
+
+        @partial(core.loop.add_reader, pty_fd)
+        def pty_to_terminal():
+            try:
+                data = os.read(pty_fd, 1024)
+            except OSError:
+                print('Stopping reading from pty')  # error reading the pty; probably bash was exited
+                core.loop.remove_reader(pty_fd)
+            else:
+                terminal.write(data)
+        @terminal.on_data
+        def terminal_to_pty(event: events.XtermDataEventArguments):
+            try:
+                os.write(pty_fd, event.data.encode('utf-8'))
+            except OSError:
+                pass  # error writing to the pty; probably bash was exited
+
+        @ui.context.client.on_disconnect
+        def kill_bash():
+            global isOpen
+            if isOpen:
+                os.close(pty_fd)
+                os.kill(pty_pid, signal.SIGKILL)
+                print('Terminal closed')
+                isOpen = False
+

ns2/ui/tests_page.py

@@ -0,0 +1,116 @@
+
+from nicegui import ui, app, background_tasks, events
+
+import pandas as pd
+import numpy as np
+
+import plotly.graph_objects as go
+import plotly.express as px
+
+import plotly.io as pio
+pio.templates.default = "plotly_dark"
+
+from ns.lib.network_delay import get_network_delay_data_locally, calculate_last_50_jitter, get_files_to_view, get_data_by_name
+
+
+    
+def set_selected_file(f):
+    global selected_file
+    selected_file = f
+
+async def tests_page():
+    global selected_file
+    
+    try:
+        selected_file = get_files_to_view()[0]['href']
+    except Exception as e:
+        print(e)
+
+    with ui.column():
+        with ui.row() as pageContainer:
+            ui.label("Network Delay Test Data").classes("text-h5")
+            
+        with ui.column():
+            for r in get_files_to_view():
+                ui.button(r["href"], on_click=lambda f=r['href']: set_selected_file(f)).classes('bg-accent')
+
+        with ui.row():
+
+            def build_plot():
+                
+                #start, ts, delays = get_network_delay_data_locally()
+
+                start, ts, delays = get_data_by_name(selected_file)
+                #delays = [0]
+                #start = ''
+
+                f1 = go.Figure(go.Scatter(x=list(range(len(delays))),y=delays))
+                f2 = go.Figure(px.histogram(x=delays))
+                jitter = calculate_last_50_jitter(delays)
+                f1.update_xaxes(title_text=f"Samples - Jitter on last 50: {jitter}")
+                f1.update_yaxes(title_text="RTT (Delay) (ms)")
+
+                f1.update_layout(title_text=f"Ping Over VPN Test Started: {start}")
+
+                mean = np.mean(delays)
+                std = np.std(delays)
+                f2.update_layout(title_text=f"Mean: {mean} STD: {std}")
+
+                #fig.update_layout(title_text="Pinging home server from novus",margin=dict(l=0, r=0, t=0, b=0))
+
+                p1 = ui.plotly(f1).classes("w-full")
+                p2 = ui.plotly(f2).classes("w-full")
+
+                return f1, p1, f2 , p2
+            f1, p1, f2, p2 = build_plot()
+
+
+            def update_plots():
+                start, ts, delays = get_data_by_name(selected_file)
+                #_, ts, delays = get_network_delay_data_locally()
+                jitter = calculate_last_50_jitter(delays)
+                f1.update_layout(title_text=f"Ping Over VPN Test Started: {start}")
+
+                f1.update_xaxes(title_text=f"Samples - Jitter on last 50: {jitter}")
+                f1.data[0].x = list(range(len(delays)))
+                f1.data[0].y = delays
+
+                f2.data[0].x = delays
+
+                p1.update()
+                p2.update()
+                mean = np.mean(delays)
+                std = np.std(delays)
+                f2.update_layout(title_text=f"Mean: {mean} STD: {std}")
+
+
+
+            timer = ui.timer(5, callback=update_plots)
+
+            def toggle_timer():
+                if timer.active:
+                    timer.deactivate()
+                else:
+                    timer.activate()
+
+        
+            def get_down_latest():
+
+                _, ts, delays = get_data_by_name(selected_file)
+
+                data = {"ts": ts, "rtt": delays}
+
+                df = pd.DataFrame(data)
+
+                df.to_excel("temp.xlsx", index=False, header=True)
+
+                ui.download.file('temp.xlsx')
+
+            
+        ui.button("Toggle Refresh", on_click=toggle_timer).classes("bg-accent")
+
+        ui.button("Download selected", on_click=get_down_latest ).classes("bg-accent")
+
+
+            
+

ns2/ui/theme.py

@@ -0,0 +1,25 @@
+from nicegui import ui, app
+
+
+def init_colors():
+    ui.colors(
+        primary="#ffffff",
+        secondary="#191a1a",
+        accent="#ff0000",
+        dark="#1f2121",
+        dark_page="#121212",
+        positive="#21ba45",
+        negative="#c10015",
+        info="#31ccec",
+        warning="#f2c037",
+        firewallcard="#3A3D3D")
+
+    dark = ui.dark_mode()
+    dark.enable()
+    ui.query("body").classes("bg-secondary")
+
+
+
+# background #191a1a
+
+# menu color #1f2121

ns2/utils.py

@@ -0,0 +1,5 @@
+from importlib.resources import files
+
+# These return Traversable objects that work like Path objects
+ASSETS_DIR = files('ns').joinpath('ui').joinpath('assets')
+INTROSPECTION_DIR = files('ns').joinpath('lib').joinpath('introspection')

ns2-0.2.6.dist-info/METADATA

@@ -0,0 +1,78 @@
+Metadata-Version: 2.4
+Name: ns2
+Version: 0.2.6
+Summary: Novus Power Products server configuration tool
+License: MIT
+Author: jowens25
+Author-email: jowens@novuspower.com
+Requires-Python: >=3.10,<4.0
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Requires-Dist: aiofiles
+Requires-Dist: aiohappyeyeballs
+Requires-Dist: aiohttp
+Requires-Dist: aiosignal
+Requires-Dist: annotated-doc
+Requires-Dist: annotated-types
+Requires-Dist: anyio
+Requires-Dist: attrs
+Requires-Dist: beautifulsoup4
+Requires-Dist: bidict
+Requires-Dist: bs4
+Requires-Dist: certifi
+Requires-Dist: charset-normalizer
+Requires-Dist: click
+Requires-Dist: dbus-next
+Requires-Dist: docutils
+Requires-Dist: fastapi
+Requires-Dist: frozenlist
+Requires-Dist: h11
+Requires-Dist: httpcore
+Requires-Dist: httptools
+Requires-Dist: httpx
+Requires-Dist: idna
+Requires-Dist: ifaddr
+Requires-Dist: itsdangerous
+Requires-Dist: jinja2
+Requires-Dist: markdown2
+Requires-Dist: markupsafe
+Requires-Dist: multidict
+Requires-Dist: narwhals
+Requires-Dist: nicegui
+Requires-Dist: numpy
+Requires-Dist: orjson
+Requires-Dist: packaging
+Requires-Dist: pam
+Requires-Dist: pandas
+Requires-Dist: plotly
+Requires-Dist: propcache
+Requires-Dist: pydantic
+Requires-Dist: pydantic-core
+Requires-Dist: pygments
+Requires-Dist: pyinstaller
+Requires-Dist: python-dateutil
+Requires-Dist: python-dotenv
+Requires-Dist: python-engineio
+Requires-Dist: python-multipart
+Requires-Dist: python-pam
+Requires-Dist: python-socketio
+Requires-Dist: pyyaml
+Requires-Dist: requests
+Requires-Dist: simple-websocket
+Requires-Dist: six
+Requires-Dist: soupsieve
+Requires-Dist: typing-extensions
+Requires-Dist: typing-inspection
+Requires-Dist: urllib3
+Requires-Dist: uvicorn
+Requires-Dist: uvloop
+Requires-Dist: watchfiles
+Requires-Dist: websockets
+Requires-Dist: wheel2deb
+Requires-Dist: wsproto
+Requires-Dist: yarl