Exemplo n.º 1
0
    def test_manga_archive_delete(self):
        ma = MangaArchive(manga=self.manga)
        ma.file.save('sample-file.zip', File(BytesIO()), save=False)
        ma.save()

        path = ma.file.path
        self.assertTrue(os.path.exists(path))

        ma.delete()
        self.assertFalse(os.path.exists(path))
Exemplo n.º 2
0
    def generate(cls, manga):
        """
        This method is locked for up to 30 seconds (per manga) until it returns.
        This prevents the server from falling over due to stampede effect.
        """

        if not cls.acquire_lock(manga): return

        try:
            manga_archive = MangaArchive.objects.get(manga=manga)
            DeletedFile.objects.create(path=manga_archive.file.path)
        except MangaArchive.DoesNotExist:
            manga_archive = MangaArchive(manga=manga)

        manga_zip_file = BytesIO()
        manga_zip = zipfile.ZipFile(manga_zip_file, 'w')

        # write manga pages into zip file
        for page in MangaPage.objects.filter(manga=manga).order_by('page'):
            if not page.image: continue
            extension = get_image_extension(page.image)
            manga_zip.write(page.image.path, '{:03d}.{}'.format(page.page, extension))

        # write info.txt into zip file
        info_text = manga.info_text
        manga_zip.writestr('info.txt', info_text)
        manga_zip.close()

        manga_archive.name = manga.archive_name
        manga_archive.file = UploadedFile(manga_zip_file, 'archive.zip')
        manga_archive.save()

        manga_zip_file.close()

        cls.release_lock(manga)

        return manga_archive