pycoze 0.1.254__py3-none-any.whl → 0.1.256__py3-none-any.whl

pycoze/api/__init__.py

@@ -0,0 +1,6 @@
+from .api import Api, api
+
+tab = api.tab
+window = api.window
+
+# from ps_view import ViewCls, WebsiteViewCls, FileViewCls, DirectoryViewCls, WorkflowCls

pycoze/api/api.py

@@ -0,0 +1,9 @@
+from .tab_cls import TabCls
+from .window_cls import WindowCls
+
+class Api:
+    def __init__(self) -> None:
+        self.tab = TabCls()
+        self.window = WindowCls()
+
+oper = Api()

pycoze/api/tab_cls.py

@@ -0,0 +1,103 @@
+import os
+from pycoze import utils
+from uuid import uuid4
+from ps_view import ViewCls, WebsiteViewCls, FileViewCls, DirectoryViewCls, WorkflowCls
+import time
+from typing import Union
+
+
+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/window_cls.py

@@ -0,0 +1,50 @@
+import sys
+import subprocess
+import os
+from ps_socket import socket
+import time
+
+
+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='info'):
+        if not isinstance(message, str):
+            message = repr(message)
+        socket.post("append-msg", {"message": message, "type": type})
+
+    def show_operation(self, text: str):
+        socket.post("show-operation", {"text": text})
+
+    def screenshot(self, name):
+        socket.post("screenshot", {"name": name})
+
+    def confirm(self, message: str) -> bool:
+        return socket.post_and_recv_result("confirm", {"message": message})
+
+    def maximize(self):
+        socket.post("maximize", {})
+
+    def open_file_with_system(self, file_path, wait: bool):
+        process_fn = subprocess.run if wait else subprocess.Popen
+        if sys.platform.startswith('linux'):
+            process_fn(['xdg-open', file_path])
+        elif sys.platform.startswith('darwin'):
+            process_fn(['open', file_path])
+        elif sys.platform.startswith('win32') or sys.platform.startswith('cygwin'):
+            os.startfile(file_path)
+        else:
+            raise OSError('Unsupported operating system')
+
+    def open_program(self, file_path, wait: bool):
+        process_fn = subprocess.run if wait else subprocess.Popen
+        process_fn([file_path])
+
+    def execute_javaScript(self, js_code: str):
+        result = socket.post_and_recv_result("executeJavaScript", {"code": js_code})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"] if "value" in result else None

pycoze/reference/bot.py

@@ -10,7 +10,7 @@ from pycoze import utils
 
 bot_index  = 0
 
-params = utils.read_params_file()
+params = utils.params
 
 def ref_bot(bot_id, as_agent_tool=False, workspace_path=None):
     global bot_index

pycoze/reference/tool.py

@@ -8,7 +8,7 @@ from .lib import ChangeDirectoryAndPath, ModuleManager, wrapped_func
 from pycoze import utils
 
 
-params = utils.read_params_file()
+params = utils.params
 
 def ref_tools(tool_id, as_agent_tool=False, workspace_path=None):
     if workspace_path is None:

pycoze/reference/workflow.py

@@ -4,11 +4,11 @@ import importlib
 from langchain.agents import tool as to_agent_tool
 import types
 import langchain_core
-from .lib import ChangeDirectoryAndPath, ModuleManager, wrapped_func
+from .lib import ModuleManager, wrapped_func
 from pycoze import utils
 
 
-params = utils.read_params_file()
+params = utils.params
 
 def _ref_workflows(workflow_id, as_agent_tool=False, workspace_path=None):
     if workspace_path is None:

pycoze/ui/base.py

@@ -4,7 +4,7 @@ import json
 import inspect
 
 
-params = utils.read_params_file()
+params = utils.params
 
 
 def get_ui():

pycoze/utils/__init__.py

@@ -1,5 +1,9 @@
 from .arg import read_arg
 from .env import read_params_file, params, read_json_file
+from .socket import TcpSocket, socket, socket_subscribe
 from .text_or_file import to_text
 
-__all__ = [read_arg, read_params_file, params, read_json_file, to_text]
+__all__ = [read_arg, 
+           read_params_file, params, read_json_file, 
+           TcpSocket, socket, socket_subscribe,
+           to_text]

pycoze/utils/socket.py

@@ -1,10 +1,11 @@
 import socket
 import json
 import uuid
+from pycoze import utils
 
 
-tcp_port = read_arg("tcp_port")
-
+params = utils.params
+tcp_port = params["tcpPort"]
 
 
 class TcpSocket:

{pycoze-0.1.254.dist-info → pycoze-0.1.256.dist-info}/METADATA

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

{pycoze-0.1.254.dist-info → pycoze-0.1.256.dist-info}/RECORD

