32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
from src.cloud_util.ticktick import TickTick
|
||
|
|
from src.config import Config
|
||
|
|
|
||
|
|
|
||
|
|
class HomeAssistant:
|
||
|
|
class PublishMessage(BaseModel):
|
||
|
|
target: str
|
||
|
|
action: str
|
||
|
|
content: str
|
||
|
|
|
||
|
|
def __init__(self, ticktick: TickTick) -> None:
|
||
|
|
self._ticktick = ticktick
|
||
|
|
|
||
|
|
def process_publish_message(self, message: PublishMessage) -> dict[str, str]:
|
||
|
|
if message.target == "ticktick":
|
||
|
|
return self._process_ticktick_message(message=message)
|
||
|
|
|
||
|
|
return {"Status": "Unknown target"}
|
||
|
|
|
||
|
|
def _process_ticktick_message(self, message: PublishMessage) -> dict[str, str]:
|
||
|
|
if message.action == "create_shopping_list":
|
||
|
|
return self._create_shopping_list(item=message.content)
|
||
|
|
|
||
|
|
return {"Status": "Unknown action"}
|
||
|
|
|
||
|
|
def _create_shopping_list(self, item: str) -> dict[str, str]:
|
||
|
|
project_id = Config.get_env("TICKTICK_SHOPPING_LIST")
|
||
|
|
task = TickTick.Task(projectId=project_id, title=item)
|
||
|
|
return self._ticktick.create_task(task=task)
|