Source code for sylib.library.plugins.export.datasource_exporters.plugin_zip_exporter

# This file is part of Sympathy for Data.
# Copyright (c) 2013, 2017, Combine Control Systems AB
#
# SYMPATHY FOR DATA COMMERCIAL LICENSE
# You should have received a link to the License with Sympathy for Data.
import os
import zipfile
from sylib.export import datasource as exportdatasource
from sympathy.platform.exceptions import sywarn




class DataExtractZip(exportdatasource.DatasourceArchiveBase):
    """Extractor for ZIP files."""

    EXPORTER_NAME = "ZIP Extractor"
    FILENAME_EXTENSION = None

    @staticmethod
    def hide_filename():
        return True

    def export_data(self, in_datasource, directory, progress=None):
        file = in_datasource.decode_path()
        filenames = self._create_filenames(file)
        if len(filenames):
            zip_infile = zipfile.ZipFile(file)
            zip_infile.extractall(directory)
        else:
            sywarn(file + ' is either empty or not a valid ZIP file.')
        return filenames

    def create_filenames(self, input_list, filename, *args):
        filenames = []
        for path in input_list:
            filenames.extend(self._create_filenames(path.decode_path()))
        return filenames

    def _create_filenames(self, path):
        if os.path.isfile(path) and zipfile.is_zipfile(path):
            return [file.filename for file in zipfile.ZipFile(path).infolist()
                    if not file.is_dir()]
        return []

    def cardinality(self):
        return self.one_to_many