ninemine/Convention-Python
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

EP Interaction

Browse Source
This commit is contained in:
LiuBai095
2025-09-30 10:40:58 +08:00
parent ecaab13948
commit b02eafcb35
2 changed files with 92 additions and 120 deletions
Download Patch File Download Diff File Expand all files Collapse all files

173
Convention/Runtime/Interaction.py
View File

@@ -3,19 +3,18 @@ from .File import ToolFile
from .Web import ToolURL from .Web import ToolURL
import json import json
import urllib.parse import urllib.parse
import urllib.request
import urllib.error
import asyncio
import os import os
import re
from typing import * from typing import *
from pydantic import BaseModel
try: try:
import aiohttp from pydantic import BaseModel, PrivateAttr, Field
except ImportError as e:
ImportingThrow(e, "Interaction", ["pydantic"])
try:
import aiofiles import aiofiles
except ImportError as e: except ImportError as e:
ImportingThrow(e, "Interaction", ["aiohttp", "aiofiles"]) ImportingThrow(e, "Interaction", ["aiofiles"])
class InteractionError(Exception): class InteractionError(Exception):
@@ -41,38 +40,30 @@ class SaveError(InteractionError):
class Interaction(BaseModel): class Interaction(BaseModel):
"""统一的文件交互类，自适应处理本地文件和网络文件""" """统一的文件交互类，自适应处理本地文件和网络文件"""
path: str originPath: str
_is_url: bool = False _is_url: bool = PrivateAttr(False)
_is_local: bool = False _is_local: bool = PrivateAttr(False)
_tool_file: Optional[ToolFile] = None _tool_file: Optional[ToolFile] = PrivateAttr(None)
_tool_url: Optional[ToolURL] = None _tool_url: Optional[ToolURL] = PrivateAttr(None)
def __init__(self, path: Union[str, 'Interaction', ToolFile, ToolURL]): def __init__(self, path):
""" """
从路径字符串创建对象，自动识别本地文件或网络URL 从路径字符串创建对象，自动识别本地文件或网络URL
Args: Args:
path: 路径字符串、Interaction对象、ToolFile对象或ToolURL对象 path: 路径字符串或是可以转换为路径字符串的对象
""" """
if isinstance(path, Interaction): super().__init__(originPath=str(path))
path = path.path
elif isinstance(path, ToolFile):
path = path.GetFullPath()
elif isinstance(path, ToolURL):
path = path.url
path_str = str(path)
super().__init__(path=path_str)
# 自动识别路径类型 # 自动识别路径类型
self._detect_path_type() self._detect_path_type()
def _detect_path_type(self): def _detect_path_type(self):
"""自动检测路径类型""" """自动检测路径类型"""
path = self.path.strip() path = self.originPath.strip()
# 检查是否为HTTP/HTTPS URL # 检查是否为HTTP/HTTPS URL
if path.startswith(('http://', 'https://')): if path.startswith(('http://', 'https://', 'file://')):
self._is_url = True self._is_url = True
self._is_local = False self._is_local = False
self._tool_url = ToolURL(path) self._tool_url = ToolURL(path)
@@ -89,7 +80,7 @@ class Interaction(BaseModel):
self._is_url = True self._is_url = True
self._is_local = False self._is_local = False
self._tool_url = ToolURL(full_url) self._tool_url = ToolURL(full_url)
self.path = full_url self.originPath = full_url
return return
# 检查是否为绝对路径或相对路径 # 检查是否为绝对路径或相对路径
@@ -109,7 +100,7 @@ class Interaction(BaseModel):
def __str__(self) -> str: def __str__(self) -> str:
"""隐式字符串转换""" """隐式字符串转换"""
return self.path return self.originPath
def __bool__(self) -> bool: def __bool__(self) -> bool:
"""隐式布尔转换，检查路径是否有效""" """隐式布尔转换，检查路径是否有效"""
@@ -194,7 +185,7 @@ class Interaction(BaseModel):
def Open(self, path: str) -> 'Interaction': def Open(self, path: str) -> 'Interaction':
"""在当前对象上打开新路径""" """在当前对象上打开新路径"""
new_obj = Interaction(path) new_obj = Interaction(path)
self.path = new_obj.path self.originPath = new_obj.originPath
self._is_url = new_obj._is_url self._is_url = new_obj._is_url
self._is_local = new_obj._is_local self._is_local = new_obj._is_local
self._tool_file = new_obj._tool_file self._tool_file = new_obj._tool_file
@@ -211,11 +202,11 @@ class Interaction(BaseModel):
""" """
if self._is_url: if self._is_url:
if not self._tool_url or not self._tool_url.IsValid: if not self._tool_url or not self._tool_url.IsValid:
raise PathValidationError(f"Invalid URL: {self.path}") raise PathValidationError(f"Invalid URL: {self.originPath}")
return self._tool_url.LoadAsText() return self._tool_url.LoadAsText()
else: else:
if not self._tool_file or not self._tool_file.Exists(): if not self._tool_file or not self._tool_file.Exists():
raise PathValidationError(f"File not found: {self.path}") raise PathValidationError(f"File not found: {self.originPath}")
return self._tool_file.LoadAsText() return self._tool_file.LoadAsText()
def LoadAsBinary(self) -> bytes: def LoadAsBinary(self) -> bytes:
@@ -227,11 +218,11 @@ class Interaction(BaseModel):
""" """
if self._is_url: if self._is_url:
if not self._tool_url or not self._tool_url.IsValid: if not self._tool_url or not self._tool_url.IsValid:
raise PathValidationError(f"Invalid URL: {self.path}") raise PathValidationError(f"Invalid URL: {self.originPath}")
return self._tool_url.LoadAsBinary() return self._tool_url.LoadAsBinary()
else: else:
if not self._tool_file or not self._tool_file.Exists(): if not self._tool_file or not self._tool_file.Exists():
raise PathValidationError(f"File not found: {self.path}") raise PathValidationError(f"File not found: {self.originPath}")
return self._tool_file.LoadAsBinary() return self._tool_file.LoadAsBinary()
def LoadAsJson(self, model_type: Optional[type] = None) -> Any: def LoadAsJson(self, model_type: Optional[type] = None) -> Any:
@@ -246,11 +237,11 @@ class Interaction(BaseModel):
""" """
if self._is_url: if self._is_url:
if not self._tool_url or not self._tool_url.IsValid: if not self._tool_url or not self._tool_url.IsValid:
raise PathValidationError(f"Invalid URL: {self.path}") raise PathValidationError(f"Invalid URL: {self.originPath}")
return self._tool_url.LoadAsJson(model_type) return self._tool_url.LoadAsJson(model_type)
else: else:
if not self._tool_file or not self._tool_file.Exists(): if not self._tool_file or not self._tool_file.Exists():
raise PathValidationError(f"File not found: {self.path}") raise PathValidationError(f"File not found: {self.originPath}")
json_data = self._tool_file.LoadAsJson() json_data = self._tool_file.LoadAsJson()
if model_type and issubclass(model_type, BaseModel): if model_type and issubclass(model_type, BaseModel):
return model_type.model_validate(json_data) return model_type.model_validate(json_data)
@@ -266,11 +257,11 @@ class Interaction(BaseModel):
""" """
if self._is_url: if self._is_url:
if not self._tool_url or not self._tool_url.IsValid: if not self._tool_url or not self._tool_url.IsValid:
raise PathValidationError(f"Invalid URL: {self.path}") raise PathValidationError(f"Invalid URL: {self.originPath}")
return await self._tool_url.LoadAsTextAsync() return await self._tool_url.LoadAsTextAsync()
else: else:
if not self._tool_file or not self._tool_file.Exists(): if not self._tool_file or not self._tool_file.Exists():
raise PathValidationError(f"File not found: {self.path}") raise PathValidationError(f"File not found: {self.originPath}")
# 异步读取本地文件 # 异步读取本地文件
async with aiofiles.open(self._tool_file.GetFullPath(), 'r', encoding='utf-8') as f: async with aiofiles.open(self._tool_file.GetFullPath(), 'r', encoding='utf-8') as f:
return await f.read() return await f.read()
@@ -284,11 +275,11 @@ class Interaction(BaseModel):
""" """
if self._is_url: if self._is_url:
if not self._tool_url or not self._tool_url.IsValid: if not self._tool_url or not self._tool_url.IsValid:
raise PathValidationError(f"Invalid URL: {self.path}") raise PathValidationError(f"Invalid URL: {self.originPath}")
return await self._tool_url.LoadAsBinaryAsync() return await self._tool_url.LoadAsBinaryAsync()
else: else:
if not self._tool_file or not self._tool_file.Exists(): if not self._tool_file or not self._tool_file.Exists():
raise PathValidationError(f"File not found: {self.path}") raise PathValidationError(f"File not found: {self.originPath}")
# 异步读取本地文件 # 异步读取本地文件
async with aiofiles.open(self._tool_file.GetFullPath(), 'rb') as f: async with aiofiles.open(self._tool_file.GetFullPath(), 'rb') as f:
return await f.read() return await f.read()
@@ -305,11 +296,11 @@ class Interaction(BaseModel):
""" """
if self._is_url: if self._is_url:
if not self._tool_url or not self._tool_url.IsValid: if not self._tool_url or not self._tool_url.IsValid:
raise PathValidationError(f"Invalid URL: {self.path}") raise PathValidationError(f"Invalid URL: {self.originPath}")
return await self._tool_url.LoadAsJsonAsync(model_type) return await self._tool_url.LoadAsJsonAsync(model_type)
else: else:
if not self._tool_file or not self._tool_file.Exists(): if not self._tool_file or not self._tool_file.Exists():
raise PathValidationError(f"File not found: {self.path}") raise PathValidationError(f"File not found: {self.originPath}")
# 异步读取本地JSON文件 # 异步读取本地JSON文件
text_content = await self.LoadAsTextAsync() text_content = await self.LoadAsTextAsync()
try: try:
@@ -318,10 +309,10 @@ class Interaction(BaseModel):
return model_type.model_validate(json_data) return model_type.model_validate(json_data)
return json_data return json_data
except json.JSONDecodeError as e: except json.JSONDecodeError as e:
raise LoadError(f"Failed to parse JSON from {self.path}: {str(e)}") raise LoadError(f"Failed to parse JSON from {self.originPath}: {str(e)}")
# 同步保存方法 # 同步保存方法
def SaveAsText(self, content: str, local_path: Optional[str] = None) -> Union[ToolFile, 'Interaction']: def SaveAsText(self, content: str, local_path: Optional[str] = None) -> 'Interaction':
""" """
同步保存为文本 同步保存为文本
@@ -339,15 +330,14 @@ class Interaction(BaseModel):
file_obj = ToolFile(local_path) file_obj = ToolFile(local_path)
file_obj.TryCreateParentPath() file_obj.TryCreateParentPath()
file_obj.SaveAsText(content) file_obj.SaveAsText(content)
return file_obj
else: else:
if not self._tool_file: if not self._tool_file:
raise PathValidationError(f"Invalid file path: {self.path}") raise PathValidationError(f"Invalid file path: {self.originPath}")
self._tool_file.TryCreateParentPath() self._tool_file.TryCreateParentPath()
self._tool_file.SaveAsText(content) self._tool_file.SaveAsText(content)
return self return self
def SaveAsBinary(self, content: bytes, local_path: Optional[str] = None) -> Union[ToolFile, 'Interaction']: def SaveAsBinary(self, content: bytes, local_path: Optional[str] = None) -> 'Interaction':
""" """
同步保存为二进制 同步保存为二进制
@@ -365,15 +355,14 @@ class Interaction(BaseModel):
file_obj = ToolFile(local_path) file_obj = ToolFile(local_path)
file_obj.TryCreateParentPath() file_obj.TryCreateParentPath()
file_obj.SaveAsBinary(content) file_obj.SaveAsBinary(content)
return file_obj
else: else:
if not self._tool_file: if not self._tool_file:
raise PathValidationError(f"Invalid file path: {self.path}") raise PathValidationError(f"Invalid file path: {self.originPath}")
self._tool_file.TryCreateParentPath() self._tool_file.TryCreateParentPath()
self._tool_file.SaveAsBinary(content) self._tool_file.SaveAsBinary(content)
return self return self
def SaveAsJson(self, data: Any, local_path: Optional[str] = None) -> Union[ToolFile, 'Interaction']: def SaveAsJson(self, data: Any, local_path: Optional[str] = None) -> 'Interaction':
""" """
同步保存为JSON 同步保存为JSON
@@ -391,16 +380,15 @@ class Interaction(BaseModel):
file_obj = ToolFile(local_path) file_obj = ToolFile(local_path)
file_obj.TryCreateParentPath() file_obj.TryCreateParentPath()
file_obj.SaveAsJson(data) file_obj.SaveAsJson(data)
return file_obj
else: else:
if not self._tool_file: if not self._tool_file:
raise PathValidationError(f"Invalid file path: {self.path}") raise PathValidationError(f"Invalid file path: {self.originPath}")
self._tool_file.TryCreateParentPath() self._tool_file.TryCreateParentPath()
self._tool_file.SaveAsJson(data) self._tool_file.SaveAsJson(data)
return self return self
# 异步保存方法 # 异步保存方法
async def SaveAsTextAsync(self, content: str, local_path: Optional[str] = None) -> Union[ToolFile, 'Interaction']: async def SaveAsTextAsync(self, content: str, local_path: Optional[str] = None) -> 'Interaction':
""" """
异步保存为文本 异步保存为文本
@@ -419,16 +407,15 @@ class Interaction(BaseModel):
file_obj.TryCreateParentPath() file_obj.TryCreateParentPath()
async with aiofiles.open(file_obj.GetFullPath(), 'w', encoding='utf-8') as f: async with aiofiles.open(file_obj.GetFullPath(), 'w', encoding='utf-8') as f:
await f.write(content) await f.write(content)
return file_obj
else: else:
if not self._tool_file: if not self._tool_file:
raise PathValidationError(f"Invalid file path: {self.path}") raise PathValidationError(f"Invalid file path: {self.originPath}")
self._tool_file.TryCreateParentPath() self._tool_file.TryCreateParentPath()
async with aiofiles.open(self._tool_file.GetFullPath(), 'w', encoding='utf-8') as f: async with aiofiles.open(self._tool_file.GetFullPath(), 'w', encoding='utf-8') as f:
await f.write(content) await f.write(content)
return self return self
async def SaveAsBinaryAsync(self, content: bytes, local_path: Optional[str] = None) -> Union[ToolFile, 'Interaction']: async def SaveAsBinaryAsync(self, content: bytes, local_path: Optional[str] = None) -> 'Interaction':
""" """
异步保存为二进制 异步保存为二进制
@@ -447,16 +434,15 @@ class Interaction(BaseModel):
file_obj.TryCreateParentPath() file_obj.TryCreateParentPath()
async with aiofiles.open(file_obj.GetFullPath(), 'wb') as f: async with aiofiles.open(file_obj.GetFullPath(), 'wb') as f:
await f.write(content) await f.write(content)
return file_obj
else: else:
if not self._tool_file: if not self._tool_file:
raise PathValidationError(f"Invalid file path: {self.path}") raise PathValidationError(f"Invalid file path: {self.originPath}")
self._tool_file.TryCreateParentPath() self._tool_file.TryCreateParentPath()
async with aiofiles.open(self._tool_file.GetFullPath(), 'wb') as f: async with aiofiles.open(self._tool_file.GetFullPath(), 'wb') as f:
await f.write(content) await f.write(content)
return self return self
async def SaveAsJsonAsync(self, data: Any, local_path: Optional[str] = None) -> Union[ToolFile, 'Interaction']: async def SaveAsJsonAsync(self, data: Any, local_path: Optional[str] = None) -> 'Interaction':
""" """
异步保存为JSON 异步保存为JSON
@@ -554,7 +540,7 @@ class Interaction(BaseModel):
return await self._tool_url.PostAsync(callback, form_data) return await self._tool_url.PostAsync(callback, form_data)
# 便利方法 # 便利方法
def Save(self, local_path: Optional[str] = None) -> Union[ToolFile, 'Interaction']: def Save(self, local_path: Optional[str] = None) -> 'Interaction':
""" """
自动选择格式保存 自动选择格式保存
@@ -564,16 +550,15 @@ class Interaction(BaseModel):
Returns: Returns:
保存的文件对象或Interaction对象 保存的文件对象或Interaction对象
""" """
# 对于本地文件，直接返回自身（已存在）
if self._is_url: if self._is_url:
# 对于URL，先下载内容再保存 # 对于URL，先下载内容再保存
if not self._tool_url: if not self._tool_url:
raise PathValidationError(f"Invalid URL: {self.path}") raise PathValidationError(f"Invalid URL: {self.originPath}")
return self._tool_url.Save(local_path) self._tool_url.Save(local_path)
else:
# 对于本地文件，直接返回自身（已存在）
return self return self
async def SaveAsync(self, local_path: Optional[str] = None) -> Union[ToolFile, 'Interaction']: async def SaveAsync(self, local_path: Optional[str] = None) -> 'Interaction':
""" """
异步自动选择格式保存 异步自动选择格式保存
@@ -583,10 +568,11 @@ class Interaction(BaseModel):
Returns: Returns:
保存的文件对象或Interaction对象 保存的文件对象或Interaction对象
""" """
# 对于本地文件，直接返回自身（已存在）
if self._is_url: if self._is_url:
# 对于URL，异步下载内容 # 对于URL，异步下载内容
if not self._tool_url: if not self._tool_url:
raise PathValidationError(f"Invalid URL: {self.path}") raise PathValidationError(f"Invalid URL: {self.originPath}")
if local_path is None: if local_path is None:
local_path = self.GetFilename() or "downloaded_file" local_path = self.GetFilename() or "downloaded_file"
@@ -605,13 +591,14 @@ class Interaction(BaseModel):
content = await self.LoadAsBinaryAsync() content = await self.LoadAsBinaryAsync()
await self.SaveAsBinaryAsync(content, local_path) await self.SaveAsBinaryAsync(content, local_path)
return file_obj
except Exception as e: except Exception as e:
raise SaveError(f"Failed to save {self.path}: {str(e)}") raise SaveError(f"Failed to save {self.originPath}: {str(e)}")
else:
# 对于本地文件，直接返回自身
return self return self
def Downloadable(self) -> bool:
"""检查是否可下载"""
return self._is_url and self._tool_url.IsValid if self._tool_url else False
def Download(self, local_path: Optional[str] = None) -> ToolFile: def Download(self, local_path: Optional[str] = None) -> ToolFile:
""" """
下载文件（仅对URL有效） 下载文件（仅对URL有效）
@@ -622,10 +609,10 @@ class Interaction(BaseModel):
Returns: Returns:
下载的文件对象 下载的文件对象
""" """
if not self._is_url: if self._is_local:
raise InteractionError("Download method is only available for URLs") raise InteractionError("Download method is only available for URLs")
if not self._tool_url: if not self._tool_url:
raise PathValidationError(f"Invalid URL: {self.path}") raise PathValidationError(f"Invalid URL: {self.originPath}")
return self._tool_url.Download(local_path) return self._tool_url.Download(local_path)
async def DownloadAsync(self, local_path: Optional[str] = None) -> ToolFile: async def DownloadAsync(self, local_path: Optional[str] = None) -> ToolFile:
@@ -638,13 +625,13 @@ class Interaction(BaseModel):
Returns: Returns:
下载的文件对象 下载的文件对象
""" """
if not self._is_url: if self._is_local:
raise InteractionError("Download method is only available for URLs") raise InteractionError("DownloadAsync method is only available for URLs")
if not self._tool_url: if not self._tool_url:
raise PathValidationError(f"Invalid URL: {self.path}") raise PathValidationError(f"Invalid URL: {self.originPath}")
return await self._tool_url.DownloadAsync(local_path) return await self._tool_url.DownloadAsync(local_path)
def Copy(self, target_path: Optional[Union[str, ToolFile, 'Interaction']] = None) -> 'Interaction': def Copy(self, target_path) -> ToolFile:
""" """
复制文件（仅对本地文件有效） 复制文件（仅对本地文件有效）
@@ -657,20 +644,10 @@ class Interaction(BaseModel):
if not self._is_local: if not self._is_local:
raise InteractionError("Copy method is only available for local files") raise InteractionError("Copy method is only available for local files")
if not self._tool_file: if not self._tool_file:
raise PathValidationError(f"Invalid file path: {self.path}") raise PathValidationError(f"Invalid file path: {self.originPath}")
return self._tool_file.Copy(str(target_path))
if target_path is None: def Move(self, target_path) -> ToolFile:
copied_file = self._tool_file.Copy()
else:
if isinstance(target_path, Interaction):
target_path = target_path.path
elif isinstance(target_path, ToolFile):
target_path = target_path.GetFullPath()
copied_file = self._tool_file.Copy(str(target_path))
return Interaction(copied_file.GetFullPath())
def Move(self, target_path: Union[str, ToolFile, 'Interaction']) -> 'Interaction':
""" """
移动文件（仅对本地文件有效） 移动文件（仅对本地文件有效）
@@ -683,16 +660,8 @@ class Interaction(BaseModel):
if not self._is_local: if not self._is_local:
raise InteractionError("Move method is only available for local files") raise InteractionError("Move method is only available for local files")
if not self._tool_file: if not self._tool_file:
raise PathValidationError(f"Invalid file path: {self.path}") raise PathValidationError(f"Invalid file path: {self.originPath}")
return self._tool_file.Move(str(target_path))
if isinstance(target_path, Interaction):
target_path = target_path.path
elif isinstance(target_path, ToolFile):
target_path = target_path.GetFullPath()
self._tool_file.Move(str(target_path))
self.path = self._tool_file.GetFullPath()
return self
def Remove(self) -> 'Interaction': def Remove(self) -> 'Interaction':
""" """
@@ -704,7 +673,7 @@ class Interaction(BaseModel):
if not self._is_local: if not self._is_local:
raise InteractionError("Remove method is only available for local files") raise InteractionError("Remove method is only available for local files")
if not self._tool_file: if not self._tool_file:
raise PathValidationError(f"Invalid file path: {self.path}") raise PathValidationError(f"Invalid file path: {self.originPath}")
self._tool_file.Remove() self._tool_file.Remove()
return self return self
@@ -728,7 +697,7 @@ class Interaction(BaseModel):
if not self._is_local: if not self._is_local:
raise InteractionError("GetSize method is only available for local files") raise InteractionError("GetSize method is only available for local files")
if not self._tool_file or not self._tool_file.Exists(): if not self._tool_file or not self._tool_file.Exists():
raise PathValidationError(f"File not found: {self.path}") raise PathValidationError(f"File not found: {self.originPath}")
return self._tool_file.GetSize() return self._tool_file.GetSize()
@@ -757,7 +726,7 @@ class Interaction(BaseModel):
""" """
if self._is_local: if self._is_local:
if not self._tool_file: if not self._tool_file:
raise PathValidationError(f"Invalid file path: {self.path}") raise PathValidationError(f"Invalid file path: {self.originPath}")
parent_dir = self._tool_file.GetParentDir() parent_dir = self._tool_file.GetParentDir()
return Interaction(parent_dir.GetFullPath()) return Interaction(parent_dir.GetFullPath())
else: else:
@@ -767,8 +736,8 @@ class Interaction(BaseModel):
def ToString(self) -> str: def ToString(self) -> str:
"""获取完整路径""" """获取完整路径"""
return self.path return self.originPath
def GetFullPath(self) -> str: def GetFullPath(self) -> str:
"""获取完整路径""" """获取完整路径"""
return self.path return self.originPath

11
[Test]/test.py
View File

@@ -1,9 +1,12 @@
import sys import sys
import os import os
from time import sleep
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from Convention.Runtime.File import * from Convention.Runtime.Interaction import *
download = Interaction("http://www.liubai.site:4000/d/storage/Convention/Convention-Unity-Demo/TEST/song.mp3?sign=pIIWFqeous--i4H5fNIpqQsS0HeMinSRA_bztq6Czgo=:0")
file = download.Download("./[Test]/test.mp3")
sleep(5)
file.Remove()
first = ToolFile("E:/dev/")
second = ToolFile("/analyze/")
print(first|second)