Preferences.py

#!/usr/bin/env python3

import os
import logging

import click

from typing import Dict
from configparser import ConfigParser

from src.model import Orders, Preferences

ROOT_PATH = os.path.dirname(os.path.abspath(__name__))
CONFIG_FILE = os.path.join(ROOT_PATH, "preferences.ini")
CONFIG_SECTION = "Preferences"


def bool_to_str(item: bool) -> str:
    if item:
        return "yes"
    else:
        return "no"


def get_preferences() -> Dict:
    parser = ConfigParser()
    parser.read(CONFIG_FILE)

    csv_order = Orders[parser.get(CONFIG_SECTION, "csv_order")]
    use_llm = parser.getboolean(CONFIG_SECTION, "use_llm")
    positive_expenses = parser.getboolean(CONFIG_SECTION, "positive_expenses")
    preferences = Preferences(csv_order, use_llm, positive_expenses)

    return preferences


def write_preferences(order: Orders, use_llm: bool, positive_expenses: bool):
    config = ConfigParser()
    config[CONFIG_SECTION] = {
        "csv_order": order,
        "use_llm": bool_to_str(use_llm),
        "positive_expenses": bool_to_str(positive_expenses),
    }

    with open(CONFIG_FILE, "w+") as config_file:
        config.write(config_file)


@click.command()
@click.option("-o", "--csv-order", type=click.Choice([order.value for order in Orders]), default=Orders.SIMPLE.value,
              help="CSV Order. Defaults to the simple order, which is Date, Amount, and Note (Description)")
@click.option("-l", "--use-llm", is_flag=True, default=False,
              help="Use local LLM (via Ollama) to categorize expenses. NOTE: Experimental Feature. Defaults to False.")
@click.option("-p", "--positive_expenses", is_flag=True, default=False,
              help="Have expenses represented as positive floats and incomes represented as negative floats. Defaults to False.")
def cli(order: Orders, use_llm: bool, positive_expenses: bool):
    """
    Writes an .ini file dictating Ena's behavioural preferences.
    """
    logging.basicConfig(level=logging.INFO)

    write_preferences(order, use_llm, positive_expenses)


if __name__ == "__main__":
    cli()