jepler/textual-totp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

re-organize and add copy/show clickable buttons

Browse source
This commit is contained in:
Jeff Epler 2023-12-04 19:07:21 -06:00
parent d156e0e2ad
commit cdac5f90e5
No known key found for this signature in database
GPG key ID: D5BF15AB975AB4DE
1 changed files with 26 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
src/ttotp/__main__.py
View file

@ -16,7 +16,7 @@ from typing import TYPE_CHECKING, Dict, Any, Sequence, cast
from textual.app import App, ComposeResult
from textual.widget import Widget
from textual.widgets import Label, Footer, ProgressBar
from textual.widgets import Label, Footer, ProgressBar, Button
from textual.binding import Binding
from textual.containers import VerticalScroll
from textual.css.query import DOMQuery
@ -126,7 +126,7 @@ class TOTPLabel(Label, can_focus=True):
Binding("down", "focus_next", show=False),
]
def __init__(self, otp: TOTPData) -> None:
def __init__(self, otp: "TOTPData") -> None:
self.otp = otp
super().__init__(
f"{otp.totp.name} / {otp.totp.issuer}",
@ -152,14 +152,10 @@ class TOTPLabel(Label, can_focus=True):
for widget in self.related:
widget.add_class(cls)
def on_blur(self) -> None:
self.related_remove_class("otp-focused")
self.otp.value_widget.update("*" * self.otp.totp.digits)
self.shown = False
def on_focus(self) -> None:
self.related_add_class("otp-focused")
class TOTPButton(Button, can_focus=False):
def __init__(self, otp: "OTPData", label: str, classes: str):
self.otp = otp
super().__init__(label=label, classes=classes)
@dataclass
class TOTPData:
@ -168,6 +164,8 @@ class TOTPData:
name_widget: Label = field(init=False)
value_widget: Label = field(init=False)
progress_widget: ProgressBar = field(init=False)
copy_widget: TOTPButton = field(init=False)
show_widget: TOTPButton = field(init=False)
@property
def id(self) -> int:
@ -175,7 +173,10 @@ class TOTPData:
def __post_init__(self) -> None:
self.name_widget = TOTPLabel(self)
self.copy_widget = TOTPButton(self, "🗐 ",
classes=f"otp-copy otp-copy-{self.id} otp-{self.id}")
self.show_widget = TOTPButton(self, "👀",
classes=f"otp-show otp-show-{self.id} otp-{self.id}")
self.value_widget = Label(
"*" * self.totp.digits,
classes=f"otp-value otp-value-{self.id} otp-{self.id}",
@ -197,21 +198,22 @@ class TOTPData:
@property
def widgets(self) -> Sequence[Widget]:
return self.value_widget, self.name_widget, self.progress_widget
return self.value_widget, self.name_widget, self.copy_widget, self.show_widget, self.progress_widget
class TTOTP(App[None]):
CSS = """
VerticalScroll {
layout: grid;
grid-size: 2;
grid-columns: 9 1fr;
grid-size: 5;
grid-rows: 1;
grid-columns: 9 1fr 6 6;
}
.otp-focused { background: $primary-background; }
ProgressBar { column-span: 2; }
TOTPLabel:focus { background: $primary-background; }
Bar > .bar--bar { color: $success; }
Bar { width: 1fr; }
Button { border: none; height: 1; width: 3; min-width: 4 }
"""
def __init__(self, tokens: Sequence[pyotp.TOTP]) -> None:
@ -266,6 +268,14 @@ class TTOTP(App[None]):
self.notify("Code copied", title="")
def on_button_pressed(self, event):
button = event.button
self.screen.set_focus(button.otp.name_widget)
if 'otp-show' in button.classes:
self.action_show()
else:
self.action_copy()
@click.command
@click.option(