From a0fcdc83cbc2b440a5836313b2afd1a35698f5ef Mon Sep 17 00:00:00 2001
From: HombreLaser <buran@silosneeded.com>
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/arg_parser.py | 18 +++++++++++++++---
 src/program.py    | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 src/table.py      | 11 +++--------
 3 files changed, 70 insertions(+), 11 deletions(-)
 create mode 100644 src/program.py

(limited to 'src')

diff --git a/src/arg_parser.py b/src/arg_parser.py
index c2db3bd..a5fdb40 100644
--- a/src/arg_parser.py
+++ b/src/arg_parser.py
@@ -1,4 +1,5 @@
 import argparse
+import sys
 
 
 AVAILABLE_COMMANDS_HELP_STRING = """Available commands:
@@ -6,13 +7,24 @@ AVAILABLE_COMMANDS_HELP_STRING = """Available commands:
                                     get-rows n: print first n rows
                                     insert "csv string": insert row at the end, the row must have exactly 3 elements.
                                     search "lookup": search "lookup" in the table.
+                                    delete-at n: delete row at nth position
                                  """
 
 
 class ArgParser:
-    def __init__(self):
+    def __init__(self, args=sys.argv):
         self.parser = argparse.ArgumentParser()
         self.parser.add_argument('-f', required=True, help='CSV file location')
         self.parser.add_argument('-c', required=True, help=AVAILABLE_COMMANDS_HELP_STRING, nargs=2)
-        self.args = self.parser.parse_args()
-        print(self.args)
\ No newline at end of file
+        self.args = self.parser.parse_args(args)
+
+    @property
+    def file(self):
+        return self.args.f
+
+    @property
+    def command(self):
+        return self.args.c
+
+    def print_help(self):
+        self.parser.print_help()
\ No newline at end of file
diff --git a/src/program.py b/src/program.py
new file mode 100644
index 0000000..1c43ea7
--- /dev/null
+++ b/src/program.py
@@ -0,0 +1,52 @@
+from src.arg_parser import ArgParser
+from src.table import Table
+from src.exceptions import IncompatibleRowLengthError
+import sys
+
+class Program():
+    def __init__(self, args=sys.argv[1:]):
+        self.arg_parser = ArgParser(args=args)
+        self.table = Table(self.arg_parser.file)
+
+    def _command_arg_to_int(self):
+        try:
+            return int(self.arg_parser.command[1])
+        except ValueError:
+            self.arg_parser.print_help()
+            exit()
+
+    def _print_table_result(self, table):
+        for row in table:
+            print(row)
+
+    def run(self):
+        match self.arg_parser.command[0]:
+            case 'row-at':
+                print(self.table.row_at(self._command_arg_to_int()))
+            case 'get-rows':
+                result = self.table.get_rows(self._command_arg_to_int())
+
+                self._print_table_result(result)
+            case 'insert':
+                new_row = tuple(self.arg_parser.command[1].rsplit(','))
+
+                try:
+                    self.table.insert(new_row)
+                    print('Row added!')
+                except IncompatibleRowLengthError as error:
+                    print(error)
+            case 'search':
+                result = self.table.search(self.arg_parser.command[1])
+
+                if result is not None:
+                    print(result)
+                else:
+                    print('Not found!')
+            case 'delete-at':
+                try: 
+                    self.table.delete_at(self._command_arg_to_int())
+                    print(f"Row at {self._command_arg_to_int()} deleted")
+                except IndexError:
+                    print(f"Couldn't find row at {self._command_arg_to_int()}")
+            case _:
+                arg_parser.print_help()
\ No newline at end of file
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