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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
"""This module implements a time picker widget"""
from datetime import datetime
from asciimatics.event import KeyboardEvent, MouseEvent
from asciimatics.screen import Screen
from asciimatics.widgets.label import Label
from asciimatics.widgets.layout import Layout
from asciimatics.widgets.listbox import ListBox
from asciimatics.widgets.temppopup import _TempPopup
from asciimatics.widgets.widget import Widget
class _TimePickerPopup(_TempPopup):
"""
An internal Frame for editing the currently selected time.
"""
def __init__(self, parent):
"""
:param parent: The widget that spawned this pop-up.
"""
# Construct the Frame
location = parent.get_location()
super().__init__(parent.frame.screen,
parent,
location[0] - 1, location[1] - 2,
10 if parent.include_seconds else 7, 5)
# Build the widget to display the time selection.
self._hours = ListBox(3, [(f"{x:02}", x) for x in range(24)], centre=True)
self._minutes = ListBox(3, [(f"{x:02}", x) for x in range(60)], centre=True)
self._seconds = ListBox(3, [(f"{x:02}", x) for x in range(60)], centre=True)
if self._parent.include_seconds:
layout = Layout([2, 1, 2, 1, 2], fill_frame=True)
else:
layout = Layout([2, 1, 2], fill_frame=True)
self.add_layout(layout)
layout.add_widget(self._hours, 0)
layout.add_widget(Label("\n:", height=3), 1)
layout.add_widget(self._minutes, 2)
if self._parent.include_seconds:
layout.add_widget(Label("\n:", height=3), 3)
layout.add_widget(self._seconds, 4)
self.fix()
# Set up the correct time.
self._hours.value = parent.value.hour
self._minutes.value = parent.value.minute
self._seconds.value = parent.value.second
def _on_close(self, cancelled):
if not cancelled:
self._parent.value = self._parent.value.replace(hour=self._hours.value,
minute=self._minutes.value,
second=self._seconds.value)
class TimePicker(Widget):
"""
A TimePicker widget allows you to pick a time from a compact, temporary, pop-up Frame.
"""
__slots__ = ["_on_change", "_child", "include_seconds"]
def __init__(self, label=None, name=None, seconds=False, on_change=None, **kwargs):
"""
:param label: An optional label for the widget.
:param name: The name for the widget.
:param seconds: Whether to include selection of seconds or not.
:param on_change: Optional function to call when the selected time changes.
Also see the common keyword arguments in :py:obj:`.Widget`.
"""
super().__init__(name, **kwargs)
self._label = label
self._on_change = on_change
self._value = datetime.now().time()
self._child = None
self.include_seconds = seconds
def update(self, frame_no):
self._draw_label()
# This widget only ever needs display the current selection - the separate Frame does all
# the clever stuff when it has the focus.
(colour, attr, background) = self._pick_colours("edit_text")
self._frame.canvas.print_at(
self._value.strftime("%H:%M:%S" if self.include_seconds else "%H:%M"),
self._x + self._offset,
self._y,
colour, attr, background)
def reset(self):
pass
def process_event(self, event):
if event is not None:
# Handle key or mouse selection events - e.g. click on widget or Enter.
if isinstance(event, KeyboardEvent):
if event.key_code in [Screen.ctrl("M"), Screen.ctrl("J"), ord(" ")]:
event = None
elif isinstance(event, MouseEvent):
if event.buttons != 0:
if self.is_mouse_over(event, include_label=False):
event = None
# Create the pop-up if needed
if event is None:
self._child = _TimePickerPopup(self)
self.frame.scene.add_effect(self._child)
return event
def required_height(self, offset, width):
return 1
@property
def value(self):
"""
The current selected time.
"""
return self._value
@value.setter
def value(self, new_value):
# Only trigger the notification after we've changed the value.
old_value = self._value
self._value = new_value
if old_value != self._value and self._on_change:
self._on_change()
|
