cli.py

import pandas as pd
from datetime import date, datetime
from enum import Enum, auto
import os
import shutil
from rich.console import Console
from rich.table import Table
from rich.text import Text

USERS_FILE = ".users.csv"
ASSETS_FILE = ".assets.csv"


class Command(Enum):
    ADD_USER = auto()
    ADD_ASSET = auto()
    GET_NETWORTH = auto()
    LIST_USERS = auto()
    LIST_ASSETS = auto()
    HELP = auto()
    EXIT = auto()

    def __str__(self):
        return self.name.lower()


COMMAND_USAGE = {
    Command.ADD_USER: "add_user <username>",
    Command.ADD_ASSET: "add_asset <username> <asset_name> <value>",
    Command.GET_NETWORTH: "get_networth <username>",
    Command.LIST_USERS: "list_users",
    Command.LIST_ASSETS: "list_assets <username>",
    Command.HELP: "help",
    Command.EXIT: "exit",
}

COMMAND_DESCRIPTIONS = {
    Command.ADD_USER: "Adds a new user to the system.",
    Command.ADD_ASSET: "Adds a new asset for a specific user.",
    Command.GET_NETWORTH: "Calculates and displays the net worth of a specific user.",
    Command.LIST_USERS: "Displays a list of all users in the system.",
    Command.LIST_ASSETS: "Lists all assets of a specific user, grouped by asset and sorted chronologically.",
    Command.HELP: "Displays this help message.",
    Command.EXIT: "Exits the application.",
}

COMMAND_EXAMPLES = {
    Command.ADD_USER: "add_user john",
    Command.ADD_ASSET: "add_asset john savings 5000",
    Command.GET_NETWORTH: "get_networth john",
    Command.LIST_USERS: "list_users",
    Command.LIST_ASSETS: "list_assets john",
    Command.HELP: "help",
    Command.EXIT: "exit",
}


def list_assets(username):
    console = Console()
    assets_df = pd.read_csv(ASSETS_FILE)
    assets_df["date"] = pd.to_datetime(assets_df["date"])

    # Filter assets for the given user
    user_assets = assets_df[assets_df["username"] == username]

    if user_assets.empty:
        console.print(f"No assets found for user [bold blue]{username}[/bold blue].")
        return

    # Sort assets by date
    user_assets = user_assets.sort_values("date")

    # Group assets by asset name
    grouped_assets = user_assets.groupby("asset")

    table = Table(title=f"Assets for [bold blue]{username}[/bold blue]")
    table.add_column("Asset", style="cyan", no_wrap=True)
    table.add_column("Date", style="magenta")
    table.add_column("Value", style="green", justify="right")

    for asset, group in grouped_assets:
        table.add_row(f"[bold]{asset}[/bold]", "", "")
        for _, row in group.iterrows():
            table.add_row("", row["date"].strftime("%Y-%m-%d"), f"${row['value']:,.2f}")

    console.print(table)


def display_networth(username, date, networth, max_line_length):
    console = Console()

    # Define the date in the required format

    # Create text components with different styles
    user_text = Text(username, style="bold blue")
    date_text = Text(date, style="bold yellow")
    networth_text = Text(f"${networth:,.2f}", style="bold green")

    # Calculate the number of dots required to fill the line
    base_str = f"Networth of {username} on {date} "
    remaining_length = (
        max_line_length - len(base_str) - len(f"{networth:,.2f}") - 2
    )  # adjusting for length of networth
    dots = "." * remaining_length

    # Create a styled output using rich Text object
    final_text = (
        Text("Networth of ")
        + user_text
        + Text(" on ")
        + date_text
        + Text(f" {dots} ")
        + networth_text
    )

    # Print the result using rich's console
    console.print(final_text)


def initialize_csv_files():
    try:
        pd.read_csv(USERS_FILE)
    except FileNotFoundError:
        pd.DataFrame(columns=["id", "username"]).to_csv(USERS_FILE, index=False)

    try:
        pd.read_csv(ASSETS_FILE)
    except FileNotFoundError:
        pd.DataFrame(columns=["id", "username", "asset", "value", "date"]).to_csv(
            ASSETS_FILE, index=False
        )


def add_user(username):
    users_df = pd.read_csv(USERS_FILE)

    # Check if the username already exists
    if username in users_df["username"].values:
        print(f"User {username} already exists. Cannot add duplicate user.")
        return

    new_id = users_df["id"].max() + 1 if not users_df.empty else 0
    new_user = pd.DataFrame({"id": [new_id], "username": [username]})
    users_df = pd.concat([users_df, new_user], ignore_index=True)
    users_df.to_csv(USERS_FILE, index=False)
    print(f"User {username} added successfully.")


