pycoze 0.1.265__py3-none-any.whl → 0.1.266__py3-none-any.whl

pycoze/api/__init__.py

@@ -1,8 +1,10 @@
 from .lib.window_cls import WindowCls
+from .lib.tab_cls import TabCls
 
 class Api:
     def __init__(self) -> None:
         self.window = WindowCls()
+        self.tab = TabCls()
 
 api = Api()
 

pycoze/api/lib/tab_cls.py

@@ -0,0 +1,103 @@
+import os
+from uuid import uuid4
+from view import ViewCls, WebsiteViewCls, FileViewCls, DirectoryViewCls, WorkflowCls
+import time
+from typing import Union
+from pycoze import utils
+
+
+socket = utils.socket
+
+class TabCls:
+
+    def open_path(self, item_path: str) -> FileViewCls | DirectoryViewCls:
+        if not os.path.isabs(item_path):
+            item_path = os.path.abspath(item_path)
+        if os.path.isdir(item_path):
+            return self.open_dir(item_path)
+        else:
+            return self.open_file(item_path)
+
+    def open_file(self, item_path: str) -> Union[FileViewCls, WebsiteViewCls, WorkflowCls]:
+        if not os.path.isabs(item_path):
+            item_path = os.path.abspath(item_path)
+        location = ["FileOrDir", item_path]
+        socket.post("add-tab", {"location": location, "name": os.path.basename(item_path)})
+        lower_suffix = item_path.lower().split(".")[-1]
+        if lower_suffix == 'workflow':
+            return WorkflowCls(location)
+        if len(location) > 1 and (location[0] == "Website" or location[1].endswith(".website") or location[1].endswith(".html") or location[1].endswith(".ipynb")):
+            return WebsiteViewCls(location)
+        return FileViewCls(location)
+
+    def open_dir(self, item_path: str) -> DirectoryViewCls:
+        if not os.path.isabs(item_path):
+            item_path = os.path.abspath(item_path)
+        location = ["FileOrDir", item_path]
+        socket.post("add-tab", {"location": location, "name": os.path.basename(item_path)})
+        return DirectoryViewCls(location)
+
+    def open_website(self, url: str, uid:str=None) -> WebsiteViewCls:
+        if uid is None:
+            uid = str(uuid4())
+        location = ["Website", url, uid]
+        socket.post("add-tab", {"location": location, "name": url})
+        return WebsiteViewCls(location)
+    
+    def get_active(self) -> ViewCls:
+        result = socket.post_and_recv_result("get-active-tab", {})
+        return self._result_to_view(result)
+
+    def _result_to_view(self, result: list[str]) -> ViewCls:
+        if result[0] == "Website":
+            return WebsiteViewCls(result)
+        elif result[0] == "FileOrDir":
+            if os.path.isdir(result[1]):
+                return DirectoryViewCls(result)
+            else:
+                if result[1].lower().split(".")[-1] == 'workflow':
+                    return WorkflowCls(result)
+                return FileViewCls(result)
+        return ViewCls(result)
+    
+    def wait_for_tab_open(self, location: list[str] | ViewCls):
+        times = 0
+        while not self.is_tab_open(location):
+            time.sleep(0.01)
+            times += 1
+            if times > 1000:
+                raise Exception("Tab open timeout")
+
+    def get_all(self) -> list[ViewCls]:
+        results = socket.post_and_recv_result("get-all-tabs", {})
+        return [self._result_to_view(result) for result in results]
+
+    def close_tab(self, location: list[str] | ViewCls):
+        if isinstance(location, ViewCls):
+            location = location.location
+        self.wait_for_tab_open(location)
+        socket.post("close-tab", {"location": location})
+
+    def switch_tab(self, location: list[str] | ViewCls):
+        if isinstance(location, ViewCls):
+            location = location.location
+        self.wait_for_tab_open(location)
+        socket.post("switchTab", {"location": location})
+
+    def is_tab_open(self, location: list[str] | ViewCls):
+        if isinstance(location, ViewCls):
+            location = location.location
+        result = socket.post_and_recv_result("is-tab-open", {"location": location})
+        return result
+
+    def pin_tab(self, location: list[str] | ViewCls):
+        if isinstance(location, ViewCls):
+            location = location.location
+        self.wait_for_tab_open(location)
+        socket.post("pin-tab", {"location": location})
+
+    def unpin_tab(self, location: list[str] | ViewCls):
+        if isinstance(location, ViewCls):
+            location = location.location
+        self.wait_for_tab_open(location)
+        socket.post("unpin-tab", {"location": location})

