mirror of
https://github.com/mikf/gallery-dl.git
synced 2024-11-23 11:12:40 +01:00
60c8e090da
Remove the trailing path separator introduced in 3284c62 before
adding the archive's filename extension.
[ci skip]
82 lines
2.4 KiB
Python
82 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2018-2019 Mike Fährmann
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License version 2 as
|
|
# published by the Free Software Foundation.
|
|
|
|
"""Store files in ZIP archives"""
|
|
|
|
from .common import PostProcessor
|
|
import zipfile
|
|
import os
|
|
|
|
|
|
class ZipPP(PostProcessor):
|
|
|
|
COMPRESSION_ALGORITHMS = {
|
|
"store": zipfile.ZIP_STORED,
|
|
"zip": zipfile.ZIP_DEFLATED,
|
|
"bzip2": zipfile.ZIP_BZIP2,
|
|
"lzma": zipfile.ZIP_LZMA,
|
|
}
|
|
|
|
def __init__(self, pathfmt, options):
|
|
PostProcessor.__init__(self)
|
|
self.delete = not options.get("keep-files", False)
|
|
ext = "." + options.get("extension", "zip")
|
|
algorithm = options.get("compression", "store")
|
|
if algorithm not in self.COMPRESSION_ALGORITHMS:
|
|
self.log.warning(
|
|
"unknown compression algorithm '%s'; falling back to 'store'",
|
|
algorithm)
|
|
algorithm = "store"
|
|
|
|
self.path = pathfmt.realdirectory
|
|
args = (self.path[:-1] + ext, "a",
|
|
self.COMPRESSION_ALGORITHMS[algorithm], True)
|
|
|
|
if options.get("mode") == "safe":
|
|
self.run = self._write_safe
|
|
self.zfile = None
|
|
self.args = args
|
|
else:
|
|
self.run = self._write
|
|
self.zfile = zipfile.ZipFile(*args)
|
|
|
|
def _write(self, pathfmt, zfile=None):
|
|
# 'NameToInfo' is not officially documented, but it's available
|
|
# for all supported Python versions and using it directly is a lot
|
|
# faster than calling getinfo()
|
|
if zfile is None:
|
|
zfile = self.zfile
|
|
if pathfmt.filename not in zfile.NameToInfo:
|
|
zfile.write(pathfmt.temppath, pathfmt.filename)
|
|
pathfmt.delete = self.delete
|
|
|
|
def _write_safe(self, pathfmt):
|
|
with zipfile.ZipFile(*self.args) as zfile:
|
|
self._write(pathfmt, zfile)
|
|
|
|
def finalize(self):
|
|
if self.zfile:
|
|
self.zfile.close()
|
|
|
|
if self.delete:
|
|
try:
|
|
# remove target directory
|
|
os.rmdir(self.path)
|
|
except OSError:
|
|
pass
|
|
|
|
if self.zfile and not self.zfile.NameToInfo:
|
|
try:
|
|
# delete empty zip archive
|
|
os.unlink(self.zfile.filename)
|
|
except OSError:
|
|
pass
|
|
|
|
|
|
__postprocessor__ = ZipPP
|