Add alternate methods of making a table (csv, table code)

Author: PokestarFan

markdowntable/__init__.py

@@ -96,6 +96,21 @@ def all_columns(self, *args):
         except Exception as e:
             raise SuppressedError(type(e).__name__, str(e), self.debug(print_d=False))
 
+    def all_columns_with_list(self, list):
+        """
+
+        :param list: list
+        :return: None
+        """
+        try:
+            all_col_data = {'function': 'all_columns', 'data': []}
+            for value in list:
+                all_col_data['data'].append(self.add_column(str(value), all_cols=True))
+            self.functions.append(all_col_data)
+        except Exception as e:
+            raise SuppressedError(type(e).__name__, str(e), self.debug(print_d=False))
+
+
     def finalize_cols(self):
         """
 
@@ -110,7 +125,7 @@ def finalize_cols(self):
         except Exception as e:
             raise SuppressedError(type(e).__name__, str(e), self.debug(print_d=False))
 
-    def add_row(self, show_warning_message=True, *args):
+    def add_row(self, *args, show_warning_message=True):
         """
 
         :param show_warning_message: bool
@@ -152,6 +167,48 @@ def add_row(self, show_warning_message=True, *args):
         except Exception as e:
             raise SuppressedError(type(e).__name__, str(e), self.debug(print_d=False))
 
+    def add_row_with_list(self, list, show_warning_message=True):
+        """
+
+        :param show_warning_message: bool
+        :param list: list
+        :return: None
+        """
+        try:
+            if self.finalized_run:
+                self.finalized_run = False
+            if not self.finalized:
+                self.finalize_cols()
+                self.finalized_run = True
+                self.finalized = True
+            add_row_data = {'function': 'add_row',
+                            'data': {'finalized_run': self.finalized_run,
+                                     'show_warning_message': show_warning_message,
+                                     'values': []}}
+            self.rows += 1
+            row = '|'
+            rows_made = 0
+            for i in range(int(len(list))):
+                row += '{}|'.format(str(list[i]))
+                rows_made += 1
+                add_row_data['data']['values'].append(list[i])
+            if self.columns > rows_made:
+                if show_warning_message:
+                    print(not_enough_values(rows_made, self.columns))
+                    add_row_data['data']['message_shown'] = True
+                for i in range(int(self.columns - rows_made)):
+                    row += ' |'
+                # noinspection PyTypeChecker
+                add_row_data['data']['values'].append('{} blank values added'.format(str(self.columns - rows_made)))
+            elif self.columns < rows_made:
+                raise TooManyValues(self.columns)
+            self.table += '\n{}'.format(row)
+            self.functions.append(add_row_data)
+        except TooManyValues:
+            raise
+        except Exception as e:
+            raise SuppressedError(type(e).__name__, str(e), self.debug(print_d=False))
+
     def export_table_to_file(self, filename='markdowntable', extension='txt', mode='w+', overwrite_warning=True):
         """
 

markdowntable/inputs.py

@@ -0,0 +1,55 @@
+import csv
+
+from .__init__ import Table
+
+
+def import_from_csv(filename, extension='csv'):
+    """
+
+    :param filename: string
+    :param extension: string
+    :return: markdown.Table object
+    """
+    global header_row, csv_table
+    with open(filename + '.' + extension, 'r', newline='') as file:
+        filereader = csv.reader(file)
+        row_num = 0
+        for row in filereader:
+            if row_num == 0:
+                header_row = True
+            if header_row:
+                first_entry = row[0]
+                row.remove(first_entry)
+                csv_table = Table(first_entry)
+                for entry in row:
+                    csv_table.add_column(entry)
+            else:
+                csv_table.add_row_with_list(row)
+            row_num += 0
+    return csv_table
+
+
+def import_from_code(code):
+    """
+
+    :param code: string
+    :return: markdown.Table object
+    """
+    global codetable
+    lines = code.split('\n')
+    altlines = []
+    for line in lines:
+        if '|---|' not in line:
+            altlines.append(line)
+    lines = altlines
+    line_no = 0
+    for line in lines:
+        entries = line.split('|')
+        if line_no == 0:
+            codetable = Table(entries[0])
+            entries.remove(entries[0])
+            codetable.all_columns_with_list(entries)
+        else:
+            codetable.add_row_with_list(entries)
+        line_no += 1
+    return codetable