xp3-tool 3.0__tar.gz → 4.0__tar.gz

{xp3_tool-3.0 → xp3_tool-4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xp3_tool
-Version: 3.0
+Version: 4.0
 Summary: xp_t3_tool
 Home-page: 
 Author: XuPeng
@@ -16,6 +16,7 @@ Requires-Python: >=3.10.0
 Description-Content-Type: text/markdown
 Requires-Dist: openai
 Requires-Dist: pandas
+Requires-Dist: oss2
 
 
-本模块提供了一个简单的 CallAi 类，用于与 OpenAI 兼容的 API 服务进行交互，支持设置提示词并发送对话请求。同时包含一个随机字符串生成工具函数，可用于生成指定长度的字母数字混合字符串。
+本模块提供了一个简单的 CallAi 类，用于与 OpenAI 兼容的 API 服务进行交互，支持设置提示词并发送对话请求。一个处理Excel文件与阿里云OSS的交互，并提供转换为pandas DataFrame的类。同时包含一个随机字符串生成工具函数，可用于生成指定长度的字母数字混合字符串。

xp3_tool-4.0/README.md

@@ -0,0 +1 @@
+本模块提供了一个简单的 CallAi 类，用于与 OpenAI 兼容的 API 服务进行交互，支持设置提示词并发送对话请求。一个处理Excel文件与阿里云OSS的交互，并提供转换为pandas DataFrame的类。同时包含一个随机字符串生成工具函数，可用于生成指定长度的字母数字混合字符串。

{xp3_tool-3.0 → xp3_tool-4.0}/setup.py

@@ -18,11 +18,11 @@ URL = ''
 EMAIL = 'xupeng23456@126.com'
 AUTHOR = 'XuPeng'
 REQUIRES_PYTHON = '>=3.10.0'
-VERSION = '3.0'
+VERSION = '4.0'
 
 # What packages are required for this module to be executed?
 REQUIRED = [
-    'openai', 'pandas'
+    'openai', 'pandas','oss2'
 ]
 
 # What packages are optional?

{xp3_tool-3.0 → xp3_tool-4.0}/xp3_tool/__init__.py

@@ -2,7 +2,8 @@ from .xp3_tool import uuid, CallAi
 
 __all__ = [
     "uuid",
-    "CallAi"
+    "CallAi",
+    "ExcelOSSHandler"
 ]
 
-__version__ = "3.0"
+__version__ = "4.0"

xp3_tool-4.0/xp3_tool/xp3_tool.py

@@ -0,0 +1,125 @@
+from openai import OpenAI
+import string
+import random
+
+def uuid(length=16):
+    chars = string.ascii_letters + string.digits
+    return ''.join(random.choice(chars) for _ in range(length))
+
+class CallAi:
+    def __init__(self,api_key,base_url,model=None):
+        self.api_key = api_key
+        self.base_url = base_url
+        self.client = OpenAI(
+            api_key = api_key,
+            base_url= base_url,
+        )
+        self.model =  model if model else 'qwen-plus'
+        self._prompt = ""
+        self.inquiry = ""
+
+    @property
+    def prompt(self):
+        return self._prompt
+
+    @prompt.setter
+    def prompt(self,content):
+        self._prompt = content
+
+    def chat(self,text,top_p = 0.9, temperature = 0.7):
+        completion = self.client.chat.completions.create(
+            model= self.model,
+            messages=[
+                {'role': 'system', 'content': f'{self._prompt}'},
+                {'role': 'user', 'content': text}],
+            temperature = temperature,
+            top_p = top_p
+        )
+        reply = completion.choices[0].message.content
+        return reply
+
+class ExcelOSSHandler:
+    """处理Excel文件与阿里云OSS的交互，并提供转换为pandas DataFrame的功能"""
+    
+    def __init__(self, access_key_id: str, access_key_secret: str, endpoint: str, bucket_name: str):
+        """
+        初始化OSS连接
+        
+        :param access_key_id: 阿里云访问密钥ID
+        :param access_key_secret: 阿里云访问密钥Secret
+        :param endpoint: OSS服务的访问域名
+        :param bucket_name: OSS存储桶名称
+        """
+        # 初始化OSS认证
+        self.auth = oss2.Auth(access_key_id, access_key_secret)
+        # 获取存储桶对象
+        self.bucket = oss2.Bucket(self.auth, endpoint, bucket_name)
+        
+    def upload_excel_to_oss(self, local_file_path: str, oss_file_path: str) -> bool:
+        """
+        将本地Excel文件上传到OSS
+        
+        :param local_file_path: 本地Excel文件路径
+        :param oss_file_path: OSS中保存的文件路径
+        :return: 上传成功返回True，否则返回False
+        """
+        try:
+            # 检查本地文件是否存在
+            if not os.path.exists(local_file_path):
+                print(f"错误: 本地文件 {local_file_path} 不存在")
+                return False
+                
+            # 检查文件是否为Excel文件
+            if not (local_file_path.endswith('.xlsx') or local_file_path.endswith('.xls')):
+                print(f"错误: {local_file_path} 不是Excel文件")
+                return False
+                
+            # 上传文件
+            self.bucket.put_object_from_file(oss_file_path, local_file_path)
+            print(f"文件 {local_file_path} 已成功上传至OSS: {oss_file_path}")
+            return True
+            
+        except ClientError as e:
+            print(f"OSS上传错误: {str(e)}")
+            return False
+        except Exception as e:
+            print(f"上传文件时发生错误: {str(e)}")
+            return False
+    
+    def get_excel_from_oss(self, oss_file_path: str) -> Optional[pd.DataFrame]:
+        """
+        从OSS获取Excel文件并转换为pandas DataFrame
+        
+        :param oss_file_path: OSS中的Excel文件路径
+        :return: 转换后的DataFrame，如果出错则返回None
+        """
+        try:
+            # 检查文件是否存在于OSS
+            if not self.bucket.object_exists(oss_file_path):
+                print(f"错误: OSS文件 {oss_file_path} 不存在")
+                return None
+                
+            # 检查文件是否为Excel文件
+            if not (oss_file_path.endswith('.xlsx') or oss_file_path.endswith('.xls')):
+                print(f"错误: {oss_file_path} 不是Excel文件")
+                return None
+                
+            # 从OSS下载文件到内存
+            response = self.bucket.get_object(oss_file_path)
+            excel_content = response.read()
+            
+            # 将内容转换为DataFrame
+            # 使用BytesIO创建内存文件对象
+            with io.BytesIO(excel_content) as excel_file:
+                # 读取Excel文件，这里假设只有一个工作表
+                df = pd.read_excel(excel_file)
+                
+            print(f"已成功从OSS获取文件 {oss_file_path} 并转换为DataFrame")
+            return df
+            
+        except ClientError as e:
+            print(f"OSS下载错误: {str(e)}")
+            return None
+        except Exception as e:
+            print(f"获取并转换文件时发生错误: {str(e)}")
+            return None

{xp3_tool-3.0 → xp3_tool-4.0}/xp3_tool.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xp3_tool
-Version: 3.0
+Version: 4.0
 Summary: xp_t3_tool
 Home-page: 
 Author: XuPeng
@@ -16,6 +16,7 @@ Requires-Python: >=3.10.0
 Description-Content-Type: text/markdown
 Requires-Dist: openai
 Requires-Dist: pandas
+Requires-Dist: oss2
 
 
-本模块提供了一个简单的 CallAi 类，用于与 OpenAI 兼容的 API 服务进行交互，支持设置提示词并发送对话请求。同时包含一个随机字符串生成工具函数，可用于生成指定长度的字母数字混合字符串。
+本模块提供了一个简单的 CallAi 类，用于与 OpenAI 兼容的 API 服务进行交互，支持设置提示词并发送对话请求。一个处理Excel文件与阿里云OSS的交互，并提供转换为pandas DataFrame的类。同时包含一个随机字符串生成工具函数，可用于生成指定长度的字母数字混合字符串。

{xp3_tool-3.0 → xp3_tool-4.0}/xp3_tool.egg-info/requires.txt

@@ -1,2 +1,3 @@
 openai
 pandas
+oss2

xp3_tool-3.0/README.md

@@ -1 +0,0 @@
-本模块提供了一个简单的 CallAi 类，用于与 OpenAI 兼容的 API 服务进行交互，支持设置提示词并发送对话请求。同时包含一个随机字符串生成工具函数，可用于生成指定长度的字母数字混合字符串。

xp3_tool-3.0/xp3_tool/xp3_tool.py

@@ -1,39 +0,0 @@
-from openai import OpenAI
-import string
-import random
-
-def uuid(length=16):
-    chars = string.ascii_letters + string.digits
-    return ''.join(random.choice(chars) for _ in range(length))
-
-class CallAi:
-    def __init__(self,api_key,base_url,model=None):
-        self.api_key = api_key
-        self.base_url = base_url
-        self.client = OpenAI(
-            api_key = api_key,
-            base_url= base_url,
-        )
-        self.model =  model if model else 'qwen-plus'
-        self._prompt = ""
-        self.inquiry = ""
-
-    @property
-    def prompt(self):
-        return self._prompt
-
-    @prompt.setter
-    def prompt(self,content):
-        self._prompt = content
-
-    def chat(self,text,top_p = 0.9, temperature = 0.7):
-        completion = self.client.chat.completions.create(
-            model= self.model,
-            messages=[
-                {'role': 'system', 'content': f'{self._prompt}'},
-                {'role': 'user', 'content': text}],
-            temperature = temperature,
-            top_p = top_p
-        )
-        reply = completion.choices[0].message.content
-        return reply