from flask import flash, Request
from app.extensions import db
from app.models.pet import Pet
from app.models.pet_kind import PetKind
from app.utils.alert_type import AlertType
from app.utils.errors.pets.pet_register_errors import PetRegisterError
from app.utils.flash_message import FlashMessage
from app.utils.helpers import pet_sex_id_to_str
from app.utils.validators.pet_validators import PetValidators
class PetService:
    
    @staticmethod
    def register_pet(request: Request):
        """Validations"""
        try:
            name =  PetValidators.is_valid_name(request.form.get('name'))
            age = PetValidators.is_valid_age(request.form.get('age'))
            sex = PetValidators.is_valid_sex(request.form.get('sex')) 
            heigth = PetValidators.is_valid_height(request.form.get('height'))
            weight = PetValidators.is_valid_weight(request.form.get('weight'))
            kind = PetValidators.is_valid_type(request.form.get('type'))
            location = PetValidators.is_valid_location(request.form.get('location'))
            """uploading image to cloudinary"""
            """saving the data in the db"""
        except PetRegisterError as e:
            flash(FlashMessage(e.message, AlertType.DANGER.value ))
    
    @staticmethod
    def get_pets(request: Request,results_per_page=8):            
        type = request.args.get('type')
        sex = pet_sex_id_to_str(request.args.get('sex'))
        age_from = request.args.get('age-from')
        age_to = request.args.get('age-to')
        query = db.select(Pet)
        page = request.args.get('page', 1)
        
        if sex != None:
            query = query.filter_by(sex=sex)
        if type != None:
            types = db.session.execute(db.select(PetKind.id)).scalars().all()
            if int(type) in types:
                query = query.filter_by(kind_id=type)
        if age_from != None:
            age_from = int(age_from)
            if age_from >= 0 and age_from <= 7:
                query = query.filter(Pet.age >= age_from)
        if age_to != None:
            age_to = int(age_to)
            if age_to >= 1 and age_to < 7:
                query = query.filter(Pet.age <= age_to)

        try:
            return db.paginate( 
                select=query, 
                page=int(page), 
                per_page=results_per_page,
                )
        except:
            return  db.paginate( 
                select=query, 
                page=1, 
                per_page=results_per_page,
                )
    
    @staticmethod
    def get_options(request: Request):
        type = request.args.get('type')
        sex = request.args.get('sex')
        age_from = request.args.get('age-from')
        options = dict()
        options["type"] = PetService.get_pets_kind_options(type)
        options["sex"] = PetService.get_sex_options(sex)
        options["age_from"] = PetService.get_age_from_options(age_from)
        return options
    
    @staticmethod
    def get_pets_kind():
        return db.session.execute(db.select(PetKind)).scalars()
    
    @staticmethod
    def get_pets_kind_options(selected_kind=None):
        pet_kinds = db.session.execute(db.select(PetKind)).scalars()
        options = []
        options.append({
                "value": "",
                "text": "Type",
                "selected": selected_kind == None
            })
        options.append({
                "value": "0",
                "text": "Any",
                "selected": selected_kind == "0"
            })
        for pet_kind in pet_kinds:
            options.append({
                "value": str(pet_kind.id),
                "text": pet_kind.name,
                "selected": selected_kind == str(pet_kind.id)
            })
        PetService.selected_safeguard(selected_kind,options[0],len(options))
        return options

    @staticmethod
    def get_age_from_options(selected_year=None):
        options = []
        options.append({
                "value": "0",
                "text": "0",
                "selected": selected_year == "0" or selected_year == None
            })
        for i in range(1,7):
            options.append({
                "value": str(i),
                "text": str(i),
                "selected": selected_year == str(i)
            })
        options.append({
                        "value": "7",
                        "text": "7+",
                        "selected": selected_year == "7"
                    })
        PetService.selected_safeguard(selected_year,options[0],len(options))
        return options
    
    @staticmethod
    def get_sex_options(selected_sex=None):
        options = [
            {
                "value": "",
                "text": "Sex",
                "selected": selected_sex == None
            },
            {
                "value": "0",
                "text": "Any",
                "selected": selected_sex == "0" 
            },
                        {
                "value": "1",
                "text": "Female",
                "selected": selected_sex == "1" 
            },
            {
                "value": "2",
                "text": "Male",
                "selected": selected_sex == "2" 
            },
        ]
        PetService.selected_safeguard(selected_sex,options[0],len(options))
        return options
    
    @staticmethod
    def selected_safeguard(selected_arg: str | None, select_default, options_length: int):
        try:
            selected_arg = int(selected_arg)
            if selected_arg < 0 or selected_arg > options_length:
                select_default["selected"] = True
        except:
            select_default["selected"] = True