コード例 #1
0
 def process_zipfile(self):
     if os.path.isfile(self.zip_file.path):
         # TODO: implement try-except here
         zip = zipfile.ZipFile(self.zip_file.path)
         bad_file = zip.testzip()
         if bad_file:
             raise Exception('"%s" in the .zip archive is corrupt.' % bad_file)
         count = 1
         album = self.album
         if not album:
             album = Album.objects.create(name=self.new_album_name)
         from cStringIO import StringIO
         for filename in sorted(zip.namelist()):
             if filename.startswith('__'): # do not process meta files
                 continue
             data = zip.read(filename)
             if len(data):
                 try:
                     # the following is taken from django.newforms.fields.ImageField:
                     #  load() is the only method that can spot a truncated JPEG,
                     #  but it cannot be called sanely after verify()
                     trial_image = PILImage.open(StringIO(data))
                     trial_image.load()
                     # verify() is the only method that can spot a corrupt PNG,
                     #  but it must be called immediately after the constructor
                     trial_image = PILImage.open(StringIO(data))
                     trial_image.verify()
                 except Exception:
                     # if a "bad" file is found we just skip it.
                     continue
                 img = Image(album=album)
                 img.image.save(filename, ContentFile(data))
                 img.save()
         zip.close()
         return album
コード例 #2
0
    def setUp(self):
        self.plugin = ImagesiftPlugin()

        self.image_file = open(
            os.path.join(os.path.dirname(__file__), 'test_img.jpg'))
        details = (
            ('i1', 'alpha            rho'),
            ('i2', '      beta       rho'),
            ('i3', '           gamma rho'),
        )
        self.image_count = len(details)
        for title, tags in details:
            Image(title=title, image=File(self.image_file), tags=tags).save()
コード例 #3
0
ファイル: upload.py プロジェクト: valeryz/imagestore
def process_zipfile(uploaded_album):
    if os.path.isfile(uploaded_album.zip_file.path):
        # TODO: implement try-except here
        zip = zipfile.ZipFile(uploaded_album.zip_file.path)
        bad_file = zip.testzip()
        if bad_file:
            raise Exception('"%s" in the .zip archive is corrupt.' % bad_file)

        if not uploaded_album.album:
            uploaded_album.album = Album.objects.create(
                name=uploaded_album.new_album_name)

        from cStringIO import StringIO
        for filename in sorted(zip.namelist()):
            if filename.startswith('__'):  # do not process meta files
                continue
            print filename.encode('ascii', errors='replace')
            data = zip.read(filename)
            if len(data):
                try:
                    # the following is taken from django.forms.fields.ImageField:
                    # load() could spot a truncated JPEG, but it loads the entire
                    # image in memory, which is a DoS vector. See #3848 and #18520.
                    # verify() must be called immediately after the constructor.
                    PILImage.open(StringIO(data)).verify()
                except Exception, ex:
                    # if a "bad" file is found we just skip it.
                    print('Error verify image: %s' % ex.message)
                    continue
                if hasattr(data, 'seek') and callable(data.seek):
                    print 'seeked'
                    data.seek(0)
                try:
                    img = Image(album=uploaded_album.album)
                    img.image.save(filename, ContentFile(data))
                    img.save()
                except Exception, ex:
                    print('error create Image: %s' % ex.message)
コード例 #4
0
ファイル: upload.py プロジェクト: DylannCordel/imagestore
def process_zipfile(uploaded_album):
    if default_storage.exists(uploaded_album.zip_file.name):
        # TODO: implement try-except here
        zip = zipfile.ZipFile(uploaded_album.zip_file)
        bad_file = zip.testzip()
        if bad_file:
            raise Exception('"%s" in the .zip archive is corrupt.' % bad_file)

        if not uploaded_album.album:
            uploaded_album.album = Album.objects.create(name=uploaded_album.new_album_name)

        for filename in sorted(zip.namelist()):
            if filename.startswith('__'):  # do not process meta files
                continue
            print(filename.encode('ascii', errors='replace'))
            data = zip.read(filename)
            if len(data):
                try:
                    # the following is taken from django.forms.fields.ImageField:
                    # load() could spot a truncated JPEG, but it loads the entire
                    # image in memory, which is a DoS vector. See #3848 and #18520.
                    # verify() must be called immediately after the constructor.
                    PILImage.open(six.moves.cStringIO(data)).verify()
                except Exception as ex:
                    # if a "bad" file is found we just skip it.
                    print(('Error verify image: %s' % ex.message))
                    continue
                if hasattr(data, 'seek') and isinstance(data.seek, collections.Callable):
                    print('seeked')
                    data.seek(0)
                try:
                    img = Image(album=uploaded_album.album)
                    img.image.save(filename, ContentFile(data))
                    img.save()
                except Exception as ex:
                    print(('error create Image: %s' % ex.message))
        zip.close()
        uploaded_album.delete()
コード例 #5
0
 def setUp(self):
     self.image_file = open(
         os.path.join(os.path.dirname(__file__), 'test_img.jpg'))
     self.album = Album(name='test album')
     self.album.save()
     details = (
         ('i1', 'alpha            rho'),
         ('i2', '      beta       rho'),
         ('i3', '           gamma rho'),
     )
     self.image_count = len(details)
     for title, tags in details:
         Image(title=title, image=File(self.image_file), tags=tags).save()
     self.image1 = Image.objects.get(title__exact='i1')
コード例 #6
0
ファイル: upload.py プロジェクト: rusbal/imagestore
def process_zipfile(uploaded_album):
    if default_storage.exists(uploaded_album.zip_file.name):
        # TODO: implement try-except here
        zip = zipfile.ZipFile(uploaded_album.zip_file)
        bad_file = zip.testzip()
        if bad_file:
            raise Exception('"%s" in the .zip archive is corrupt.' % bad_file)

        if not uploaded_album.album:
            try:
                uploaded_album.album = Album.objects.get(name=uploaded_album.new_album_name,
                                                         user=uploaded_album.user)
            except Album.DoesNotExist:
                uploaded_album.album = Album.objects.create(name=uploaded_album.new_album_name,
                                                            user=uploaded_album.user)

        from cStringIO import StringIO
        for filename in sorted(zip.namelist()):
            if filename.startswith('__'):  # do not process meta files
                continue
            # print filename.encode('ascii', errors='replace')
            data = zip.read(filename)
            if len(data):
                try:
                    # the following is taken from django.forms.fields.ImageField:
                    # load() could spot a truncated JPEG, but it loads the entire
                    # image in memory, which is a DoS vector. See #3848 and #18520.
                    # verify() must be called immediately after the constructor.
                    PILImage.open(StringIO(data)).verify()
                except Exception, ex:
                    # if a "bad" file is found we just skip it.
                    print('Error verify image: %s' % ex.message)
                    continue
                if hasattr(data, 'seek') and callable(data.seek):
                    print 'seeked'
                    data.seek(0)
                try:
                    img = Image()
                    img.image.save(filename, ContentFile(data))
                    img.user = uploaded_album.album.user
                    img.title = reverse_slug(filename, remove_extension=True, title=True)
                    img.tags = uploaded_album.tags
                    img.save()

                    AlbumImage.objects.create(album=uploaded_album.album,
                                              image=img,
                                              order=0)
                except Exception, ex:
                    print('error create Image: %s' % ex.message)