2020-10-08 00:11:06 +02:00
|
|
|
import pandas as pd
|
2021-11-19 16:34:36 +01:00
|
|
|
from utils.utils import fetch_provider_features, run_provider_cleaning_script
|
2020-10-08 00:11:06 +02:00
|
|
|
|
|
|
|
sensor_data_files = dict(snakemake.input)
|
|
|
|
|
|
|
|
provider = snakemake.params["provider"]
|
|
|
|
provider_key = snakemake.params["provider_key"]
|
|
|
|
sensor_key = snakemake.params["sensor_key"]
|
|
|
|
|
2021-11-19 16:34:36 +01:00
|
|
|
if sensor_key == "all_cleaning_individual" or sensor_key == "all_cleaning_overall":
|
|
|
|
# Data cleaning
|
|
|
|
sensor_features = run_provider_cleaning_script(provider, provider_key, sensor_key, sensor_data_files)
|
|
|
|
else:
|
|
|
|
# Extract sensor features
|
|
|
|
del sensor_data_files["time_segments_labels"]
|
|
|
|
time_segments_file = snakemake.input["time_segments_labels"]
|
|
|
|
sensor_features = fetch_provider_features(provider, provider_key, sensor_key, sensor_data_files, time_segments_file)
|
2020-10-08 00:11:06 +02:00
|
|
|
|
|
|
|
sensor_features.to_csv(snakemake.output[0], index=False)
|