Migrate TickTick OAuth and action tasks
This commit is contained in:
@@ -1,10 +1,13 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
from datetime import UTC, datetime, time, timedelta
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
from app.integrations.ticktick import TICKTICK_DATETIME_FORMAT, TickTickClient, TickTickTask
|
||||
from app.schemas.homeassistant import HomeAssistantPublishEnvelope
|
||||
from app.schemas.location import LocationRecordRequest
|
||||
from app.schemas.ticktick import TickTickActionTaskRequest
|
||||
from app.services.location import record_location
|
||||
|
||||
|
||||
@@ -13,12 +16,18 @@ class UnsupportedHomeAssistantMessage(RuntimeError):
|
||||
|
||||
|
||||
def handle_homeassistant_message(
|
||||
session: Session, envelope: HomeAssistantPublishEnvelope
|
||||
session: Session,
|
||||
envelope: HomeAssistantPublishEnvelope,
|
||||
ticktick_client: TickTickClient | None = None,
|
||||
) -> None:
|
||||
if envelope.target == "location_recorder":
|
||||
_handle_location_message(session, envelope)
|
||||
return
|
||||
|
||||
if envelope.target == "ticktick":
|
||||
_handle_ticktick_message(envelope, ticktick_client)
|
||||
return
|
||||
|
||||
raise UnsupportedHomeAssistantMessage(
|
||||
f"Unsupported Home Assistant target/action: {envelope.target}/{envelope.action}"
|
||||
)
|
||||
@@ -33,3 +42,38 @@ def _handle_location_message(session: Session, envelope: HomeAssistantPublishEnv
|
||||
content = json.loads(envelope.content.replace("'", '"'))
|
||||
payload = LocationRecordRequest.model_validate(content)
|
||||
record_location(session, payload)
|
||||
|
||||
|
||||
def _handle_ticktick_message(
|
||||
envelope: HomeAssistantPublishEnvelope,
|
||||
ticktick_client: TickTickClient | None,
|
||||
) -> None:
|
||||
if envelope.action != "create_action_task":
|
||||
raise UnsupportedHomeAssistantMessage(
|
||||
f"Unsupported Home Assistant target/action: {envelope.target}/{envelope.action}"
|
||||
)
|
||||
if ticktick_client is None:
|
||||
raise UnsupportedHomeAssistantMessage("TickTick client is unavailable")
|
||||
|
||||
content = json.loads(envelope.content.replace("'", '"'))
|
||||
payload = TickTickActionTaskRequest.model_validate(content)
|
||||
project_id = ticktick_client.settings.home_assistant_action_task_project_id
|
||||
if not project_id:
|
||||
raise RuntimeError(
|
||||
"TickTick action task integration is missing HOME_ASSISTANT_ACTION_TASK_PROJECT_ID"
|
||||
)
|
||||
|
||||
ticktick_client.create_task(
|
||||
TickTickTask(
|
||||
projectId=project_id,
|
||||
title=payload.action,
|
||||
dueDate=build_action_task_due_date(datetime.now().astimezone(), payload.due_hour),
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
def build_action_task_due_date(now: datetime, due_hour: int) -> str:
|
||||
local_now = now.astimezone()
|
||||
due = local_now + timedelta(hours=due_hour)
|
||||
next_midnight = datetime.combine(due.date(), time.min, tzinfo=local_now.tzinfo) + timedelta(days=1)
|
||||
return next_midnight.astimezone(UTC).strftime(TICKTICK_DATETIME_FORMAT)
|
||||
|
||||
Reference in New Issue
Block a user