pycoze/api/lib/view.py

@@ -0,0 +1,242 @@
+import time
+from typing import Union
+from pycoze import utils
+
+
+socket = utils.socket
+
+class ViewCls:
+
+    def __init__(self, location: list[str], nodeIds: list[str] = []):
+        self.location = location
+        self.nodeIds = nodeIds
+        self.is_in_workflow = len(nodeIds) > 0
+        self.is_tab = len(nodeIds) == 0
+
+    def __getitem__(self, key):
+        return self.location[key]
+    
+    def wait_for_tab_open(self):
+        times = 0
+        while not self.is_tab_open():
+            time.sleep(0.01)
+            times += 1
+            if times > 1000:
+                raise Exception("Tab open timeout")
+
+    def close_tab(self):
+        assert self.is_tab
+        self.wait_for_tab_open()
+        socket.post("close-tab", {"location": self.location})
+
+    def set_activate(self):
+        assert self.is_tab
+        self.wait_for_tab_open()
+        socket.post("switchTab", {"location": self.location})
+
+    def is_tab_open(self):
+        assert self.is_tab
+        result = socket.post_and_recv_result("is-tab-open", {"location": self.location})
+        return result
+    
+    def pin_tab(self):
+        assert self.is_tab
+        self.wait_for_tab_open()
+        socket.post("pin-tab", {"location": self.location})
+
+    def unpin_tab(self):
+        assert self.is_tab
+        self.wait_for_tab_open()
+        socket.post("unpin-tab", {"location": self.location})
+
+
+class WebsiteViewCls(ViewCls):
+    def execute_javaScript(self, js_code: str):
+        result = socket.post_and_recv_result("webview.executeJavaScript", {"location": self.location, "code": js_code})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"] if "value" in result else None
+    
+    def capture_page(self, rect=None):
+        result = socket.post_and_recv_result("webview.capturePage", {"location": self.location, "rect": rect})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+    
+    def get_webview_height(self):
+        result = socket.post_and_recv_result("get-webview-height", {"location": self.location})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+    
+    def set_webview_height(self, height: int):
+        result = socket.post_and_recv_result("set-webview-height", {"location": self.location, "height": height})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+    
+    def set_webview_allow_new_window(self, allow: bool):
+        result = socket.post_and_recv_result("set-webview-allow-new-window", {"location": self.location, "allow": allow})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+
+class FileViewCls(ViewCls):
+
+    def get_file_path(self):
+        return self.location[1]
+
+
+class DirectoryViewCls(FileViewCls):
+
+    def get_dir_path(self):
+        return self.location[1]
+
+
+class WorkflowCls(FileViewCls):
+
+    def layout_arrangement(self):
+        # NetworkX layout arrangement，直接用python的布局即可，虽然没有输入输出port的，但是用节点坐标，再缩放几倍一般情况就可以了
+        pass
+
+    def run(self, wait_for_end=False):
+        result = socket.post_and_recv_result("workflow-run", {"location": self.location, "wait_for_end": wait_for_end})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]  # task_id
+
+    def stop(self):
+        result = socket.post_and_recv_result("workflow-stop", {"location": self.location})
+        if not result["ok"]:
+            raise Exception(result["value"])
+
+    def get_state(self):
+        result = socket.post_and_recv_result("workflow-get-state", {"location": self.location})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+
+    def get_nodes(self):
+        result = socket.post_and_recv_result("workflow-get-graph-nodes", {"location": self.location})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+        
+    def get_edges(self):
+        result = socket.post_and_recv_result("workflow-get-graph-edges", {"location": self.location})
+        return result["value"]
+       
+    def add_text_node(self, text: str, position='center'):
+        result = socket.post_and_recv_result("workflow-add-text-node", {
+            "location": self.location,
+            "text": text,
+            "position": position
+        })
+        if not result["ok"]:
+            raise Exception(result["value"])
+        time.sleep(0.01)
+        return result["value"]  # node
+
+    def add_markdown_node(self, text: str, position='center'):
+        result = socket.post_and_recv_result("workflow-add-markdown-node", {
+            "location": self.location,
+            "text": text,
+            "position": position
+        })
+        if not result["ok"]:
+            raise Exception(result["value"])
+        time.sleep(0.01)
+        return result["value"]  # node
+
+    def add_file_node(self, file_path: str, position='center'):
+        is_step = False
+        if file_path.lower().endswith(".py"):
+            with open(file_path, "r", encoding="utf-8") as f:
+                content = f.read()
+                if "ps_workflow" in content:
+                    is_step = True
+        if is_step:
+            return self.add_step_file_node(file_path, position)
+        result = socket.post_and_recv_result("workflow-add-file-node", {
+            "location": self.location,
+            "file_path": file_path,
+            "position": position
+        })
+        if not result["ok"]:
+            raise Exception(result["value"])
+        time.sleep(0.01)
+        return result["value"]  # node
+
+    def add_step_file_node(self, file_path: str, position='center'):
+        result = socket.post_and_recv_result("workflow-add-step-file-node", {
+            "location": self.location,
+            "file_path": file_path,
+            "position": position
+        })
+        if not result["ok"]:
+            raise Exception(result["value"])
+        time.sleep(0.01)
+        return result["value"]  # node
+
+    def add_edge(self, from_node_id: str, from_port_id: str, to_node_id: str, to_port_id: str):
+        result = socket.post_and_recv_result(
+            "workflow-add-edge", {
+                "location": self.location,
+                "source": {
+                    "cell": from_node_id,
+                    "port": from_port_id
+                },
+                "target": {
+                    "cell": to_node_id,
+                    "port": to_port_id
+                }
+            })
+        if not result["ok"]:
+            raise Exception(result["value"])
+        time.sleep(0.01)
+        return result["value"]  # edge
+
+    def get_node_info(self, node: Union[str, object]):
+        if isinstance(node, object):
+            node_id = node["id"]
+        result = socket.post_and_recv_result("workflow-get-node-info", {"location": self.location, "node_id": node_id})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+
+    def get_node_state(self, node: Union[str, object]):
+        if isinstance(node, object):
+            node_id = node["id"]
+        result = socket.post_and_recv_result("workflow-get-node-state", {"location": self.location, "node_id": node_id})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+
+    def get_node_log(self, node: Union[str, object]):
+        if isinstance(node, object):
+            node_id = node["id"]
+        result = socket.post_and_recv_result("workflow-get-node-log", {"location": self.location, "node_id": node_id})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+
+    def get_node_input_data(self, node: Union[str, object], port:str):
+        if isinstance(node, object):
+            node_id = node["id"]
+        result = socket.post_and_recv_result("workflow-get-node-input-data", {"location": self.location, "node_id": node_id, "port":port})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+    
+    def get_node_output_data(self, node: Union[str, object], port:str):
+        if isinstance(node, object):
+            node_id = node["id"]
+        result = socket.post_and_recv_result("workflow-get-node-output-data", {"location": self.location, "node_id": node_id, "port":port})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+    
+    def parse_node_data(self, data: Union[dict|list]):
+        if isinstance(data, list):
+            return [parse_data(d) for d in data]
+        return parse_data(data)

