Fix pickle error with pool_init

This commit is contained in:
Simon Conseil
2022-04-05 19:28:25 +02:00
parent 246aed53ff
commit 5c287be4c1

View File

@@ -793,11 +793,11 @@ class Gallery:
self.logger.info("Using %s cores", ncpu)
if ncpu > 1:
def pool_init():
if self.settings['max_img_pixels']:
PILImage.MAX_IMAGE_PIXELS = self.settings['max_img_pixels']
self.pool = multiprocessing.Pool(processes=ncpu, initializer=pool_init)
self.pool = multiprocessing.Pool(
processes=ncpu,
initializer=pool_init,
initargs=(self.settings['max_img_pixels'],),
)
else:
self.pool = None
@@ -931,6 +931,11 @@ class Gallery:
yield f
def pool_init(max_img_pixels):
if max_img_pixels:
PILImage.MAX_IMAGE_PIXELS = max_img_pixels
def process_file(media):
processor = None
if media.type == 'image':