83 lines
2.4 KiB
Python
83 lines
2.4 KiB
Python
"""Org module."""
|
|
|
|
import locale
|
|
from datetime import datetime
|
|
|
|
from orgparse import loads
|
|
|
|
from settings import ORG_CAPTURE_FILENAME, ORG_LINKS_FILENAME, ORG_PLAN_FILENAME
|
|
|
|
locale.setlocale(locale.LC_ALL, "es_ES.utf8")
|
|
|
|
|
|
class OrgData:
|
|
"""Manage org-mode files for TODO items, plans, and links.
|
|
|
|
Provides functionality to read and write org-mode formatted data, including
|
|
TODOs, daily plans, and links. Handles file operations and data formatting.
|
|
"""
|
|
|
|
def _generate_today(self):
|
|
"""Generate a formatted timestamp for today's date.
|
|
|
|
Returns:
|
|
str: Formatted string with date, day, and time (YYYY-MM-DD day HH:MM)
|
|
"""
|
|
today = datetime.today()
|
|
today_ymd = today.strftime("%Y-%m-%d")
|
|
today_day = today.strftime("%a").lower()
|
|
today_hour = today.strftime("%H:%M")
|
|
return f"{today_ymd} {today_day} {today_hour}"
|
|
|
|
def add_new_todo(self, keyword: str, description: str, outcome: str, extra: str) -> None:
|
|
"""Add a new TODO item to the capture file.
|
|
|
|
Args:
|
|
keyword: Category of TODO (e.g., 'TODO', 'REPEAT', 'NEXT')
|
|
description: Title/description of the task
|
|
outcome: Desired outcome or objective
|
|
extra: Additional notes or context
|
|
"""
|
|
today = self._generate_today()
|
|
|
|
todo_template = f"""
|
|
** {keyword.upper()} {description} :NEW::BOT:
|
|
Desired outcome: {outcome}
|
|
:LOGBOOK:
|
|
- Added: [{today}]
|
|
:END:
|
|
|
|
{extra}
|
|
"""
|
|
|
|
with open(ORG_CAPTURE_FILENAME, "a") as capture_file:
|
|
capture_file.write(todo_template)
|
|
|
|
def list_plan(self, filename: str) -> str:
|
|
"""Get and format the daily plan from an org file.
|
|
|
|
Args:
|
|
filename: Type of plan to retrieve ('free' or 'work')
|
|
|
|
Returns:
|
|
str: Formatted plan with checkboxes converted to emoji markers
|
|
"""
|
|
try:
|
|
with open(ORG_PLAN_FILENAME.replace("{filename}", filename), "r") as agenda:
|
|
plan = agenda.read()
|
|
|
|
plan = loads(plan)
|
|
|
|
return plan[-1].get_body().replace("[X]", "✅").replace("[ ]", "❌")
|
|
except FileNotFoundError:
|
|
return f"No plan found for {filename}."
|
|
|
|
def add_new_link(self, link: str) -> None:
|
|
"""Add a new link to the links file.
|
|
|
|
Args:
|
|
link: URL to be saved
|
|
"""
|
|
with open(ORG_LINKS_FILENAME, "a") as capture_file:
|
|
capture_file.write(link)
|