save.py

import csv
import os
from datetime import datetime


data_path = 'data.csv'
# if not os.path.exists(data_path):


def load():
    state = [4, 4, 4, 4, 4, 4, 0, 4, 4, 4, 4, 4, 4, 0]
    if not os.path.exists(data_path):
        return state, 1, 1, 0
    with open(data_path, 'r') as data:
        reader = list(csv.reader(data))
        i = 0
        for row in reader:
            i += 1
        
            print(row)

        if i < 2:
            return state, 1, 1, 0
        game_num = int(input("enter game number to be loaded: "))
        for j, row in enumerate(reader):
            if j == game_num:
                return [int(x) for x in row[1].split()], int(row[3]), int(row[4]), int(row[5])


def save(state, stealing, level, interrupt_mode):
    with open(data_path, 'r+') as data:
        writer = csv.writer(data)
        reader = list(csv.reader(data))
        ln = len(reader)

        if os.stat(data_path).st_size == 0:
            ln += 1
            writer.writerow(["index", "state", "time",
                            "stealing", "level", "irp"])

        s = ""
        for i in state:
            s = s + str(i) + " "
        s = s.strip()
        writer.writerow(
            [ln, s, datetime.now(), stealing, level, interrupt_mode])

    return True


def clear_data():
    f = open(data_path, "w")
    f.truncate()
    f.close()


if __name__ == "__main__":
    clear_data()