""" This file contains functionality to find data to the database. """ from app.database.models import Competition, Media, Question, Slide, User def image(pagination_parameters=None, filename=None, order=1, order_by=None): """Finds and returns an image from the file name.""" query = Media.query.filter(Media.type_id == 1) if filename: query = query.filter(Media.filename.like(f"%{filename}%")) return query.paginate_api(pagination_parameters) def user( pagination_parameters=None, email=None, name=None, city_id=None, role_id=None, ): """Finds and returns any number of users from the provided parameters.""" query = User.query if name: query = query.filter(User.name.like(f"%{name}%")) if email: query = query.filter(User.email.like(f"%{email}%")) if city_id: query = query.filter(User.city_id == city_id) if role_id: query = query.filter(User.role_id == role_id) return query.paginate_api(pagination_parameters, order_columns=(User.name,)) def competition( pagination_parameters=None, name=None, year=None, city_id=None, ): """Finds and returns a competition from the provided parameters.""" query = Competition.query if name: query = query.filter(Competition.name.like(f"%{name}%")) if year: query = query.filter(Competition.year == year) if city_id: query = query.filter(Competition.city_id == city_id) return query.paginate_api(pagination_parameters, order_columns=(Competition.year.desc(), Competition.name)) def slide( pagination_paramters=None, slide_order=None, title=None, body=None, competition_id=None, ): """Finds and returns a slide from the provided parameters.""" query = Slide.query if slide_order: query = query.filter(Slide.order == slide_order) if title: query = query.filter(Slide.title.like(f"%{title}%")) if body: query = query.filter(Slide.body.like(f"%{body}%")) if competition_id: query = query.filter(Slide.competition_id == competition_id) return query.paginate_api(pagination_paramters) def questions( name=None, total_score=None, type_id=None, slide_id=None, competition_id=None, page=0, page_size=15, order=1, order_by=None, ): """Finds and returns a question from the provided parameters.""" query = Question.query if name: query = query.filter(Question.name.like(f"%{name}%")) if total_score: query = query.filter(Question.total_score == total_score) if type_id: query = query.filter(Question.type_id == type_id) if slide_id: query = query.filter(Question.slide_id == slide_id) if competition_id: query = query.join( Slide, (Slide.competition_id == competition_id) & (Slide.id == Question.slide_id), ) order_column = Question.id # Default order_by if order_by: order_column = getattr(Question.__table__.c, order_by) return query.pagination(page, page_size, order_column, order)