Init auto insert extra columns

Author: atthaboon

Examples/AutoAddTestDataColumnsExample.robot

@@ -0,0 +1,8 @@
+*** Setting ***
+Library    ExcelDataDriver
+
+
+*** Test Cases ***
+Add basic test result column
+    ${extra columns} =    Create List    [Status2]    [Log Message2]    [Screenshot2]    [Tags2]
+    auto_insert_extra_columns    ./test_data/BasicDemoData.xlsx    username    ${extra columns}

Examples/test_data/BasicNonTestResultColumns.xlsx

@@ -29,5 +29,13 @@ def parse(self, wb):
             test_data_rows = self.parser_strategy.parse_test_data_properties(ws, ws_column_indexs)
             ws_test_data_rows[ws.title] = test_data_rows
         print('Done validate reference data')
-
         return ws_test_data_rows
+
+    def insert_extra_columns(self, wb, columns):
+        for ws in list(wb):
+            ws_column_indexs = self.parser_strategy.parsing_column_indexs(ws)
+            for column in reversed(columns):
+                if column in ws_column_indexs:
+                    continue
+                ws.insert_cols(1)
+                ws['A' + str(self.parser_strategy.start_row)] = column

ExcelDataDriver/ExcelParser/ParserContext.py

@@ -523,6 +523,21 @@ def save_report(self, newfile=None):
         """
         self.excelTestDataService.save_report(newfile)
 
+    ####################################################
+    #
+    # Insert selective columns
+    #
+    ####################################################
+    @keyword
+    def auto_insert_extra_columns(self, filename, main_column_key, columns=[],
+                                  custom_parser_module='ExcelDataDriver.ExcelParser.DefaultReferenceParserStrategy',
+                                  custom_parser_class='DefaultReferenceParserStrategy'):
+        reference_wb = OpenpyxlHelper.load_excel_file(filename)
+        CustomExcelParser = getattr(importlib.import_module(custom_parser_module), custom_parser_class)
+        parser_context = ParserContext(CustomExcelParser(main_column_key))
+        parser_context.insert_extra_columns(reference_wb, columns)
+        reference_wb.save('./test_data/BasicDemoData2.xlsx')
+
     ####################################################
     #
     # Load reference excel data