Started working on the local ncurses UI. Also logging...
This commit is contained in:
Kalzu Rekku
230
alert_api/ncurses_ui.py
Normal file
230
alert_api/ncurses_ui.py
Normal file
@@ -0,0 +1,230 @@
|
||||
import curses
|
||||
import time
|
||||
from datetime import datetime, date, timedelta
|
||||
import os
|
||||
from big_digits import BIG_DIGITS
|
||||
from ncurses_ui_draw import _draw_big_digit, _draw_main_clock, _draw_add_alarm, _draw_list_alarms, _draw_error
|
||||
|
||||
class UI:
|
||||
def __init__(self, alarm_system_manager):
|
||||
"""
|
||||
Initialize the ncurses UI for the alarm system
|
||||
|
||||
Args:
|
||||
alarm_system_manager (AlarmSystemManager): The main alarm system manager
|
||||
"""
|
||||
self.alarm_system = alarm_system_manager
|
||||
self.stop_event = alarm_system_manager.stop_event
|
||||
self.storage = alarm_system_manager.storage
|
||||
|
||||
# UI state variables
|
||||
self.selected_menu = 0
|
||||
self.new_alarm_name = "Alarm"
|
||||
self.new_alarm_hour = datetime.now().hour
|
||||
self.new_alarm_minute = datetime.now().minute
|
||||
self.new_alarm_selected = 0
|
||||
self.new_alarm_date = None
|
||||
self.new_alarm_weekdays = []
|
||||
self.new_alarm_enabled = True
|
||||
self.weekday_names = ['Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat', 'Sun']
|
||||
self.alarms = []
|
||||
self.error_message = None
|
||||
self.error_timestamp = None
|
||||
|
||||
def run(self):
|
||||
"""
|
||||
Start the ncurses UI
|
||||
"""
|
||||
def ui_thread():
|
||||
try:
|
||||
curses.wrapper(self._main_loop)
|
||||
except Exception as e:
|
||||
print(f"Error in UI thread: {e}")
|
||||
finally:
|
||||
self.stop_event.set()
|
||||
|
||||
import threading
|
||||
ui_thread_obj = threading.Thread(target=ui_thread, daemon=True)
|
||||
ui_thread_obj.start()
|
||||
return ui_thread_obj
|
||||
|
||||
def _handle_add_alarm_input(self, key):
|
||||
"""
|
||||
Handle input for adding a new alarm
|
||||
"""
|
||||
if key == 27: # Escape
|
||||
self.selected_menu = 0
|
||||
return
|
||||
|
||||
if key == 10: # Enter
|
||||
try:
|
||||
# Prepare alarm data
|
||||
alarm_data = {
|
||||
"name": self.new_alarm_name,
|
||||
"time": f"{self.new_alarm_hour:02d}:{self.new_alarm_minute:02d}:00",
|
||||
"enabled": self.new_alarm_enabled
|
||||
}
|
||||
|
||||
# Add repeat rule if applicable
|
||||
if self.new_alarm_weekdays:
|
||||
alarm_data["repeat_rule"] = {
|
||||
"type": "weekly",
|
||||
"days": self.new_alarm_weekdays
|
||||
}
|
||||
elif self.new_alarm_date:
|
||||
alarm_data["repeat_rule"] = {
|
||||
"type": "once",
|
||||
"date": self.new_alarm_date.strftime("%Y-%m-%d")
|
||||
}
|
||||
|
||||
# Save new alarm
|
||||
self.storage.save_new_alert(alarm_data)
|
||||
|
||||
# Reset form
|
||||
self.selected_menu = 0
|
||||
self.new_alarm_hour = datetime.now().hour
|
||||
self.new_alarm_minute = datetime.now().minute
|
||||
self.new_alarm_date = None
|
||||
self.new_alarm_weekdays = []
|
||||
self.new_alarm_name = "Alarm"
|
||||
self.new_alarm_enabled = True
|
||||
except Exception as e:
|
||||
# TODO: Show error on screen
|
||||
print(f"Failed to save alarm: {e}")
|
||||
return
|
||||
|
||||
if key == curses.KEY_LEFT:
|
||||
self.new_alarm_selected = (self.new_alarm_selected - 1) % 6
|
||||
elif key == curses.KEY_RIGHT:
|
||||
self.new_alarm_selected = (self.new_alarm_selected + 1) % 6
|
||||
elif key == 32: # Space
|
||||
if self.new_alarm_selected == 2: # Date
|
||||
self.new_alarm_date = None
|
||||
elif self.new_alarm_selected == 3: # Weekdays
|
||||
current_day = len(self.new_alarm_weekdays)
|
||||
if current_day < 7:
|
||||
if current_day in self.new_alarm_weekdays:
|
||||
self.new_alarm_weekdays.remove(current_day)
|
||||
else:
|
||||
self.new_alarm_weekdays.append(current_day)
|
||||
self.new_alarm_weekdays.sort()
|
||||
elif self.new_alarm_selected == 5: # Enabled toggle
|
||||
self.new_alarm_enabled = not self.new_alarm_enabled
|
||||
|
||||
elif key == curses.KEY_UP:
|
||||
if self.new_alarm_selected == 0:
|
||||
self.new_alarm_hour = (self.new_alarm_hour + 1) % 24
|
||||
elif self.new_alarm_selected == 1:
|
||||
self.new_alarm_minute = (self.new_alarm_minute + 1) % 60
|
||||
elif self.new_alarm_selected == 2:
|
||||
if not self.new_alarm_date:
|
||||
self.new_alarm_date = date.today()
|
||||
else:
|
||||
self.new_alarm_date += timedelta(days=1)
|
||||
elif self.new_alarm_selected == 4: # Name
|
||||
self.new_alarm_name += " "
|
||||
|
||||
elif key == curses.KEY_DOWN:
|
||||
if self.new_alarm_selected == 0:
|
||||
self.new_alarm_hour = (self.new_alarm_hour - 1) % 24
|
||||
elif self.new_alarm_selected == 1:
|
||||
self.new_alarm_minute = (self.new_alarm_minute - 1) % 60
|
||||
elif self.new_alarm_selected == 2 and self.new_alarm_date:
|
||||
self.new_alarm_date -= timedelta(days=1)
|
||||
elif self.new_alarm_selected == 4 and len(self.new_alarm_name) > 1:
|
||||
self.new_alarm_name = self.new_alarm_name.rstrip()[:-1]
|
||||
|
||||
def _handle_list_alarms_input(self, key):
|
||||
"""
|
||||
Handle input for the list alarms screen
|
||||
"""
|
||||
if key == 27: # Escape
|
||||
self.selected_menu = 0
|
||||
elif key == ord('d'):
|
||||
# Delete last alarm if exists
|
||||
if self.alarms:
|
||||
last_alarm = self.alarms[-1]
|
||||
try:
|
||||
self.storage.remove_saved_alert(last_alarm['id'])
|
||||
except Exception as e:
|
||||
print(f"Failed to delete alarm: {e}")
|
||||
|
||||
def _show_error(self, message, duration=3):
|
||||
"""Display an error message for a specified duration"""
|
||||
self.error_message = message
|
||||
self.error_timestamp = time.time()
|
||||
|
||||
def _clear_error_if_expired(self):
|
||||
"""Clear error message if it has been displayed long enough"""
|
||||
if self.error_message and self.error_timestamp:
|
||||
if time.time() - self.error_timestamp > 3: # 3 seconds
|
||||
self.error_message = None
|
||||
self.error_timestamp = None
|
||||
|
||||
def _main_loop(self, stdscr):
|
||||
"""
|
||||
Main ncurses event loop
|
||||
"""
|
||||
# Initialize color pairs
|
||||
curses.start_color()
|
||||
curses.init_pair(1, curses.COLOR_GREEN, curses.COLOR_BLACK)
|
||||
curses.init_pair(2, curses.COLOR_YELLOW, curses.COLOR_BLACK)
|
||||
curses.init_pair(3, curses.COLOR_RED, curses.COLOR_BLACK)
|
||||
curses.curs_set(0)
|
||||
|
||||
# Configure screen
|
||||
stdscr.keypad(1)
|
||||
stdscr.timeout(100)
|
||||
|
||||
while not self.stop_event.is_set():
|
||||
# Clear the screen
|
||||
stdscr.clear()
|
||||
|
||||
# Draw appropriate screen based on selected menu
|
||||
if self.selected_menu == 0:
|
||||
_draw_main_clock(stdscr)
|
||||
elif self.selected_menu == 1:
|
||||
_draw_add_alarm(stdscr, {
|
||||
'new_alarm_selected': self.new_alarm_selected,
|
||||
'new_alarm_name': self.new_alarm_name,
|
||||
'new_alarm_hour': self.new_alarm_hour,
|
||||
'new_alarm_minute': self.new_alarm_minute,
|
||||
'new_alarm_enabled': self.new_alarm_enabled,
|
||||
'new_alarm_date': self.new_alarm_date,
|
||||
'new_alarm_weekdays': self.new_alarm_weekdays,
|
||||
'weekday_names': self.weekday_names,
|
||||
'new_alarm_snooze_enabled': getattr(self, 'new_alarm_snooze_enabled', False),
|
||||
'new_alarm_snooze_duration': getattr(self, 'new_alarm_snooze_duration', 5),
|
||||
'new_alarm_snooze_max_count': getattr(self, 'new_alarm_snooze_max_count', 3)
|
||||
})
|
||||
elif self.selected_menu == 2:
|
||||
_draw_list_alarms(stdscr, {
|
||||
'alarms': self.alarms,
|
||||
'weekday_names': self.weekday_names
|
||||
})
|
||||
|
||||
# Refresh the screen
|
||||
stdscr.refresh()
|
||||
|
||||
# Small sleep to reduce CPU usage
|
||||
time.sleep(0.1)
|
||||
|
||||
# Handle input
|
||||
key = stdscr.getch()
|
||||
if key != -1:
|
||||
# Menu navigation and input handling
|
||||
if key == ord('q') or key == 27: # 'q' or Escape
|
||||
break
|
||||
elif key == ord('c'): # Clock/Home screen
|
||||
self.selected_menu = 0
|
||||
elif key == ord('a'): # Add Alarm
|
||||
self.selected_menu = 1
|
||||
elif key == ord('l'): # List Alarms
|
||||
self.selected_menu = 2
|
||||
self.alarms = self.storage.get_saved_alerts()
|
||||
|
||||
# Context-specific input handling
|
||||
if self.selected_menu == 1:
|
||||
self._handle_add_alarm_input(key)
|
||||
elif self.selected_menu == 2:
|
||||
self._handle_list_alarms_input(key)
|
||||
Reference in New Issue
Block a user