notion_service.py

import os

import requests
from dotenv import load_dotenv

from notion_setup import create_task
from .calendar_service import create_calendar_event, format_due_display

load_dotenv()

NOTION_API_KEY = os.getenv("NOTION_API_KEY")
DATABASE_ID = os.getenv("NOTION_DB_ID") or os.getenv("NOTION_DATABASE_ID")


def get_all_tasks_debug():
    if not NOTION_API_KEY or not DATABASE_ID:
        return []

    url = f"https://api.notion.com/v1/databases/{DATABASE_ID}/query"
    headers = {
        "Authorization": f"Bearer {NOTION_API_KEY}",
        "Notion-Version": "2022-06-28",
    }

    response = requests.post(url, headers=headers)
    if not response.ok:
        print(f"Notion query failed: {response.status_code} {response.text}")
        return []

    return response.json().get("results", [])


def send_to_notion(assignments):
    for assignment in assignments:
        if not isinstance(assignment, dict):
            print(f"Skipping invalid assignment payload: {assignment}")
            continue

        raw_title = (assignment.get("title") or "").strip()
        if not raw_title:
            print(f"Skipping assignment with missing title: {assignment}")
            continue

        title = raw_title[:100]
        due_date = assignment.get("due_date")
        time = assignment.get("time", "18:00:00")
        subject = assignment.get("subject", "General")

        try:
            create_task(
                title=title,
                due_date=due_date,
                subject=subject,
                description=assignment.get("description"),
                priority=assignment.get("priority", "LOW"),
            )
        except Exception as error:
            print(f"Notion: ERROR failed to create task '{title}': {error}")
            continue

        print(f"Task created: {title}")
        if due_date:
            print(f"Due: {format_due_display(due_date, time)}")
        else:
            print("Due: Not found")
        print(f"Subject: {subject}")
        print("Notion: OK Task added")

        if due_date:
            create_calendar_event(
                title=title,
                due_date=due_date,
                description=assignment.get("description", ""),
                time=time,
            )

        print("\n----------------------\n")