from Lib.Json import json_decode
from Lib.Yaml import yaml_decode
import csv
import os


class LoadData:
    """
    Load the data file from the data directory.
    """
    def __init__(self):
        home = '%s%s%s' % (os.path.dirname(os.path.dirname(__file__)), os.sep, 'Data')
        try:
            os.path.exists(home) or os.makedirs(home)
        except FileExistsError:
            pass
        self.home = home

    def fromJson(self, filename: str, from_data_home=True):
        return json_decode(open('%s/%s' % (self.home, filename) if from_data_home else filename, encoding='utf-8').read())

    def fromYaml(self, filename: str, from_data_home=True):
        return yaml_decode(open('%s/%s' % (self.home, filename) if from_data_home else filename, encoding='utf-8').read())

    def fromCsv(self, filename: str, from_data_home=True):
        """
        Load data from CSV file, compatible standard format.
        :param filename: Filename or path.
        :param from_data_home: Whether to load from the data directory, otherwise use the absolute path.
        :return: List.
        """
        file = '%s/%s' % (self.home, filename) if from_data_home else filename
        try:
            open(file).read(4096)
            encoding = None
        except UnicodeDecodeError:
            encoding = 'utf-8'
        return [row for row in csv.reader(open(file, encoding=encoding))]

    def fromTxt(self, filename: str, from_data_home=True):
        """
        Load the data from the txt text file, and one line is used as a data.
        :param filename: Filename or path.
        :param from_data_home: Whether to load from the data directory, otherwise use the absolute path.
        :return: List.
        """
        file = '%s/%s' % (self.home, filename) if from_data_home else filename
        try:
            open(file).read(4096)
            encoding = None
        except UnicodeDecodeError:
            encoding = 'utf-8'
        return [row.rstrip("\n") for row in open(file, encoding=encoding).readlines()]