生成器函数执行两次?

2024-06-16 13:01:54 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在使用python生成器函数为我提供当前目录中的图像列表。然而,我看到这个函数发出整个列表两次而不是一次,我不知道为什么。我正在使用Python PIL库创建批缩略图。在

谁能给我指出正确的方向吗?在

脚本:


import os
import sys
import Image

class ThumbnailGenerator:
    def __init__(self, width, height, image_path, thumb_path):
        self.width = width
        self.height = height
        self.image_path = image_path
        self.thumb_path = "%s%s%s" % (self.image_path, os.sep, thumb_path)

    def __call__(self):
        self.__create_thumbnail_dir()

        for filename, image in self.__generate_image_list():
            try:
                thumbnail = "%s%s%s" % (self.thumb_path, os.sep, filename)
                image.thumbnail((self.width, self.height))
                image.save(thumbnail, 'JPEG')
                print "Thumbnail gemaakt voor: %s" % filename
            except IOError:
                print "Fout: thumbnail kon niet gemaakt worden voor: %s" % filename

    def __generate_image_list(self):
        for dirpath, dirnames, filenames in os.walk(self.image_path):
            count = 0
            for filename in filenames:
                try:
                    image = Image.open(filename)
                    print '=========', count, filename
                    count += 1
                    yield (filename, image)
                except IOError:
                    pass

    def __create_thumbnail_dir(self):
        try:
            os.mkdir(self.thumb_path)
        except OSError as exception:
            print "Fout: %s" % exception

if __name__ == '__main__':
    try:
        thumbnail_generator = ThumbnailGenerator(80, 80, '.', 'thumbs')
        thumbnail_generator()
    except keyboardInterrupt:
        print 'Programma gestopt'

此时脚本的输出(带有一些测试图像)是:

^{pr2}$

它应该是:

========= 0 124415main_image_feature_380a_ys_full.jpg
Thumbnail gemaakt voor: 124415main_image_feature_380a_ys_full.jpg
========= 1 60130main_image_feature_182_jwfull.jpg
Thumbnail gemaakt voor: 60130main_image_feature_182_jwfull.jpg
========= 2 assetImage.jpg
Thumbnail gemaakt voor: assetImage.jpg
========= 3 devcon-c1-image.gif
Fout: thumbnail kon niet gemaakt worden voor: devcon-c1-image.gif
========= 4 image-646313.jpg
Thumbnail gemaakt voor: image-646313.jpg
========= 5 Image-Schloss_Nymphenburg_Munich_CC.jpg
Thumbnail gemaakt voor: Image-Schloss_Nymphenburg_Munich_CC.jpg
========= 6 image1w.jpg
Thumbnail gemaakt voor: image1w.jpg
========= 7 New%20Image.jpg
Thumbnail gemaakt voor: New%20Image.jpg
========= 8 samsung-gx20-image.jpg
Thumbnail gemaakt voor: samsung-gx20-image.jpg
========= 9 samsung-image.jpg
Thumbnail gemaakt voor: samsung-image.jpg

如您所见,generator函数返回列表两次(我验证了它,它只被调用一次)。在

@heikogerlach: 手术室步行无法找到缩略图,因为我正在遍历当前目录的文件名,并且缩略图被写入当前目录名为“thumb”的子文件夹中。该列表是在将缩略图写入“thumb”目录之前生成的,我验证了(使用WinPDB)缩略图不包括在列表中。在

@S.Lott: 谢谢你的建议。os.path.join操作系统解决了这个问题。在


Tags: pathimageself列表osfilenamejpgprint
2条回答

在调试中,打印完整路径。我想您在遍历.目录之后,正在遍历thumbs子目录。在

还有。在

class ThumbnailGenerator( object ):

从长远来看通常效果更好。在

请不要在方法名前面使用__(生成图像列表并创建缩略图目录)。在

不要使用"%s%s%s" % (self.image_path, os.sep, thumb_path)创建路径名,请使用os.path.join。在

您的缩略图位于self.image_path的子目录中,与原始图像同名。你能检查一下walk在你创建thumnails时是否找到了它们?只需将图像的路径和名称一起打印。在

相关问题 更多 >