160 lines
5.6 KiB
Python
160 lines
5.6 KiB
Python
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
|