def add_asset(username, asset_name, value):
    assets_df = pd.read_csv(ASSETS_FILE)
    new_id = assets_df["id"].max() + 1 if not assets_df.empty else 0
    new_asset = pd.DataFrame(
        {
            "id": [new_id],
            "username": [username],
            "asset": [asset_name],
            "value": [float(value)],
            "date": [date.today().strftime("%Y-%m-%d")],
        }
    )
    assets_df = pd.concat([assets_df, new_asset], ignore_index=True)
    assets_df.to_csv(ASSETS_FILE, index=False)
    print(f"Asset {asset_name} added for user {username}.")


def get_networth(username):
    assets_df = pd.read_csv(ASSETS_FILE)
    assets_df["date"] = pd.to_datetime(assets_df["date"])

    # Filter assets for the given user
    user_assets = assets_df[assets_df["username"] == username]

    # Group by asset and get the latest entry for each
    latest_assets = user_assets.loc[user_assets.groupby("asset")["date"].idxmax()]

    # Calculate total value of latest assets
    total_value = latest_assets["value"].sum()

    # formatted_value = "{:,.2f}".format(total_value)
    current_date = date.today().strftime("%Y-%m-%d")

    display_networth(username, current_date, total_value, 80)
    # print(f"Networth of {username} on {current_date} is ${formatted_value}")


def list_users():
    try:
        users_df = pd.read_csv(USERS_FILE)
        if users_df.empty:
            print("No users found.")
        else:
            print("\nUser List:")
            print(users_df.to_string(index=False))
    except FileNotFoundError:
        print("Users file not found. No users have been added yet.")


def print_help():
    help_text = "Net Worth Tracker CLI - Available Commands:\n\n"
    for command in Command:
        if command != Command.EXIT:
            help_text += f"{COMMAND_USAGE[command]}\n"
            help_text += f"   {COMMAND_DESCRIPTIONS[command]}\n"
            help_text += f"   Example: {COMMAND_EXAMPLES[command]}\n\n"
    help_text += f"{COMMAND_USAGE[Command.EXIT]}\n"
    help_text += f"   {COMMAND_DESCRIPTIONS[Command.EXIT]}"
    print(help_text)


def parse_command(input_string):
    parts = input_string.split()
    if not parts:
        return None, []

    command_str = parts[0].lower()

    try:
        command = next(cmd for cmd in Command if str(cmd) == command_str)
    except StopIteration:
        return None, []

    return command, parts[1:]


def create_backup(files_to_backup):
    # Generate timestamp for the backup folder
    timestamp = datetime.now().strftime("%Y-%m-%d-%H-%M")

    # Create main backup folder if it doesn't exist
    main_backup_folder = "backups"
    if not os.path.exists(main_backup_folder):
        os.mkdir(main_backup_folder)

    # Create timestamped subfolder
    backup_subfolder = os.path.join(main_backup_folder, timestamp)
    os.mkdir(backup_subfolder)

    # Copy files to the backup subfolder
    for file in files_to_backup:
        if os.path.exists(file):
            shutil.copy2(file, backup_subfolder)
            print(f"Backed up: {file} to {backup_subfolder}")
        else:
            print(f"Warning: {file} not found. Skipping backup.")

    print(f"Backup completed in folder: {backup_subfolder}")


def main():
    initialize_csv_files()
    print("Welcome to the Net Worth Tracker CLI!")
    print_help()

    create_backup([USERS_FILE, ASSETS_FILE])

    while True:
        input_string = input(
            "Enter command (or 'exit' to quit, or 'help' to list the API): "
        )
        command, args = parse_command(input_string)

        if command == Command.EXIT:
            print("Thank you for using Net Worth Tracker CLI. Goodbye!")
            break
        elif command == Command.ADD_USER and len(args) == 1:
            add_user(args[0])
        elif command == Command.ADD_ASSET and len(args) == 3:
            add_asset(args[0], args[1], args[2])
        elif command == Command.GET_NETWORTH and len(args) == 1:
            get_networth(args[0])
        elif command == Command.LIST_USERS and len(args) == 0:
            list_users()
        elif command == Command.LIST_ASSETS and len(args) == 1:
            list_assets(args[0])
        elif command == Command.HELP and len(args) == 0:
            print_help()
        else:
            print(f"Invalid usage of '{command}'. Type 'help' for correct usage.")


if __name__ == "__main__":
    main()