@@ -416,17 +416,18 @@ def populate_fields(self, values, wikidata_id):
416
416
return fields
417
417
418
418
class Processor (object ):
419
- batch_size = 100
420
419
client = OpenChurchClient ()
421
420
redis_url = os .getenv ('REDIS_URL' )
422
421
redis_client = redis .from_url (redis_url )
423
422
424
- def __init__ (self , verbosity_level , type ):
423
+ def __init__ (self , verbosity_level , type , batch_size ):
425
424
self .q = Query (verbosity_level = verbosity_level )
426
425
self .verbosity_level = verbosity_level
427
426
self .type = type
427
+ self .batch_size = batch_size
428
428
429
429
def process_batch (self , data , method , run_id ):
430
+ print (self .batch_size )
430
431
batches = Query .split_into_batches (data , self .batch_size )
431
432
self .redis_client .hset (self .type , "batchCount" , len (batches ))
432
433
iteration = 1
@@ -508,5 +509,5 @@ def get_redis_key(type, origin, to):
508
509
parser .add_argument ("-v" , "--verbose" , action = "count" , default = 0 , help = "Augmente le niveau de verbosité (utilisez -vvv pour plus de détails)." )
509
510
args = parser .parse_args ()
510
511
511
- processor = Processor (verbosity_level = args .verbose , type = args .entity_only )
512
+ processor = Processor (verbosity_level = args .verbose , type = args .entity_only , batch_size = 50 )
512
513
processor .process_entity ()
0 commit comments