diff --git a/mossutils/cluster.py b/mossutils/cluster.py index 4c49cb58b34d938e7570a77d5ef217d0ae326ce9..d3e0f0aed648fa6a3bc94e9d0674864bfb90061e 100644 --- a/mossutils/cluster.py +++ b/mossutils/cluster.py @@ -29,10 +29,10 @@ class MossResult: lines: int match_file: str - def flat_headers(self): + def fields_flattened(self): return [f.name for f in dataclasses.fields(self)] - def flat_data(self): + def values_flattened(self): return dataclasses.astuple(self) @dataclasses.dataclass @@ -42,15 +42,15 @@ class ClusterEntry: dot_file: str svg_file: str - def flat_headers(self): + def fields_flattened(self): field_names = [f.name for f in dataclasses.fields(self)] - result_headers = self.result.flat_headers() + result_headers = self.result.fields_flattened() return field_names[:1] + result_headers + field_names[2:] - def flat_data(self): + def values_flattened(self): field_data = dataclasses.astuple(self) - result_data = self.result.flat_data() + result_data = self.result.values_flattened() return field_data[:1] + result_data + field_data[2:] @@ -174,10 +174,10 @@ def create_summary_csv_file(cluster_entries, summary_csv_file): with open(summary_csv_file, "w", newline="") as csv_fh: csv_writer = csv.writer(csv_fh) - csv_writer.writerow(cluster_entries[0].flat_headers()) + csv_writer.writerow(cluster_entries[0].fields_flattened()) for entry in cluster_entries: - csv_writer.writerow(entry.flat_data()) + csv_writer.writerow(entry.values_flattened()) def main( results_csv_file=DEFAULT_RESULTS_CSV_FILE,