c-busqueda/handler.cr

@@ -177,6 +177,7 @@ class Handler
     DB.open(DB_URL) do |cursor|
       cursor.query sql do |result_set|
         result_set.each do
+          puts "loop"
           valor = result_set.read(Int32)
           nombre = result_set.read(String)
           datos.push({valor, nombre})
@@ -202,11 +203,11 @@ class Handler
 
     if genero
       DB.open(DB_URL) do |cursor|
-        datos = datos.find do |item|
+        filtered = Array(Tuple(Int32, String)).new
+        datos.map { |item|
           # How feminine is this name?
           # Yes this database is upper case
           nombre = item[1].upcase
-          puts "Checking #{nombre}"
           feminidad = 0
           sql = %(
             SELECT COALESCE((SELECT frecuencia FROM mujeres WHERE nombre='#{nombre}'), 0) AS mujeres,
@@ -222,13 +223,13 @@ class Handler
             end
           end
           # El overlap en 0.5 es intencional!
-          if (feminidad >= 0.5 && genero == "f") ||
-             (feminidad <= 0.5 && genero == "m")
-            true
-          else
-            false
-          end
+          if feminidad >= 0.5 && genero == "f"
+            filtered << item
+          elsif feminidad <= 0.5 && genero == "m"
+            filtered << item
           end
+        }
+        datos = filtered
         puts "Data split by gender"
       end
     end