commands with enum

Author: nathan-vm

dependencies/requirements_0.txt

@@ -10,8 +10,8 @@
             "memory": 128
         },
         "limits": {
-            "cpu": 100,
-            "memory": 128
+            "cpu": 1000,
+            "memory": 1024
         }
     },
     "tags": [

pieces/SeleniumWebBrowserPiece/metadata.json

@@ -1,10 +1,14 @@
 from pydantic import BaseModel, Field
 from typing import List
+from enum import Enum
 
+class Commands(str,Enum):
+    get="get"
+    maximize_window="maximize_window"
+    save_screenshot="save_screenshot"
 class CommandInput(BaseModel):
-    name: str
-    cmd: str
-    stop_on_error: bool
+    name: Commands
+    value: str
 
 class CommandOutput(BaseModel):
     name: str

pieces/SeleniumWebBrowserPiece/models.py

@@ -1,8 +1,8 @@
 from domino.base_piece import BasePiece
-from .models import InputModel, OutputModel, CommandOutput
+from .models import InputModel, OutputModel,CommandInput, CommandOutput
 from selenium import webdriver
 from selenium.webdriver.chrome.options import Options
-
+from pathlib import Path
 class SeleniumWebBrowserPiece(BasePiece):
 
     def piece_function(self, input_data: InputModel):
@@ -15,18 +15,18 @@ def piece_function(self, input_data: InputModel):
         self.logger.info("Start chrome web driver.")
         driver = webdriver.Chrome(options=options)
 
-        self.logger.info("Going to URL.")
-        driver.get("https://webscraper.io/test-sites/e-commerce/static")
-
-        self.logger.info("Maximize window.")
-        driver.maximize_window()
-
-        self.logger.info("Taking screenshot.")
-        driver.save_screenshot('ss.png')
+        results_path = Path(self.results_path)
 
-        self.logger.info("Quit chrome driver.")
+        for cmd in input_data.commands:
+            if cmd.name.value == "save_screenshot":
+                cmd.value = str(results_path/cmd.value)
+                self.display_result = dict(file_type="png",filepath=cmd.value)
+                
+            getattr(driver, cmd.name)(cmd.value)
+        
         driver.quit()
 
         return OutputModel(
             outputs=[dict(name="out1", result="res1")]
         )
+