def get_abilities(search=None, unused=False, query_gerneration=None): abilities = [] if search is None: search = "" abilities = [] for ability in Ability.select().order_by(Ability.id): if search in ability.name: abilities.append(ability) if unused: abilities = [ ability for ability in abilities if len(ability.pokemons) == 0 ] if query_gerneration is not None: filtered_abilities = [] for ability in abilities: # abilities = [] generation_de_ce_ability = Generation.select().where( Generation.id == ability.generation) # print((generation_de_ce_ability[0].name)) if query_gerneration in generation_de_ce_ability[0].name: filtered_abilities.append(ability) abilities = filtered_abilities return abilities
def search_abilities(query=None, limits=None, offset=None): """:return abilities containing query :parameter limits limit the number of abilities :parameter offset the offset""" if limits == None: limits = 10 if offset == None: offset = 0 abilities = (Ability.select(Ability.id, Ability.name, Ability.is_main_series, Ability.generation).join(Generation)).order_by( Ability.name).limit(limits).offset(offset) if query != None: abilities = abilities.where(query in Ability.name) if len(abilities) > (limits + offset): abilities = abilities.limits(limits).offset(offset) data = [ dict({ "id": ability.id, "name": ability.name, "is_main_series": ability.is_main_series, "generation": ability.generation.name }) for ability in abilities ] return data
def get_abilities(generation=None, limit=None, offset=None): abilities = Ability.select().offset(offset).limit(limit) if generation is not None: generation_id = Generation.get_or_none(name=generation) if generation_id is not None: abilities = Generation.select().where( Generation.name == generation) return abilities
def get_number_of_abilities_by_generation(generation=None): ability_of_generation = Ability.select().where( Ability.generation == generation) number_of_abilities_of_generation = ability_of_generation.count() return number_of_abilities_of_generation
def search_abilities(query, limit=None): query = query.lower() abilities = Ability.select().where(Ability.name.contains(query)) return abilities
def get_abilities(): abilities = Ability.select() return abilities
def number_of_abilities(generation_id): counter = len( Ability.select().where(Ability.generation_id == generation_id)) return counter