pycoze 0.1.255__tar.gz → 0.1.256__tar.gz

{pycoze-0.1.255 → pycoze-0.1.256}/PKG-INFO

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

pycoze-0.1.256/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-0.1.256/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-0.1.256/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-0.1.256/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-0.1.256/pycoze/utils/__init__.py

@@ -0,0 +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, 
+           TcpSocket, socket, socket_subscribe,
+           to_text]

{pycoze-0.1.255 → pycoze-0.1.256}/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.255 → pycoze-0.1.256}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.255 → pycoze-0.1.256}/pycoze.egg-info/SOURCES.txt

@@ -12,6 +12,10 @@ pycoze/ai/llm/__init__.py
 pycoze/ai/llm/chat.py
 pycoze/ai/llm/text_to_image_prompt.py
 pycoze/ai/llm/think.py
+pycoze/api/__init__.py
+pycoze/api/api.py
+pycoze/api/tab_cls.py
+pycoze/api/window_cls.py
 pycoze/bot/__init__.py
 pycoze/bot/agent_chat.py
 pycoze/bot/bot.py

{pycoze-0.1.255 → pycoze-0.1.256}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="pycoze",
-    version="0.1.255",
+    version="0.1.256",
     packages=find_packages(),
     install_requires=[],
     author="Yuan Jie Xiong",

pycoze-0.1.255/pycoze/utils/__init__.py

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