Merge pull request #10 from RedTanny/gconfig

read lang filter from sheets

Author: zvigrinberg

src/analysis.py

@@ -399,7 +399,29 @@ def _dataframe_to_test_entries(self, df: pd.DataFrame, language: str) -> List[Di
             )
 
         return tests
-    
+    def get_google_test_configuration(self) -> Dict:
+        """
+        Get Google test configuration.
+        
+        Returns:
+            Dictionary containing Google test configuration
+        """
+        dict_config = {}
+        sheet_name = "AutoTesting_Config"
+        headers, data_rows = self._read_sheet_data(
+                    self.gsheets_config.input_sheet_id,
+                    sheet_name)
+        
+        if not headers or not data_rows:
+            logger.info(f"No configuration found in sheet '{sheet_name}', skipping")
+            return dict_config
+        
+        for key,value in data_rows:
+            dict_config[key] = value
+        logger.info(f"Google test configuration: {dict_config}")    
+
+        return dict_config
+
     def generate_scan_json_from_sheets(self) -> None:
         """
         Generate scan.json file from Google Sheets data.

src/vulnerability_main_automation.py

@@ -194,6 +194,10 @@ def main():
             try:
                 data_loader = GoogleSheetsDataLoader(config)
                 data_loader.generate_scan_json_from_sheets()
+                config_override = data_loader.get_google_test_configuration()
+                if "language" in config_override:
+                    logger.info("Override Language filter: %s with value %s from Google Sheets", config.language, config_override["language"])
+                    config.language = config_override["language"]
                 logger.info("Successfully generated scan.json from Google Sheets")
             except Exception as e:
                 logger.error(f"Failed to generate scan.json from Google Sheets: {e}", exc_info=True)