diff options
Diffstat (limited to 'src/table.py')
| -rw-r--r-- | src/table.py | 11 |
1 files changed, 3 insertions, 8 deletions
diff --git a/src/table.py b/src/table.py index 798fc9f..004bd4d 100644 --- a/src/table.py +++ b/src/table.py @@ -9,8 +9,7 @@ ROW_SIZE = 3 class Table: def __init__(self, file_path: str): self.csv_params = { 'delimiter': ',', 'quotechar': '"' } - self.csv_file = open(file_path, 'r+', newline='') - self.writer = csv.writer(self.csv_file, **self.csv_params) + self.csv_file = open(file_path, 'r', newline='') self.rows = [tuple(row) for row in csv.reader(self.csv_file, **self.csv_params)] def __del__(self): @@ -18,7 +17,7 @@ class Table: self.csv_file.close() def __write_changes(self): - with open(f"new_{token_hex(8)}.csv", 'w', newline='') as new_table: + with open(f"./new_{token_hex(8)}.csv", 'w', newline='') as new_table: writer = csv.writer(new_table, **self.csv_params) for row in self.rows: writer.writerow(row) @@ -37,13 +36,9 @@ class Table: raise IncompatibleRowLengthError(len(new_row)) self.rows.append(new_row) - self.writer.writerow(new_row) def delete_at(self, to_delete: int): - try: - self.rows.pop(to_delete) - except IndexError: - return None + self.rows.pop(to_delete) def search(self, lookup: str): for row in self.rows: |