Module: postgraphs¶

def find_graph_files(config:ConfigParser) -> list:
    """
    Find the graph image and its corresponding alt text file.

    Args:
      config: A ConfigParser object from the [config](module-config.md) module

    Returns:
        A tuple containing (graph_image_path, graph_alt_text) or (None, None) if not found
    """
    hashtag = config.get("mastoscore", "hashtag")
    year = config.get("mastoscore", "event_year")
    month = config.get("mastoscore", "event_month")
    day = config.get("mastoscore", "event_day")
    date_str = f"{year}{month}{day}"

    # Define paths
    graphs_dir = create_journal_directory(config)
    graph_pattern = os.path.join(graphs_dir, f"{hashtag}-{date_str}.png")
    alt_text_pattern = os.path.join(graphs_dir, f"{hashtag}-{date_str}.txt")

    # Find files
    graph_files = glob.glob(graph_pattern)
    alt_text_files = glob.glob(alt_text_pattern)

    if graph_files and os.path.exists(graph_files[0]):
        graph_image_path = graph_files[0]

        # Read alt text if available
        if alt_text_files and os.path.exists(alt_text_files[0]):
            with open(alt_text_files[0], "r") as f:
                graph_alt_text = f.read()
        else:
            graph_alt_text = f"Graph of #{hashtag} activity on {date_str}"

        return graph_image_path, graph_alt_text

    return None, None

def find_wordcloud_files(config:ConfigParser) -> list:
    """
    Find the wordcloud image and its corresponding alt text file.

    Args:
        config: A ConfigParser object from the [config](module-config.md) module

    Returns:
        A tuple containing (wordcloud_image_path, wordcloud_alt_text) or (None, None) if not found
    """
    hashtag = config.get("mastoscore", "hashtag")
    year = config.get("mastoscore", "event_year")
    month = config.get("mastoscore", "event_month")
    day = config.get("mastoscore", "event_day")
    date_str = f"{year}{month}{day}"

    # Get hashtag_fix value, default to 'as-is'
    hashtag_fix = config.get("wordcloud", "hashtag_fix", fallback="as-is")

    # Define paths
    wordcloud_dir = create_journal_directory(config)
    wordcloud_pattern = os.path.join(
        wordcloud_dir, f"wordcloud-{hashtag}-{date_str}-{hashtag_fix}.png"
    )
    alt_text_pattern = os.path.join(
        wordcloud_dir, f"wordcloud-{hashtag}-{date_str}-{hashtag_fix}.txt"
    )

    # Find files
    wordcloud_files = glob.glob(wordcloud_pattern)
    alt_text_files = glob.glob(alt_text_pattern)

    if wordcloud_files and os.path.exists(wordcloud_files[0]):
        wordcloud_image_path = wordcloud_files[0]

        # Read alt text if available
        if alt_text_files and os.path.exists(alt_text_files[0]):
            with open(alt_text_files[0], "r") as f:
                wordcloud_alt_text = f.read()
        else:
            wordcloud_alt_text = f"Word cloud for #{hashtag} on {date_str}"

        return wordcloud_image_path, wordcloud_alt_text

    return None, None

def post_graphs(config:ConfigParser) -> int:
    """
    Find graph and wordcloud images with their alt text and post them to Mastodon as threaded replies.
    Histogram is posted as a reply to the last text post, wordcloud as a reply to histogram.

    ## Config Parameters Used
    | Option | Description |
    | ------- | ------- |
    | `postgraphs:message` | Custom message to include with the post |
    | `post:thread_visibility` | Visibility for graph posts (from post config) |
    | `mastoscore:hashtag` | Hashtag used for the analysis |

    Args:
      config: A ConfigParser object from the [config](module-config.md) module

    Returns:
      Number of posts made (0, 1, or 2)
    """
    from .config import get_debug_level
    from .fetch import read_json

    # Get configuration values
    debug = get_debug_level(config, "mastoscore")
    thread_visibility = config.get("post", "thread_visibility", fallback="unlisted")
    custom_message = config.get("postgraphs", "message", fallback="")
    hashtag = config.get("mastoscore", "hashtag")

    # Set up logging
    logger = logging.getLogger(__name__)
    logging.basicConfig(
        format="%(asctime)s %(levelname)-8s %(message)s",
        level=logging.ERROR,
        datefmt="%H:%M:%S",
    )
    logger.setLevel(debug)

    # Get the last post ID from analysis results to thread from
    analysis = read_json(config, "analysis")
    reply_to_id = None
    if analysis and "tootlist" in analysis and analysis["tootlist"]:
        # Get the last post ID from the text posts
        reply_to_id = analysis["tootlist"][-1]
        logger.info(f"Will thread graphs from post ID: {reply_to_id}")

    # Find graph and wordcloud files
    graph_image_path, graph_alt_text = find_graph_files(config)
    wordcloud_image_path, wordcloud_alt_text = find_wordcloud_files(config)

    if not graph_image_path and not wordcloud_image_path:
        logger.error("No graph or wordcloud images found")
        return 0

    # Create Tooter object
    try:
        tooter = Tooter(config, "post")  # Use 'post' credentials
    except Exception as e:
        logger.critical("Failed to create Tooter object")
        logger.critical(e)
        return 0

    posts_made = 0
    graph_urls = []

    # Post histogram first
    if graph_image_path:
        try:
            media = tooter.media_post(graph_image_path, description=graph_alt_text)
            message = f"{custom_message}\n\n" if custom_message else ""
            message += f"{graph_alt_text}\n\n#{hashtag}"

            status = tooter.status_post(
                message,
                media_ids=[media["id"]],
                visibility=thread_visibility,
                in_reply_to_id=reply_to_id
            )
            logger.info(f"Posted histogram: {status['url']}")
            graph_urls.append({"name": "histogram", "url": status["url"]})
            reply_to_id = status["id"]  # Next post replies to this
            posts_made += 1
        except Exception as e:
            logger.error(f"Failed to post histogram: {e}")

    # Post wordcloud as reply to histogram
    if wordcloud_image_path:
        try:
            media = tooter.media_post(wordcloud_image_path, description=wordcloud_alt_text)
            message = f"{wordcloud_alt_text}\n\n#{hashtag}"

            status = tooter.status_post(
                message,
                media_ids=[media["id"]],
                visibility=thread_visibility,
                in_reply_to_id=reply_to_id
            )
            logger.info(f"Posted wordcloud: {status['url']}")
            graph_urls.append({"name": "wordcloud", "url": status["url"]})
            posts_made += 1
        except Exception as e:
            logger.error(f"Failed to post wordcloud: {e}")

    # Save graph URLs to data-{hashtag}-posts.json
    if graph_urls:
        from .fetch import update_json
        update_json(config, "posts", {"graph_posts": graph_urls})

    return posts_made