Manage Mail with the Kiteworks REST API

import requests
from get_access_token import KWOAuthClient

def create_email_draft(base_url, access_token, subject, body, recipient):
    """Create an email draft and return its mail_id."""
    url = f"{base_url}/rest/mail/actions/sendFile?returnEntity=true"
    headers = {
        "X-Accellion-Version": "28",
        "Authorization": f"Bearer {access_token}",
    }
    payload = {
        "subject": subject,
        "body": body,
        "to": [recipient],
        "cc": [],
        "bcc": [],
        "draft": "true",
        "secureBody": "false",
    }
    response = requests.post(url, headers=headers, json=payload)
    response.raise_for_status()
    return response.json()["id"]

import os
import requests
from get_access_token import KWOAuthClient

def send_mail(base_url, access_token, file_name, file_path, subject, body, recipient):
    """
    Compose an email, attach a file, and send it.

    file_name:   Name the attachment will have on the email
    file_path:   Local path to the file, e.g. "/tmp/report.pdf"
    subject:     Email subject line
    body:        Email message body
    recipient:   Recipient email address
    """
    headers = {
        "X-Accellion-Version": "28",
        "Authorization": f"Bearer {access_token}",
    }

    # Step 1: Create the email draft
    create_url = f"{base_url}/rest/mail/actions/sendFile?returnEntity=true"
    create_payload = {
        "subject": subject,
        "body": body,
        "to": [recipient],
        "draft": "true",
        "secureBody": "false",
    }
    create_response = requests.post(create_url, headers=headers, json=create_payload)
    create_response.raise_for_status()
    mail_id = create_response.json()["id"]
    print(f"Draft created. mail_id: {mail_id}")

    # Step 2a: Initiate the attachment upload
    file_size = os.path.getsize(file_path)
    initiate_url = f"{base_url}/rest/mail/{mail_id}/actions/initiateUpload?returnEntity=true"
    initiate_payload = {
        "filename": file_name,
        "totalSize": file_size,
        "totalChunks": 1,
    }
    initiate_response = requests.post(initiate_url, headers=headers, json=initiate_payload)
    initiate_response.raise_for_status()
    # Mail attachment uploads use the uri field, not the id field
    upload_uri = initiate_response.json()["uri"]

    # Step 2b: Upload the attachment content
    upload_url = f"{base_url}/{upload_uri}?returnEntity=true"
    with open(file_path, "rb") as f:
        payload = {
            "compressionMode": (None, "NORMAL"),
            "compressionSize": (None, str(file_size)),
            "originalSize": (None, str(file_size)),
            "index": (None, "1"),
            "lastChunk": (None, "1"),
            "content": (file_name, f),
        }
        upload_response = requests.post(upload_url, headers=headers, files=payload)

    if upload_response.status_code != 201:
        print(f"Attachment upload failed: {upload_response.status_code} — {upload_response.text}")
        return

    file_id = upload_response.json()["id"]
    print(f"Attachment uploaded. file_id: {file_id}")

    # Step 3: Send the email
    send_url = f"{base_url}/rest/mail/{mail_id}/actions/sendFile?returnEntity=true"
    send_payload = {
        "files": [file_id],
        "draft": "false",
        "uploading": 0,
    }
    send_response = requests.put(send_url, headers=headers, json=send_payload)

    if send_response.status_code == 200:
        print("Email sent successfully.")
    else:
        print(f"Failed to send email: {send_response.status_code} — {send_response.text}")

def main():
    base_url = input("Enter Base URL for Kiteworks instance: ")
    client = KWOAuthClient(base_url)
    access_token = client.get_access_token()
    file_path = input("Enter local file path: ")
    file_name = input("Enter attachment name: ")
    subject = input("Enter email subject: ")
    body = input("Enter email body: ")
    recipient = input("Enter recipient email address: ")
    send_mail(base_url, access_token, file_name, file_path, subject, body, recipient)

if __name__ == "__main__":
    main()