# copyright: Canonical
"Tools for managing results."
import csv
from uuid import uuid4


class Results:
    def __init__(self, *args, **kwargs):
        self.result_list = [["Run num", "test_case", "files", "MB/s", "time", "full size"]]

    def add(self, row: list) -> None:
        """Add a row to the results lists.

        Args:
            row (list): a row of results.
        """
        self.result_list.append(row)

    def save_to_csv(self) -> None:
        """Save results to csv file in the current directory."""
        with open(f"results{uuid4()}.csv", "w") as f:
            cw = csv.writer(f)
            cw.writerows(self.result_list)