Refactor with pyautogui

desktop_env/controllers/python.py

@@ -1,15 +1,31 @@
-import requests
 import json
 
+import requests
+
 
 class PythonController:
-    def __init__(self, http_server: str):
+    def __init__(self, http_server: str, pkgs_prefix: str = "py -c \"import pyautogui; {command}\""):
         self.http_server = http_server
+        self.pkgs_prefix = pkgs_prefix  # fixme: this is a hacky way to execute python commands. fix it and combine it with installation of packages
 
-    def _execute_python_command(self, command: str) -> None:
-        payload = json.dumps({
-            "command": command
-        })
+    def get_screenshot(self):
+        """
+        Gets a screenshot from the server. With the cursor.
+        """
+        response = requests.get(self.http_server + "/screenshot")
+        if response.status_code == 200:
+            return response.content
+        else:
+            print("Failed to get screenshot. Status code:", response.status_code)
+            return None
+
+    def execute_python_command(self, command: str) -> None:
+        """
+        Executes a python command on the server.
+        It can be used to execute the pyautogui commands, or... any other python command. who knows?
+        """
+        command = self.pkgs_prefix.format(command=command)
+        payload = json.dumps({"command": command})
         headers = {
             'Content-Type': 'application/json'
         }
@@ -23,15 +39,3 @@ class PythonController:
             return response.json()
         except requests.exceptions.RequestException as e:
             print("An error occurred while trying to execute the command:", e)
-
-
-# example usage
-if __name__ == '__main__':
-    # replace with your actual server URL of the vm
-    server_url = "http://192.168.7.129:5000"
-    controller = PythonController(server_url)
-
-    # example commands    
-    python_command = "python -c \"import keyboard; keyboard.write('hello world')\""
-    python_command = "python -c \"import mouse; mouse.move(100,100);mouse.right_click()\""
-    controller._execute_python_command(python_command)