Przeglądaj źródła

Tamaño de segmento configurable

Hugo 3 lat temu
rodzic
commit
10d2d6b895
1 zmienionych plików z 5 dodań i 1 usunięć
  1. 5 1
      ondemand/ondemand.py

+ 5 - 1
ondemand/ondemand.py

@@ -128,6 +128,9 @@ def procesar_siguiente_pila():
 def procesar_trabajo(pendiente):
     ciudad = pendiente['origen']
     estacion = pendiente['estacion']
+    segmento = 10
+    if "segmento" in pendiente:
+        segmento = int(pendiente["segmento"])
 
     # Descarga de anuncios
     log.info("Descargando anuncios")
@@ -179,7 +182,7 @@ def procesar_trabajo(pendiente):
 
         for anuncio in anuncios:
             log.info("Buscando anuncio %s" % (anuncio,))
-            for i in dejavu.recognize(recognizer, anuncio, 10):
+            for i in dejavu.recognize(recognizer, anuncio, segmento):
                 if not "id" in i:
                     continue
 
@@ -226,6 +229,7 @@ def procesar_trabajo(pendiente):
                     break
 
         log.info("[Resultado] %s" % (json.dumps(resultados)))
+        pendiente["media"] = None
         enviar_resultados(pendiente)
 
     except Exception as ex: