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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
|
"""This module defines a datepicker widget"""
from datetime import date, datetime
from asciimatics.event import KeyboardEvent, MouseEvent
from asciimatics.exceptions import InvalidFields
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 _DatePickerPopup(_TempPopup):
"""
An internal Frame for editing the currently selected date.
"""
def __init__(self, parent, year_range=None):
"""
:param parent: The widget that spawned this pop-up.
:param year_range: Optional range to limit the year selection to.
"""
# Create the lists for each entry.
now = parent.value if parent.value else date.today()
if year_range is None:
year_range = range(now.year - 50, now.year + 50)
self._days = ListBox(3,
[(f"{x:02}", x) for x in range(1, 32)],
centre=True,
validator=self._check_date)
self._months = ListBox(3,
[(now.replace(day=1, month=x).strftime("%b"), x)
for x in range(1, 13)],
centre=True,
on_change=self._refresh_day)
self._years = ListBox(3,
[(f"{x:04}", x) for x in year_range],
centre=True,
on_change=self._refresh_day)
# Construct the Frame
location = parent.get_location()
super().__init__(parent.frame.screen,
parent,
location[0] - 1, location[1] - 2,
13, 5)
# Build the widget to display the time selection.
layout = Layout([2, 1, 3, 1, 4], fill_frame=True)
self.add_layout(layout)
layout.add_widget(self._days, 0)
layout.add_widget(Label("\n/", height=3), 1)
layout.add_widget(self._months, 2)
layout.add_widget(Label("\n/", height=3), 3)
layout.add_widget(self._years, 4)
self.fix()
# Set up the correct time.
self._years.value = parent.value.year
self._months.value = parent.value.month
self._days.value = parent.value.day
def _check_date(self, value):
try:
date(self._years.value, self._months.value, value)
return True
except (TypeError, ValueError):
return False
def _refresh_day(self):
self._days.value = self._days.value
def _on_close(self, cancelled):
try:
if not cancelled:
self._parent.value = self._parent.value.replace(day=self._days.value,
month=self._months.value,
year=self._years.value)
except ValueError as e:
raise InvalidFields([self._days]) from e
class DatePicker(Widget):
"""
A DatePicker widget allows you to pick a date from a compact, temporary, pop-up Frame.
"""
__slots__ = ["_on_change", "_child", "_year_range"]
def __init__(self, label=None, name=None, year_range=None, on_change=None, **kwargs):
"""
:param label: An optional label for the widget.
:param name: The name for the widget.
: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().date()
self._child = None
self._year_range = year_range
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("%d/%b/%Y"),
self._x + self._offset,
self._y,
colour, attr, background)
def reset(self):
pass
def process_event(self, event):
if event is not None:
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
if event is None:
self._child = _DatePickerPopup(self, year_range=self._year_range)
self.frame.scene.add_effect(self._child)
return event
def required_height(self, offset, width):
return 1
@property
def value(self):
"""
The current selected date.
"""
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()
|
