129 lines
4.7 KiB
Python
129 lines
4.7 KiB
Python
import datetime
|
|
import os.path
|
|
from .base import Job
|
|
|
|
# Wrap imports to prevent crashing if dependencies are missing
|
|
try:
|
|
import requests
|
|
import icalendar
|
|
import recurring_ical_events
|
|
import pytz
|
|
HAS_ICAL_DEPS = True
|
|
except ImportError:
|
|
HAS_ICAL_DEPS = False
|
|
|
|
class TasksJob(Job):
|
|
def __init__(self):
|
|
self.days_to_print = 1
|
|
|
|
def get_name(self):
|
|
return "UKOLY DNE"
|
|
|
|
def configure(self):
|
|
print("\n--- Configure Calendar URL ---")
|
|
print("To find your Google Calendar URL:")
|
|
print("1. Go to Settings > Select Calendar > Integrate calendar")
|
|
print("2. Copy 'Secret address in iCal format'")
|
|
print("-" * 30)
|
|
print("Enter the public or secret address in iCal format (.ics).")
|
|
print("Example: https://calendar.google.com/calendar/ical/.../basic.ics")
|
|
url = input("URL: ").strip()
|
|
if url:
|
|
with open('calendar_url.txt', 'w') as f:
|
|
f.write(url)
|
|
print("URL saved.")
|
|
|
|
print("-" * 30)
|
|
try:
|
|
d_input = input(f"Number of days to print [{self.days_to_print}]: ").strip()
|
|
if d_input:
|
|
self.days_to_print = int(d_input)
|
|
except ValueError:
|
|
pass
|
|
|
|
def print_body(self, p):
|
|
if not HAS_ICAL_DEPS:
|
|
print("Error: Missing iCal Libraries.")
|
|
p.text("Error: Missing iCal Libraries.\n")
|
|
p.text("Run: pip install requests icalendar recurring-ical-events pytz\n")
|
|
return
|
|
|
|
if not os.path.exists('calendar_url.txt'):
|
|
print("Error: Calendar URL not configured.")
|
|
p.text("Error: Calendar URL not configured.\n")
|
|
p.text("Run configuration in TUI first.\n")
|
|
return
|
|
|
|
with open('calendar_url.txt', 'r') as f:
|
|
url = f.read().strip()
|
|
|
|
try:
|
|
# 1. Fetch the .ics file
|
|
# Timeout is important so the printer doesn't hang forever
|
|
response = requests.get(url, timeout=30)
|
|
response.raise_for_status()
|
|
|
|
# 2. Parse Calendar
|
|
cal = icalendar.Calendar.from_ical(response.content)
|
|
|
|
# 3. Loop over days
|
|
base_now = datetime.datetime.now().astimezone()
|
|
|
|
for i in range(self.days_to_print):
|
|
p.cut()
|
|
|
|
target_date = base_now + datetime.timedelta(days=i)
|
|
|
|
# Print Calendar Name
|
|
cal_name = cal.get('X-WR-CALNAME')
|
|
if cal_name:
|
|
p.set(align='center', bold=True)
|
|
p.text(f"{str(cal_name)}\n")
|
|
days_cz = ["Pondělí", "Úterý", "Středa", "Čtvrtek", "Pátek", "Sobota", "Neděle"]
|
|
p.text(f"{days_cz[target_date.weekday()]} {target_date.strftime('%d.%m.%Y')}\n")
|
|
p.set(align='left', bold=False)
|
|
p.text("-" * 32 + "\n")
|
|
|
|
start_of_day = target_date.replace(hour=0, minute=0, second=0, microsecond=0)
|
|
end_of_day = target_date.replace(hour=23, minute=59, second=59, microsecond=0)
|
|
|
|
# 4. Expand Recurring Events
|
|
events = recurring_ical_events.of(cal).between(start_of_day, end_of_day)
|
|
|
|
# 5. Sort by start time
|
|
def get_start_time(e):
|
|
dt = e.get('DTSTART').dt
|
|
if not isinstance(dt, datetime.datetime):
|
|
return datetime.datetime.combine(dt, datetime.time.min).replace(tzinfo=base_now.tzinfo)
|
|
return dt
|
|
|
|
events.sort(key=get_start_time)
|
|
|
|
if not events:
|
|
p.text("Zadne ukoly.\n")
|
|
else:
|
|
for event in events:
|
|
summary = str(event.get('SUMMARY', '(bez nazvu)'))
|
|
dtstart = event.get('DTSTART').dt
|
|
|
|
if not isinstance(dtstart, datetime.datetime):
|
|
time_str = "Cely den"
|
|
else:
|
|
local_dt = dtstart.astimezone(base_now.tzinfo)
|
|
time_str = local_dt.strftime('%H:%M')
|
|
|
|
p.set(bold=True)
|
|
p.text(f"{time_str} [_]")
|
|
p.set(bold=False)
|
|
p.text(f" {summary}\n")
|
|
|
|
location = event.get('LOCATION')
|
|
if location:
|
|
p.text(f" @ {str(location)}\n")
|
|
|
|
p.text("-" * 32 + "\n")
|
|
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
p.text(f"Error: {e}\n")
|