Comma Device
101 lines
3.3 KiB
Python
101 lines
3.3 KiB
Python
import datetime
|
|
import time
|
|
import threading
|
|
|
|
from openpilot.common.params import Params
|
|
from openpilot.common.realtime import Ratekeeper
|
|
|
|
def calculate_easter(year):
|
|
a = year % 19
|
|
b = year // 100
|
|
c = year % 100
|
|
d = b // 4
|
|
e = b % 4
|
|
f = (b + 8) // 25
|
|
g = (b - f + 1) // 3
|
|
h = (19 * a + b - d - g + 15) % 30
|
|
i = c // 4
|
|
k = c % 4
|
|
l = (32 + 2 * e + 2 * i - h - k) % 7
|
|
m = (a + 11 * h + 22 * l) // 451
|
|
month = (h + l - 7 * m + 114) // 31
|
|
day = ((h + l - 7 * m + 114) % 31) + 1
|
|
return datetime.datetime(year, month, day)
|
|
|
|
def calculate_thanksgiving(year):
|
|
fourth_thursday = 4
|
|
november_first = datetime.datetime(year, 11, 1)
|
|
day_of_week = november_first.weekday()
|
|
thanksgiving = november_first + datetime.timedelta(days=((fourth_thursday - day_of_week) % 7) + 21)
|
|
return thanksgiving
|
|
|
|
def is_within_week_of(target_date, current_date):
|
|
start_of_week = target_date - datetime.timedelta(days=target_date.weekday())
|
|
end_of_week = start_of_week + datetime.timedelta(days=6)
|
|
return start_of_week <= current_date <= end_of_week
|
|
|
|
def check_date(params_memory):
|
|
current_date = datetime.datetime.now()
|
|
year = current_date.year
|
|
previous_theme_id = params_memory.get_int("CurrentHolidayTheme")
|
|
|
|
holidays = {
|
|
"april_fools_day": (datetime.datetime(year, 4, 1), 1),
|
|
"christmas": (datetime.datetime(year, 12, 25), 2),
|
|
"cinco_de_mayo": (datetime.datetime(year, 5, 5), 3),
|
|
"easter": (calculate_easter(year), 4),
|
|
"fourth_of_july": (datetime.datetime(year, 7, 4), 5),
|
|
"halloween": (datetime.datetime(year, 10, 31), 6),
|
|
"new_years_day": (datetime.datetime(year, 1, 1), 7),
|
|
"st_patricks_day": (datetime.datetime(year, 3, 17), 8),
|
|
"thanksgiving": (calculate_thanksgiving(year), 9),
|
|
"valentines_day": (datetime.datetime(year, 2, 14), 10),
|
|
"world_frog_day": (datetime.datetime(year, 3, 20), 11)
|
|
}
|
|
|
|
for holiday, (date, theme_id) in holidays.items():
|
|
if holiday in ["easter", "thanksgiving"] or holiday.endswith("_week"):
|
|
if is_within_week_of(date, current_date):
|
|
params_memory.put_int("CurrentHolidayTheme", theme_id)
|
|
if theme_id != previous_theme_id:
|
|
threading.Thread(target=update_holiday_theme, args=(params_memory,)).start()
|
|
return
|
|
elif current_date.date() == date.date():
|
|
params_memory.put_int("CurrentHolidayTheme", theme_id)
|
|
if theme_id != previous_theme_id:
|
|
threading.Thread(target=update_holiday_theme, args=(params_memory,)).start()
|
|
return
|
|
else:
|
|
params_memory.put_int("CurrentHolidayTheme", 0)
|
|
if previous_theme_id != 0:
|
|
threading.Thread(target=update_holiday_theme, args=(params_memory,)).start()
|
|
|
|
def update_holiday_theme(params_memory):
|
|
params_memory.put_bool("FrogPilotTogglesUpdated", True)
|
|
time.sleep(1)
|
|
params_memory.put_bool("FrogPilotTogglesUpdated", False)
|
|
|
|
def wait_for_next_minute():
|
|
now = datetime.datetime.now()
|
|
next_minute = (now + datetime.timedelta(minutes=1)).replace(second=0, microsecond=0)
|
|
wait_seconds = (next_minute - now).total_seconds()
|
|
time.sleep(wait_seconds)
|
|
|
|
def frogpilot_process_thread():
|
|
params_memory = Params("/dev/shm/params")
|
|
|
|
rk = Ratekeeper(1.0)
|
|
|
|
while True:
|
|
current_time = datetime.datetime.now()
|
|
if current_time.second == 0:
|
|
check_date(params_memory)
|
|
rk.keep_time()
|
|
|
|
def main():
|
|
wait_for_next_minute()
|
|
frogpilot_process_thread()
|
|
|
|
if __name__ == "__main__":
|
|
main()
|