import pandas as pd
import glob
import os

import abstract_importer as ai
import sample_path as sp


class CSVImporter(ai.AbstractImporter):

    def __init__(self, file_path):
        self._df_samples_list = None
        super(CSVImporter, self).__init__(file_path)

    def import_data(self):
        self.read_csv_file()
        self.import_variables()
        self.import_structure()
        self.compute_row_delta_in_all_samples_frames(self._df_samples_list)

    def read_csv_file(self):
        df = pd.read_csv(self._file_path)
        df.drop(df.columns[[0]], axis=1, inplace=True)
        self._df_samples_list = [df]

    def import_variables(self):
        variables_list = list(self._df_samples_list[0].columns)[1:]
        #wrong_vars_labels = ['Y','Z','X']
        self._sorter = variables_list
        values_list = [3 for var in variables_list]
        # initialize list of lists
        data = {'Name':variables_list, 'Value':values_list}

        # Create the pandas DataFrame
        self._df_variables = pd.DataFrame(data)

    def import_structure(self):
        data = {'From':['X','Y','Z'], 'To':['Z','Z','Y']}
        self._df_structure = pd.DataFrame(data)


read_files = glob.glob(os.path.join('../data', "*.csv"))
print(read_files[0])
csvimp = CSVImporter(read_files[0])

#csvimp.import_data()


s1 = sp.SamplePath(csvimp)
s1.build_trajectories()
s1.build_structure()
print(s1.structure)
print(s1.trajectories)