@@ -5,6 +5,10 @@ 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=c4COCanllUOYJ6-P3cu8jjuiirxODmTwJQvDkkRwfC8,151
+pycoze/api/api.py,sha256=6c75iq44ryedNTGfKrOKxfA1-6eF7l77PjuNg1J6_1k,191
+pycoze/api/tab_cls.py,sha256=0HYzi0At7j2xjMPPDXw9lH5psYKxaPrDCO11cM6razw,4205
+pycoze/api/window_cls.py,sha256=OTVZH1izCasx8a7NPFaS6w-21gZX5jE36aO7Y_MX_mQ,1766
 pycoze/bot/__init__.py,sha256=JxnRoCCqx_LFyVb3pLu0qYCsH8ZLuAaMXAtvVUuCHuE,78
 pycoze/bot/agent_chat.py,sha256=ARXXsIVCTpmBojz2C4BR69nB0QhM6gkEngL3iq34JPo,4178
 pycoze/bot/bot.py,sha256=_qmUTZ09FmRLifHrW5stDZWGVK6yuMEMBC3fmeYJnqk,844
@@ -16,22 +20,22 @@ pycoze/bot/agent/agent_types/__init__.py,sha256=zmU2Kmrv5mCdfg-QlPn2H6pWxbGeq8s7
 pycoze/bot/agent/agent_types/const.py,sha256=BfUKPrhAHREoMLHuFNG2bCIEkC1-f7K0LEqNg4RwiRE,70
 pycoze/bot/agent/agent_types/openai_func_call_agent.py,sha256=6aKnUQDINyUaCW24oa9Qjkm5w3ctZ6lxAgcE4m9YHwE,6701
 pycoze/reference/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pycoze/reference/bot.py,sha256=BDflTV3zYoZqWnJpD5lMM_1vU_5b20M3XiFt1p-RHWM,2427
+pycoze/reference/bot.py,sha256=EQUngYFn5K7DdQWsPHKOXOsrMAz-3nMjiZEPc3yHfvo,2415
 pycoze/reference/lib.py,sha256=0xQJTLTHedGzQBsjuTFNBVqYc4-8Yl65gGCrAhWyOX8,2155
-pycoze/reference/tool.py,sha256=9OxHawxjkh6S8eAOlGNrbfhkk5oRlRc7LBpnV6RmhYI,1722
-pycoze/reference/workflow.py,sha256=xIxv4EAeIR663bRfXkLhgBC0OhfMs5RdfvdORI80CoQ,2068
+pycoze/reference/tool.py,sha256=CYUS95M_XGnUcl4uWayjzKOM9jGpylS7vWJ66JfNTjI,1710
+pycoze/reference/workflow.py,sha256=2EcqN9xefI_c2wHjwmMrUUakM50FPoSVH5_qEedqT8k,2032
 pycoze/ui/__init__.py,sha256=uaXet23wUk64TcZjpBX8qOx4aUhwA_ucrmcxy7Q4Qr4,929
-pycoze/ui/base.py,sha256=sbBZGMUtlosWHQJpxMULa1bGByeSlcldtE9QXNyiJmM,1093
+pycoze/ui/base.py,sha256=bz9mHZwIXA8LErEHTIonH347u6LP7rxV2EADMMjNZos,1081
 pycoze/ui/color.py,sha256=cT9Ib8uNzkOKxyW0IwVj46o4LwdB1xgNCj1_Rou9d_4,854
 pycoze/ui/typ.py,sha256=NpT0FrbHvByOszBZMFtroRp7I7pN-38tYz_zPOPejF4,1723
 pycoze/ui/ui_def.py,sha256=lGWZGpzRoegP34D562PvK0EJHrmVZrlHW1JjsIG9A9Q,4521
-pycoze/utils/__init__.py,sha256=6bQ3erQHKVOUk4ylyC-6PCASbAV0pYFQfC_f1KqkbUw,191
+pycoze/utils/__init__.py,sha256=H-2KRUsUG47owL0sbD1KwDOuRm-j_0K4RkeNhzr7ISo,319
 pycoze/utils/arg.py,sha256=jop1tBfe5hYkHW1NSpCeaZBEznkgguBscj_7M2dWfrs,503
 pycoze/utils/env.py,sha256=5pWlXfM1F5ZU9hhv1rHlDEanjEW5wf0nbyez9bNRqqA,559
-pycoze/utils/socket.py,sha256=CpyyC-zgibi_GvTeHP14VlqIfC7QuRE6BNbQH3Hb45c,2370
+pycoze/utils/socket.py,sha256=yyYgx7G40x58sV466OQH0vj3i55xgR3uMDEjbQMbGks,2413
 pycoze/utils/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
-pycoze-0.1.254.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.254.dist-info/METADATA,sha256=hD65Lc7e26JX0Th3OYDX8OFcJPZcQCR90fy9doVnkno,699
-pycoze-0.1.254.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-pycoze-0.1.254.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.254.dist-info/RECORD,,
+pycoze-0.1.256.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.256.dist-info/METADATA,sha256=UFuAyV9kjFN62yPq0kVYFc3rJwRbMPwAx9OYDysFdA4,699
+pycoze-0.1.256.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+pycoze-0.1.256.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.256.dist-info/RECORD,,