def process_amenities()

in gemini/agents/research-multi-agents/ev_agent/api_handler/api_01_NeighborhoodSummary.py [0:0]

• 22 lines of code
• 11 McCabe index (conditional complexity)

    def process_amenities(elements: List[Dict]) -> PublicAmenities:
        amenities = PublicAmenities()
        for element in elements:
            tags = element.get("tags", {})

            amenity_type = tags.get("amenity")
            if amenity_type == "post_office":
                amenities.post_offices += 1
            elif amenity_type == "bank":
                amenities.banks += 1
            elif amenity_type == "atm":
                amenities.atms += 1
            elif amenity_type == "toilets":
                amenities.toilets += 1
            elif amenity_type == "recycling":
                amenities.recycling += 1
            elif amenity_type == "waste_disposal":
                amenities.waste_disposal += 1
            elif amenity_type == "water_point" or amenity_type == "drinking_water":
                amenities.water_points += 1
            elif amenity_type == "bench":
                amenities.benches += 1
        return amenities