pycoze/api/lib/window_cls.py

@@ -8,11 +8,7 @@ socket = utils.socket
 
 class WindowCls:
 
-    def get_slected_text(self) -> str:
-        result = socket.post_and_recv_result("get-selected-text", {})
-        return result
-
-    def append_msg(self, message: str, type: str = 'info'):
+    def message(self, message: str, type: str = 'info'):
         assert type in ['info', 'warning', 'success', 'error'], 'type must be info, warning, success or error'
         if not isinstance(message, str):
             message = repr(message)
@@ -24,6 +20,10 @@ class WindowCls:
     def maximize(self):
         socket.post("maximize", {})
 
+    def get_slected_text(self) -> str:
+        result = socket.post_and_recv_result("get-selected-text", {})
+        return result
+
     def open_file_with_system(self, file_path, wait: bool):
         process_fn = subprocess.run if wait else subprocess.Popen
         if sys.platform.startswith('linux'):
@@ -35,9 +35,9 @@ class WindowCls:
         else:
             raise OSError('Unsupported operating system')
 
-    def open_program(self, file_path, wait: bool):
+    def open_program(self, program_path, wait: bool):
         process_fn = subprocess.run if wait else subprocess.Popen
-        process_fn([file_path])
+        process_fn([program_path])
 
     def execute_javaScript(self, js_code: str):
         result = socket.post_and_recv_result("executeJavaScript", {"code": js_code})

