notion_setup.py

import os
import re

from dotenv import load_dotenv
from notion_client import Client

load_dotenv()

notion = Client(auth=os.getenv("NOTION_API_KEY"))
database_id = os.getenv("NOTION_DB_ID") or os.getenv("NOTION_DATABASE_ID")


def sanitize_select_option(value):
    if not value:
        return None

    cleaned = value.replace(",", " ")
    cleaned = re.sub(r"\s+", " ", cleaned).strip()
    return cleaned[:100] or None


def create_task(title, due_date=None, subject=None, description=None, priority="Medium"):
    properties = {
        "Name": {
            "title": [{"text": {"content": title}}],
        },
        "Status": {
            "status": {"name": "Not started"},
        },
        "Priority": {
            "select": {"name": priority},
        },
    }

    if due_date:
        properties["Due Date"] = {
            "date": {"start": due_date},
        }

    subject = sanitize_select_option(subject)
    if subject:
        properties["Subject"] = {
            "select": {"name": subject},
        }

    return notion.pages.create(
        parent={"database_id": database_id},
        properties=properties,
    )