From a0fcdc83cbc2b440a5836313b2afd1a35698f5ef Mon Sep 17 00:00:00 2001 From: HombreLaser Date: Tue, 12 Mar 2024 23:46:02 -0600 Subject: AƱade pruebas MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/table.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) (limited to 'src/table.py') 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: -- cgit v1.2.3