changelog.py

"""Utility script for updating the CHANGELOG using Anthropic."""

import subprocess
import os
import requests
from datetime import datetime, timedelta
import anthropic

"""Utility script for updating the CHANGELOG using Anthropic."""
import dotenv

dotenv.load_dotenv()

# Anthropic API key
ANTHROPIC_API_KEY = os.environ.get("ANTHROPIC_API_KEY")


# Function to get git log
def get_git_log(since_date):
    """Return the git commit log entries since *since_date*."""
    cmd = f"git log --since='{since_date}' --pretty=format:'%h - %s (%an, %ad)' --date=short"
    return subprocess.check_output(cmd, shell=True).decode("utf-8")


# Function to read existing CHANGELOG.md
def read_changelog():
    """Return the contents of the existing CHANGELOG.md file."""
    if os.path.exists("CHANGELOG.md"):
        with open("CHANGELOG.md", "r") as f:
            return f.read()
    return ""


# Function to write updated CHANGELOG.md
def write_changelog(content):
    """Write *content* to CHANGELOG.md."""
    with open("CHANGELOG.md", "w") as f:
        f.write(content)


# Function to get the last date from CHANGELOG.md
def get_last_changelog_date(changelog):
    """Extract and return the most recent date entry from the changelog."""
    lines = changelog.split("\n")
    for line in lines:
        if line.startswith("## "):
            try:
                return datetime.strptime(line.split(" ")[1], "%Y-%m-%d")
            except ValueError:
                continue
    return None


# Function to update CHANGELOG using Anthropic API
def update_changelog_with_anthropic(git_log, existing_changelog):
    """Return an updated changelog generated via Anthropic."""
    client = anthropic.Anthropic(api_key=ANTHROPIC_API_KEY)

    system_prompt = f"""
    You are a helpful assistant that creates and updates CHANGELOG.md files.
    """

    prompt = f"""
    Here's the existing CHANGELOG.md content:

    {existing_changelog}

    And here's the new git log entries:

    {git_log}

    Please update the CHANGELOG.md with the following rules:
    1. Add a new section with today's date if it doesn't exist.
    2. Summarize and categorize the new changes under the new section.
    3. Don't duplicate any existing entries.
    4. Keep the existing content intact.
    5. Use Markdown formatting.

    Output the entire updated CHANGELOG.md content.
    """

    message = client.messages.create(
        model="claude-3-5-sonnet-20240620",
        max_tokens=4000,
        temperature=1.0,
        system=system_prompt,
        messages=[
            {
                "role": "user",
                "content": [{"type": "text", "text": prompt}],
            },
        ],
    )

    for m in message.content:
        if isinstance(m, anthropic.types.text_block.TextBlock):
            return m.text


# Main function
def main():
    """Update CHANGELOG.md with entries generated from recent commits."""
    existing_changelog = read_changelog()
    last_date = get_last_changelog_date(existing_changelog)

    if last_date:
        since_date = last_date.strftime("%Y-%m-%d")
    else:
        since_date = (datetime.now() - timedelta(days=180)).strftime("%Y-%m-%d")

    git_log = get_git_log(since_date)

    if git_log:
        updated_changelog = update_changelog_with_anthropic(git_log, existing_changelog)
        write_changelog(updated_changelog)
        print("CHANGELOG.md has been updated.")
    else:
        print("No new changes to add to the CHANGELOG.")


if __name__ == "__main__":
    main()