import yaml

import json

import csv

import configparser

class IoUtils(object):

    """

    dependency:  pip install pyyaml

    yaml_r: read yaml ,return dict

    yaml_w: write yaml,data type is dict

    json_r: read json file ,return dict

    json_w: write json file ,data type is dict

    csv_r: read csv file ,return list format is:

    [

        [header],

        [column1,column2,....],

         .....

     ]

    csv_w: write data into csv file

    data format is :

        [

         [],[],

         ......

        ]

    or like this format:

        [

        (),()

        ......

        ]

    read_config: read ini file :

    return items of section or object of config or only option value

    """

    def yaml_r(self, filepath) -> dict:

        with open(filepath, 'r') as f:

            data = yaml.load(f, Loader=yaml.Loader)

        return data

    def yaml_w(self, filepath, data: dict):

        with open(filepath, 'w', encoding="utf-8") as f:

            yaml.dump(data, f)

    def json_r(self, filepath) -> dict:

        with open(filepath, 'r+')as f:

            return json.load(f)

    def json_w(self, filepath, data: dict):

        with open(filepath, "w+", encoding="utf-8")as f:

            json.dump(data, f, indent=2, ensure_ascii=False)

    def csv_r(self, csv_path) -> list:

        with open(file=csv_path, mode="r")as f:

            data = csv.reader(f, dialect='excel', delimiter=',', quotechar='|')

            data_set = [i for i in data]

        return data_set

    def csv_w(self, csv_path, data):

        with open(file=csv_path, mode='w', newline='')as f:

            wt = csv.writer(f)

            wt.writerows(data)

    def readConfig(self, filepath, section=None, option=None,

                   section_only=False, option_only=False,):

        config = configparser.ConfigParser()

        config.read(filepath)

        if section and section_only:

            return config.items(section)

        if option and option_only:

            return config.get(section=section,option=option)

        if not section_only and not option_only:

            return config

    def opens(self, filepath, mode, data=None):

        """r or w file """

        if mode == "r" or mode == "r+":

            with open(filepath, mode)as file:

                lines = file.readlines()

            return lines

        if mode == "w" or mode == "r+" and not data:

            with open(filepath, mode)as file:

                file.write(data)