1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
from textual import on
from textual.app import ComposeResult
from textual.widgets import Input, Button, Label
from textual.containers import Vertical, Horizontal
from textual.screen import ModalScreen
from ..back.scanner_profiles_manager import ScannerProfilesManager
class SaveProfilePushScreen(ModalScreen[str]):
def __init__(self, manager: ScannerProfilesManager, nmap_input, cve_check, *args, **kwargs):
super().__init__(*args, **kwargs)
self.manager = manager
self.nmap_input = nmap_input
self.cve_check = cve_check
def compose(self) -> ComposeResult:
with Vertical(classes="modal-window small-modal"):
yield Label("Save scan profile", classes="modal-header")
yield Input(placeholder="Profile name:", id="profile-name", classes="input")
with Horizontal(id="buttons-row", classes="modal-buttons modal-footer"):
yield Button("Confirm", id="confirm-button", variant="success", classes="button")
yield Button("Cancel", id="cancel-button", variant="error", classes="button")
@on(Button.Pressed)
async def on_button_pressed(self, event: Button.Pressed):
if event.button.id == "confirm-button":
profile_name = self.query_one("#profile-name", Input).value.strip()
self.manager.add_profile(profile_name)
self.manager.update_profile(profile_name,"nmap_input",self.nmap_input,notify=False)
self.manager.update_profile(profile_name,"cve",self.cve_check, notify=False)
self.dismiss(None)
else:
self.dismiss(None)
|
