# -*- coding:utf-8 -*-
"""
作者：sunli
日期：2022年06月07日21:04
"""
# -*- coding:utf-8 -*-
"""
作者：sunli
日期：2022年06月07日20:35
"""
import pandas as pd
import os

RAW_DATA_PATH = 'D:/Fall/train_model/data/test'
Label = {0: 0, 1: 1, 2: 2, 3: 3, 4: 4, 5: 5, 6: 6, 7: 7, 8: 8, 10: 9}


def extract_data(data_file):
    """
    从mobileFall中提取数据，用于做实验测试
    :param data_file:  原始数据文件
    :param sampling_frequency: 原始数据采集频率
    :return:
    """
    data = pd.read_csv(data_file)
    data_size = len(data.label)
    for i in range(data_size):
        data.iat[i, 6] = Label[data.iloc[i, 6]]
    save_path = 'D:/Fall/train_model/data/test' + data_file.replace(RAW_DATA_PATH, '')
    data.to_csv(save_path, index=0)


def find_all_data_and_extract(path):
    """
    递归的查找所有文件并进行转化
    :param path:
    :return:
    """
    if not os.path.exists(path):
        print('路径存在问题：', path)
        return None

    for i in os.listdir(path):
        if os.path.isfile(path + "/" + i):
            if 'csv' in i:
                extract_data(path + "/" + i)
        else:
            find_all_data_and_extract(path + "/" + i)


if __name__ == '__main__':
    find_all_data_and_extract(RAW_DATA_PATH)