from django.core.exceptions import ObjectDoesNotExist

from apps.vehicle import models


def get_connector_id_by_vehicle_make(vehicle_make_id):
    """
    Get the connector ID based on the given vehicle make ID.

    Args:
        vehicle_make_id (int): The ID of the vehicle make.

    Returns:
        int or None: The connector ID if found, otherwise None.
    """
    try:
        vehicle_model = models.VehicleModelMaster.objects.get(
            vehicle_make__id=vehicle_make_id
        )
        return vehicle_model.connector.id
    except ObjectDoesNotExist:
        return None


def get_connector_id_by_vehicle_model(vehicle_model_id):
    """
    Get the connector ID based on the given vehicle model ID.

    Args:
        vehicle_model_id (int): The ID of the vehicle model.

    Returns:
        int or None: The connector ID if found, otherwise None.
    """
    try:
        vehicle_model = models.VehicleModelMaster.objects.get(id=vehicle_model_id)
        return vehicle_model.connector.id
    except ObjectDoesNotExist:
        return None