Source code for dr_data.utilities.file

import logging
import json
import string
import csv
import os
import time
from dr_data.static_strings import *

__author__ = AUTHOR
__copyright__ = COPYRIGHT
__license__ = LICENSE

_logger = logging.getLogger(__name__)


class FileUtility:
    @staticmethod
    def read_file(file_path):
        file = open(file_path)
        contents = file.read()
        file.close()
        return contents

    @staticmethod
    def append_to_file(json_data, filename):
        with open(filename, 'w') as json_file:
            json.dump(json_data, json_file, indent=4, separators=(',', ': '))

    @staticmethod
    def get_filename(name):
        file_date = time.strftime("%Y%m%d")
        return "{database}_{date}".format(database=name, date=file_date)

    @staticmethod
    def generate_json_file(name, path, data):
        file_name = FileUtility.get_filename(name)
        json_schema = json.dumps(data, indent=4)
        with open('{path}/{file_name}.json'.format(path=path, file_name=file_name),
                  'w') as outfile:
            outfile.write(json_schema)
        return file_name

    @staticmethod
    def is_csv_file(selected_file):
        try:
            with open(selected_file, newline='') as csv_file:
                start = csv_file.read(4096)
                if not all([c in string.printable or c.isprintable() for c in start]):
                    return False
                dialect = csv.Sniffer().sniff(start)
                return True
        except csv.Error:
            return False

    @staticmethod
    def get_directory_files(directory):
        directory_files = dict()
        for root, dirs, files in os.walk(os.path.abspath(directory)):
            for file in files:
                key = os.path.splitext(file)[0]
                path = os.path.join(root, file)
                directory_files[key] = path
        return directory_files