isrpa 1.0.4__py3-none-any.whl → 1.0.6__py3-none-any.whl

isrpa/utils.py

@@ -1,11 +1,14 @@
 import os
 
+import redis
 import requests
 import tldextract
 from dotenv import load_dotenv
 
 load_dotenv("/isearch/aiAgent/.env", override=True)
 address = os.getenv("address")
+client = redis.Redis(host=os.getenv("REDIS_HOST"), port=os.getenv("REDIS_PORT"), db=os.getenv("REDIS_DB"),
+                     password=os.getenv("REDIS_PASSWORD"))
 
 
 def getPath(user_name):
@@ -20,13 +23,44 @@ def getPath(user_name):
     :param port:
     :return:
     """
+
+    # address = os.environ.get("address", "192.168.12.249")
+
+    url = f"{address}/client/getPath"
+    headers = {
+        'Content-Type': 'application/json',  # 说明请求体是 JSON 格式
+    }
+    # 请求体的数据
+    data = {
+        'user_name': user_name
+    }
+    response = requests.post(url, headers=headers, json=data)
+    if response.status_code != 200:
+        return "failure"
+    data = response.json()
+    return f"ws://{data.get('ip_address')}:{data.get('port')}"
+
+
+def get_ssh_path(user_name):
+    """
+        环境变量设置
+        windows
+        setx address "192.168.12.249"
+
+        linux
+        export address = "192.168.12.249"
+        source /etc/profile
+    :param port:
+    :return:
+    """
     url = f"{address}/client/getPath"
     headers = {'Content-Type': 'application/json'}
     data = {'user_name': user_name}
     response = requests.post(url, headers=headers, json=data)
     if response.status_code != 200:
         return "failure"
-    port = response.json()
+    data = response.json()
+    port = data.get('port')
 
     def get_ws_url():
         url = f"http://localhost:{port}/json/version"
@@ -49,7 +83,8 @@ def get_active_url(user_name):
     response = requests.post(url, headers=headers, json=data)
     if response.status_code != 200:
         return "failure"
-    port = response.json()
+    data = response.json()
+    port = data.get('port')
 
     def get_json():
         url = f"http://localhost:{port}/json"
@@ -74,6 +109,10 @@ def get_active_page(browser, active_url):
             return page
 
 
+def set_playwright_file_path(key, value):
+    client.set(key, value, ex=30)
+
+
 def upload_file(file_path, dest_file, user_name):
     """
     通知客户端上传文件

{isrpa-1.0.4.dist-info → isrpa-1.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: isrpa
-Version: 1.0.4
+Version: 1.0.6
 Summary: isrpa package
 Home-page: https://github.com/yourusername/mypackage
 Author: ysq

{isrpa-1.0.4.dist-info → isrpa-1.0.6.dist-info}/RECORD

@@ -5,8 +5,8 @@ isrpa/__init__.py,sha256=pG-YPVG0gJIJ6s4xcAz9S_CnUxpUcz33wl9eNUSgxGk,20
 isrpa/credentials.py,sha256=5paT4rxtZIUPqADmi1bLkoSKEYoosgS7rIjrJsgMNIs,3350
 isrpa/llm_chat.py,sha256=jTuK9wcm3X36V9lXuGuTCdNup_zi2NeG2GuYHzO7gm4,1541
 isrpa/message.py,sha256=N0s6hMaC5S-Pi4EhwcSp-ngfBTBQyW6ymU9sI9dUSeE,11329
-isrpa/utils.py,sha256=y24xUNheQP89D0tityC7uDztJwCuItor4eJucHoqpFA,3996
-isrpa-1.0.4.dist-info/METADATA,sha256=Eb2LwO4tVLsGeLQNaTpx8X99p6KuQhLGv9xXyMf5DNc,511
-isrpa-1.0.4.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
-isrpa-1.0.4.dist-info/top_level.txt,sha256=J5HJbtR2WAeKnW1rrpkiuapwnlswv3RgM-riyZD87o0,6
-isrpa-1.0.4.dist-info/RECORD,,
+isrpa/utils.py,sha256=KjZbsSnypJT5OhfBPPogUef7Q831g2cAYFyD5BmWDKs,5079
+isrpa-1.0.6.dist-info/METADATA,sha256=kuCJISHQM1RTf3MIOl-RV7oyq2cDtvj0YHbGlQTqi5o,511
+isrpa-1.0.6.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
+isrpa-1.0.6.dist-info/top_level.txt,sha256=J5HJbtR2WAeKnW1rrpkiuapwnlswv3RgM-riyZD87o0,6
+isrpa-1.0.6.dist-info/RECORD,,