utils.py

import csv
from typing import Any, Dict, Iterable, Optional
from pathlib import Path


def isnumber(v):
    try:
        int(v)
    except ValueError:
        return False
    return True


def get_data_from(filename):
    path = Path(__file__).parent / "csv" / filename

    with open(path) as f:
        reader = csv.DictReader(f)
        data = list({k: int(v) if isnumber(v) else v for k, v in row.items() if v != ""} for row in reader)

    return data


def comma_formatted(iterable: Iterable, *, conjunction: Optional[str] = "and") -> str:
    """Function to take in a list and return a comma formatted string of its elements"""

    iterable = list(iterable)
    if len(iterable) == 0:
        return ""
    elif len(iterable) == 1:
        return str(iterable[0])

    return f"{', '.join(map(str, iterable[:-1]))} {conjunction} {iterable[-1]}"


def unwind(dictionary: Dict[tuple, Any], *, include_values: Optional[bool] = False):
    """Unwinds a dictionary with tuples keys, returning a dictionary where each tuple element is assigned to their respective values"""

    result = {key: value for tuple, value in dictionary.items() for key in tuple}
    # If include_values is true, add each item's value as a key aswell.
    # Useful for shortcutting items to include the original key
    if include_values is True:
        result.update({v: v for k, v in dictionary.items()})

    return result