@@ -247,16 +247,16 @@ def json_writer(filename, fields, task_queue, error_queue, format):
247247 pass
248248
249249
250- def json_gz_writer (filename , fields , task_queue , error_queue , format , compression_level ):
250+ def jsongz_writer (filename , fields , task_queue , error_queue , format , compression_level ):
251251 try :
252252 with open (filename , "wb" ) as out :
253253 # wbits 31 = MAX_WBITS + gzip header and trailer
254254 compressor = zlib .compressobj (compression_level , zlib .DEFLATED , 31 )
255- def compress_write (str ):
255+ def compress_and_write (str ):
256256 out .write (compressor .compress (str .encode ("utf-8" )))
257257
258258 first = True
259- compress_write ("[" )
259+ compress_and_write ("[" )
260260 item = task_queue .get ()
261261 while not isinstance (item , StopIteration ):
262262 row = item [0 ]
@@ -265,15 +265,15 @@ def compress_write(str):
265265 if item not in fields :
266266 del row [item ]
267267 if first :
268- compress_write ("\n " )
268+ compress_and_write ("\n " )
269269 first = False
270270 else :
271- compress_write (",\n " )
271+ compress_and_write (",\n " )
272272
273- compress_write (json .dumps (row ))
273+ compress_and_write (json .dumps (row ))
274274 item = task_queue .get ()
275275
276- compress_write ("\n ]\n " )
276+ compress_and_write ("\n ]\n " )
277277 out .write (compressor .flush ())
278278 except BaseException :
279279 ex_type , ex_class , tb = sys .exc_info ()
@@ -392,7 +392,7 @@ def export_table(
392392 elif options .format == "jsongz" :
393393 filename = directory + "/%s/%s.jsongz" % (db , table )
394394 writer = multiprocessing .Process (
395- target = json_gz_writer ,
395+ target = jsongz_writer ,
396396 args = (
397397 filename ,
398398 options .fields ,
0 commit comments