aswi2020sebela-gitlab/python-module/Utilities/FolderProcessor.py @ b0d2180b

import os
import zipfile
from Utilities import ConfigureLoader


def list_of_all_files(path):
    files_in_dir = os.listdir(path)

    ignore_set = load_ignore_set(path)

    return set(files_in_dir).difference(ignore_set)


def load_ignore_set(path):
    ignore_set = set()

    with open(path + "ignore.txt", "r") as file:

        for line in file:
            ignore_set.add(line[:-1])

    return ignore_set


def update_ignore_set(path,file_name):

    with open(path + "ignore.txt", "a") as file:
        file.write(file_name + '\n')


def unzip_all_csv_zip_files_in_folder(folder):

    files_in_dir = os.listdir(folder)
    zips = []

    for file in files_in_dir:
        if file.endswith(".zip"):
            zips.append(folder + file)

    for zip_file in zips:

        with zipfile.ZipFile(zip_file, "r") as unziped_file:
            unziped_file.extractall(folder)

        os.remove(zip_file)