dev: добавлена возможность обрабатывать трейсы вне зависимости от регистра символов в названии колонок csv файла

This commit is contained in:
Andrew 2025-02-04 16:37:57 +03:00
parent b15b079221
commit 8d0151cb97
2 changed files with 7482 additions and 1 deletions

View File

@ -24,10 +24,34 @@ class DataConverter(BaseDataConverter):
# TODO: Осмысленное сообщение в лог. Без traceback и прочего подобного.
return None
@staticmethod
def _fix_headers(dataframe: pd.DataFrame) -> pd.DataFrame:
correct_columns = [
"time", "Closing", "Electrode Force, N FE", "Electrode Force, N ME",
"Force Control FE", "Force Control ME", "Hold Position ME", "Oncomming",
"Position Control FE", "Position Control ME", "Relief",
"Rotor Position, mm FE", "Rotor Position, mm ME",
"Rotor Speed, mm/s FE", "Rotor Speed, mm/s ME",
"Squeeze", "Welding", "Welding Current ME", "Welding Voltage ME"
]
try:
correct_mapping = {name.lower(): name for name in correct_columns}
new_columns = []
for col in dataframe.columns:
fixed_col = correct_mapping.get(col.lower(), col)
new_columns.append(fixed_col)
dataframe.columns = new_columns
return dataframe
except:
# TODO: Обработка исключений!
# TODO: Осмысленное сообщение в лог. Без traceback и прочего подобного.
return None
def convert_data(self, files: list[str]) -> None:
try:
dataframes = [pd.read_csv(file) if file != '' else None for file in files]
converted_dataframes = list(map(self._replace_bool, dataframes))
renamed_dataframes = list(map(self._fix_headers, dataframes))
converted_dataframes = list(map(self._replace_bool, renamed_dataframes))
except:
# TODO: Обработка исключений!
# Get the traceback object

File diff suppressed because it is too large Load Diff