{pycoze-0.1.265.dist-info → pycoze-0.1.266.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycoze
-Version: 0.1.265
+Version: 0.1.266
 Summary: Package for pycoze only!
 Author: Yuan Jie Xiong
 Author-email: aiqqqqqqq@qq.com

{pycoze-0.1.265.dist-info → pycoze-0.1.266.dist-info}/RECORD

@@ -5,9 +5,11 @@ pycoze/ai/llm/__init__.py,sha256=5_AnOrzXI2V6ZZsLUWW1v5ATRWmJy53JLN9jfSZQXCg,249
 pycoze/ai/llm/chat.py,sha256=izriC7nCp5qeJRqcUVQBVqTHiH6MJS77ROzGBJufdNI,5133
 pycoze/ai/llm/text_to_image_prompt.py,sha256=0bx2C_YRvjAo7iphHGp1-pmGKsKqwur7dM0t3SiA8kA,3398
 pycoze/ai/llm/think.py,sha256=sUgTBdGzcZtL3r-Wx8M3lDuVUmDVz8g3qC0VU8uiKAI,5143
-pycoze/api/__init__.py,sha256=mflQ7hWK3S56kl11UUytHEOhkPv5ZiYSw8rxOaaqQXc,219
+pycoze/api/__init__.py,sha256=q34K7gjDIyRPwsRaTMwfvZXops_15yYp7WD7tXwfp8A,279
 pycoze/api/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pycoze/api/lib/window_cls.py,sha256=nB6vIRq76hYOPR9uYYBOkXRMRVgvr2_NGkw6Pve4IQ0,1734
+pycoze/api/lib/tab_cls.py,sha256=bMNF5e-fAjKRF3ZwXXyVReqnXuE4GJEZOi3Ni7Xruyk,4202
+pycoze/api/lib/view.py,sha256=WrDcI5Du38BpSv1u4TXZ3idToe7VAtDM3Zrjd7BNfVc,8850
+pycoze/api/lib/window_cls.py,sha256=Yezy-SR_EwVtUjrgXnHRC69WGgXVaXeJHjaxn09Ophg,1737
 pycoze/bot/__init__.py,sha256=JxnRoCCqx_LFyVb3pLu0qYCsH8ZLuAaMXAtvVUuCHuE,78
 pycoze/bot/agent_chat.py,sha256=ARXXsIVCTpmBojz2C4BR69nB0QhM6gkEngL3iq34JPo,4178
 pycoze/bot/bot.py,sha256=_qmUTZ09FmRLifHrW5stDZWGVK6yuMEMBC3fmeYJnqk,844
@@ -33,8 +35,8 @@ pycoze/utils/arg.py,sha256=jop1tBfe5hYkHW1NSpCeaZBEznkgguBscj_7M2dWfrs,503
 pycoze/utils/env.py,sha256=5pWlXfM1F5ZU9hhv1rHlDEanjEW5wf0nbyez9bNRqqA,559
 pycoze/utils/socket.py,sha256=yyYgx7G40x58sV466OQH0vj3i55xgR3uMDEjbQMbGks,2413
 pycoze/utils/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
-pycoze-0.1.265.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.265.dist-info/METADATA,sha256=D9LK3HoqJZGMxbeyhxi45zSTKbTP7nlcCUD1R_sIfjY,699
-pycoze-0.1.265.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-pycoze-0.1.265.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.265.dist-info/RECORD,,
+pycoze-0.1.266.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.266.dist-info/METADATA,sha256=iOKp4KKH2GscWVx3ljJaxwGwawNRaD9GQvFczQ56Cik,699
+pycoze-0.1.266.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+pycoze-0.1.266.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.266.dist-info/RECORD,,