from measurement.measures import Weight


class WeightUnits:
    # KILOGRAM = "kg"
    POUND = "lb"
    OUNCE = "oz"
    # GRAM = "g"

    CHOICES = [
        # (KILOGRAM, "kg"),
        (POUND, "lb"),
        (OUNCE, "oz"),
        # (GRAM, "g"),
    ]


def zero_weight():
    """Represent the zero weight value."""
    return Weight(lb=0)


def convert_weight(weight: Weight, unit: str) -> Weight:
    """Covert weight to given unit and round it to 3 digits after decimal point."""
    # Weight amount from the Weight instance can be retrived in serveral units
    # via its properties. eg. Weight(lb=10).kg
    converted_weight = getattr(weight, unit)
    weight = Weight(**{unit: converted_weight})
    weight.value = round(weight.value, 3)
    return weight


def convert_weight_to_default_weight_unit(weight: Weight) -> Weight:
    """Weight is kept in one unit, but should be returned in site default unit."""
    default_unit = get_default_weight_unit()
    if weight is not None:
        if weight.unit != default_unit:
            weight = convert_weight(weight, default_unit)
        else:
            weight.value = round(weight.value, 3)
    return weight