from apps.companies.models import CompanyBotSettings
from django.utils import timezone


def get_bot_name(phone_number):
    bot = CompanyBotSettings.objects.get(phone_number=phone_number)
    return bot.bot_name if bot else None

def get_date_range_from_request(request, serializer_cls):
    serializer = serializer_cls(data=request.query_params)
    serializer.is_valid(raise_exception=True)
    validated = serializer.validated_data

    start_dt = validated["start_date"]
    end_dt = validated["end_date"]

    if timezone.is_naive(start_dt):
        start_dt = timezone.make_aware(start_dt)

    if timezone.is_naive(end_dt):
        end_dt = timezone.make_aware(end_dt)

    return start_dt, end_dt