"""
General-purpose utilities for handling command-line
arguments and flags in ctsm python codes.
Types for command-lines error handling.
"""

import logging
import argparse

from ctsm.config_utils import lon_range_0_to_360

logger = logging.getLogger(__name__)


def plat_type(plat):
    """
    Function to define lat type for the parser
    and
    raise error if latitude is not between -90 and 90.

    Args:
        plat(str): latitude
    Raises:
        Error (ArgumentTypeError): when plat (latitude) is not between -90 and 90.
    Returns:
        plat_out (float): latitude in float
    """
    plat_out = float(plat)
    if plat_out < -90 or plat_out > 90:
        raise argparse.ArgumentTypeError("ERROR: Latitude should be between -90 and 90.")
    return plat_out


def plon_type(plon):
    """
    Function to define lon type for the parser and
    convert negative longitudes to 0-360 and
    raise error if lon is not between -180 and 360.

    Args:
        plon (str): longitude
    Raises:
        Error (ArgumentTypeError): when longitude is <-180 and >360.
    Returns:
        plon_out (float): converted longitude between 0 and 360
    """
    plon_float = float(plon)
    if plon_float < -180 or plon_float > 360:
        raise argparse.ArgumentTypeError(
            "ERROR: Longitude should be between 0 and 360 or -180 and 180."
        )
    plon_out = lon_range_0_to_360(plon_float)
    return plon_out