tamar-file-hub-client 0.1.6__tar.gz → 0.1.7__tar.gz

{tamar_file_hub_client-0.1.6 → tamar_file_hub_client-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tamar-file-hub-client
-Version: 0.1.6
+Version: 0.1.7
 Summary: A Python SDK for gRPC-based file management system
 Home-page: https://github.com/Tamar-Edge-AI/file-hub-client
 Author: Oscar Ou

{tamar_file_hub_client-0.1.6 → tamar_file_hub_client-0.1.7}/file_hub_client/schemas/context.py

@@ -1,171 +1,171 @@
-"""
-用户上下文和请求上下文相关的Schema定义
-"""
-from dataclasses import dataclass, field
-from typing import Optional, Dict, Any
-from datetime import datetime
-
-from file_hub_client.enums import Role
-
-
-@dataclass
-class UserContext:
-    """
-    用户上下文信息
-    
-    包含两大类信息：
-    1. ownership（所有权）: org_id, user_id - 表示资源归属
-    2. operator（操作者）: actor_id, role - 表示实际操作者（可能是用户、agent或系统）
-    3. request info（请求信息）: user_ip - 表示请求来源IP（用于审计和安全）
-    """
-    # Ownership - 资源所有权信息
-    org_id: str  # 组织ID
-    user_id: str  # 用户ID
-
-    # Operator - 操作者信息
-    actor_id: Optional[str] = None  # 实际操作者ID（如果为空，默认使用user_id）
-    role: Role = Role.ACCOUNT  # 操作者角色（ACCOUNT, AGENT, SYSTEM等）
-    
-    # Request info - 请求信息
-    user_ip: Optional[str] = None  # 用户IP地址（请求来源IP）
-
-    def __post_init__(self):
-        """初始化后处理，如果actor_id为空，默认使用user_id"""
-        if self.actor_id is None:
-            self.actor_id = self.user_id
-
-    def to_metadata(self) -> Dict[str, str]:
-        """转换为gRPC metadata格式"""
-        metadata = {
-            'x-org-id': self.org_id,
-            'x-user-id': self.user_id,
-            'x-actor-id': self.actor_id,
-            'x-role': self.role,
-        }
-        
-        # 只有当user_ip不为None时才添加x-user-ip
-        if self.user_ip is not None:
-            metadata['x-user-ip'] = self.user_ip
-            
-        return metadata
-
-    @classmethod
-    def from_metadata(cls, metadata: Dict[str, str]) -> Optional['UserContext']:
-        """从metadata中解析用户上下文"""
-        org_id = metadata.get('x-org-id')
-        user_id = metadata.get('x-user-id')
-
-        if not org_id or not user_id:
-            return None
-
-        return cls(
-            org_id=org_id,
-            user_id=user_id,
-            actor_id=metadata.get('x-actor-id'),
-            role=Role(metadata.get('x-role', Role.ACCOUNT)),
-            user_ip=metadata.get('x-user-ip')
-        )
-
-
-@dataclass
-class RequestContext:
-    """
-    请求上下文信息
-    
-    包含请求相关的元数据，如客户端信息、请求追踪等
-    """
-    request_id: Optional[str] = None  # 请求ID，用于追踪
-    client_ip: Optional[str] = None  # 客户端IP地址
-    client_version: Optional[str] = None  # 客户端版本
-    client_type: Optional[str] = None  # 客户端类型（web, mobile, desktop, cli等）
-    user_agent: Optional[str] = None  # User-Agent信息
-    timestamp: Optional[datetime] = field(default_factory=datetime.now)  # 请求时间戳
-    extra: Dict[str, Any] = field(default_factory=dict)  # 其他扩展信息
-
-    def to_metadata(self) -> Dict[str, str]:
-        """转换为gRPC metadata格式"""
-        metadata = {}
-
-        if self.request_id:
-            metadata['x-request-id'] = self.request_id
-        if self.client_ip:
-            metadata['x-client-ip'] = self.client_ip
-        if self.client_version:
-            metadata['x-client-version'] = self.client_version
-        if self.client_type:
-            metadata['x-client-type'] = self.client_type
-        if self.user_agent:
-            metadata['x-user-agent'] = self.user_agent
-        if self.timestamp:
-            metadata['x-timestamp'] = self.timestamp.isoformat()
-
-        # 添加扩展信息
-        for key, value in self.extra.items():
-            metadata[f'x-{key}'] = str(value)
-
-        return metadata
-
-    @classmethod
-    def from_metadata(cls, metadata: Dict[str, str]) -> 'RequestContext':
-        """从metadata中解析请求上下文"""
-        # 提取标准字段
-        request_id = metadata.get('x-request-id')
-        client_ip = metadata.get('x-client-ip')
-        client_version = metadata.get('x-client-version')
-        client_type = metadata.get('x-client-type')
-        user_agent = metadata.get('x-user-agent')
-
-        # 解析时间戳
-        timestamp = None
-        if 'x-timestamp' in metadata:
-            try:
-                timestamp = datetime.fromisoformat(metadata['x-timestamp'])
-            except:
-                pass
-
-        # 提取扩展字段
-        extra = {}
-        for key, value in metadata.items():
-            if key.startswith('x-') and key not in [
-                'x-request-id', 'x-client-ip', 'x-client-version',
-                'x-client-type', 'x-user-agent', 'x-timestamp',
-                'x-org-id', 'x-user-id', 'x-actor-id', 'x-role'
-            ]:
-                extra[key[2:]] = value  # 去掉 'x-' 前缀
-
-        return cls(
-            request_id=request_id,
-            client_ip=client_ip,
-            client_version=client_version,
-            client_type=client_type,
-            user_agent=user_agent,
-            timestamp=timestamp,
-            extra=extra
-        )
-
-
-@dataclass
-class FullContext:
-    """完整的上下文信息，包含用户上下文和请求上下文"""
-    user_context: Optional[UserContext] = None
-    request_context: Optional[RequestContext] = None
-
-    def to_metadata(self) -> Dict[str, str]:
-        """转换为gRPC metadata格式"""
-        metadata = {}
-
-        if self.user_context:
-            metadata.update(self.user_context.to_metadata())
-
-        if self.request_context:
-            metadata.update(self.request_context.to_metadata())
-
-        return metadata
-
-    @classmethod
-    def from_metadata(cls, metadata: Dict[str, str]) -> 'FullContext':
-        """从metadata中解析完整上下文"""
-        return cls(
-            user_context=UserContext.from_metadata(metadata),
-            request_context=RequestContext.from_metadata(metadata)
-        )
+"""
+用户上下文和请求上下文相关的Schema定义
+"""
+from dataclasses import dataclass, field
+from typing import Optional, Dict, Any
+from datetime import datetime
+
+from file_hub_client.enums import Role
+
+
+@dataclass
+class UserContext:
+    """
+    用户上下文信息
+    
+    包含两大类信息：
+    1. ownership（所有权）: org_id, user_id - 表示资源归属
+    2. operator（操作者）: actor_id, role - 表示实际操作者（可能是用户、agent或系统）
+    3. request info（请求信息）: user_ip - 表示请求来源IP（用于审计和安全）
+    """
+    # Ownership - 资源所有权信息
+    org_id: str  # 组织ID
+    user_id: str  # 用户ID
+
+    # Operator - 操作者信息
+    actor_id: Optional[str] = None  # 实际操作者ID（如果为空，默认使用user_id）
+    role: Role = Role.ACCOUNT  # 操作者角色（ACCOUNT, AGENT, SYSTEM等）
+    
+    # Request info - 请求信息
+    user_ip: Optional[str] = None  # 用户IP地址（请求来源IP）
+
+    def __post_init__(self):
+        """初始化后处理，如果actor_id为空，默认使用user_id"""
+        if self.actor_id is None:
+            self.actor_id = self.user_id
+
+    def to_metadata(self) -> Dict[str, str]:
+        """转换为gRPC metadata格式"""
+        metadata = {
+            'x-org-id': self.org_id,
+            'x-user-id': self.user_id,
+            'x-actor-id': self.actor_id,
+            'x-role': self.role,
+        }
+        
+        # 只有当user_ip不为None时才添加x-user-ip
+        if self.user_ip is not None:
+            metadata['x-user-ip'] = self.user_ip
+            
+        return metadata
+
+    @classmethod
+    def from_metadata(cls, metadata: Dict[str, str]) -> Optional['UserContext']:
+        """从metadata中解析用户上下文"""
+        org_id = metadata.get('x-org-id')
+        user_id = metadata.get('x-user-id')
+
+        if not org_id or not user_id:
+            return None
+
+        return cls(
+            org_id=org_id,
+            user_id=user_id,
+            actor_id=metadata.get('x-actor-id'),
+            role=Role(metadata.get('x-role', Role.ACCOUNT)),
+            user_ip=metadata.get('x-user-ip')
+        )
+
+
+@dataclass
+class RequestContext:
+    """
+    请求上下文信息
+    
+    包含请求相关的元数据，如客户端信息、请求追踪等
+    """
+    request_id: Optional[str] = None  # 请求ID，用于追踪
+    client_ip: Optional[str] = None  # 客户端IP地址
+    client_version: Optional[str] = None  # 客户端版本
+    client_type: Optional[str] = None  # 客户端类型（web, mobile, desktop, cli等）
+    user_agent: Optional[str] = None  # User-Agent信息
+    timestamp: Optional[datetime] = field(default_factory=datetime.now)  # 请求时间戳
+    extra: Dict[str, Any] = field(default_factory=dict)  # 其他扩展信息
+
+    def to_metadata(self) -> Dict[str, str]:
+        """转换为gRPC metadata格式"""
+        metadata = {}
+
+        if self.request_id:
+            metadata['x-request-id'] = self.request_id
+        if self.client_ip:
+            metadata['x-client-ip'] = self.client_ip
+        if self.client_version:
+            metadata['x-client-version'] = self.client_version
+        if self.client_type:
+            metadata['x-client-type'] = self.client_type
+        if self.user_agent:
+            metadata['x-user-agent'] = self.user_agent
+        if self.timestamp:
+            metadata['x-timestamp'] = self.timestamp.isoformat()
+
+        # 添加扩展信息
+        for key, value in self.extra.items():
+            metadata[f'x-{key}'] = str(value)
+
+        return metadata
+
+    @classmethod
+    def from_metadata(cls, metadata: Dict[str, str]) -> 'RequestContext':
+        """从metadata中解析请求上下文"""
+        # 提取标准字段
+        request_id = metadata.get('x-request-id')
+        client_ip = metadata.get('x-client-ip')
+        client_version = metadata.get('x-client-version')
+        client_type = metadata.get('x-client-type')
+        user_agent = metadata.get('x-user-agent')
+
+        # 解析时间戳
+        timestamp = None
+        if 'x-timestamp' in metadata:
+            try:
+                timestamp = datetime.fromisoformat(metadata['x-timestamp'])
+            except:
+                pass
+
+        # 提取扩展字段
+        extra = {}
+        for key, value in metadata.items():
+            if key.startswith('x-') and key not in [
+                'x-request-id', 'x-client-ip', 'x-client-version',
+                'x-client-type', 'x-user-agent', 'x-timestamp',
+                'x-org-id', 'x-user-id', 'x-actor-id', 'x-role'
+            ]:
+                extra[key[2:]] = value  # 去掉 'x-' 前缀
+
+        return cls(
+            request_id=request_id,
+            client_ip=client_ip,
+            client_version=client_version,
+            client_type=client_type,
+            user_agent=user_agent,
+            timestamp=timestamp,
+            extra=extra
+        )
+
+
+@dataclass
+class FullContext:
+    """完整的上下文信息，包含用户上下文和请求上下文"""
+    user_context: Optional[UserContext] = None
+    request_context: Optional[RequestContext] = None
+
+    def to_metadata(self) -> Dict[str, str]:
+        """转换为gRPC metadata格式"""
+        metadata = {}
+
+        if self.user_context:
+            metadata.update(self.user_context.to_metadata())
+
+        if self.request_context:
+            metadata.update(self.request_context.to_metadata())
+
+        return metadata
+
+    @classmethod
+    def from_metadata(cls, metadata: Dict[str, str]) -> 'FullContext':
+        """从metadata中解析完整上下文"""
+        return cls(
+            user_context=UserContext.from_metadata(metadata),
+            request_context=RequestContext.from_metadata(metadata)
+        )

{tamar_file_hub_client-0.1.6 → tamar_file_hub_client-0.1.7}/file_hub_client/services/file/async_blob_service.py

@@ -453,18 +453,33 @@ class AsyncBlobService(BaseFileService):
             # 使用下载的内容作为file参数
             file = downloaded_content
 
-            # 基于文件名计算MIME类型
-            mime_type = get_file_mime_type(Path(file_name))
+            # MIME类型优先级：用户指定 > 内容检测 > URL文件名推断
+            if mime_type:
+                # 用户明确提供的MIME类型优先级最高，无需进行内容检测
+                final_mime_type = mime_type
+            else:
+                # 用户未提供MIME类型，进行内容检测和文件名推断
+                content_detected_mime = self._detect_mime_from_content(downloaded_content)
+                url_filename_mime = get_file_mime_type(Path(file_name))
+                
+                if content_detected_mime != "application/octet-stream":
+                    # 内容检测到了具体的MIME类型，使用内容检测的
+                    final_mime_type = content_detected_mime
+                else:
+                    # 内容检测失败，使用从URL文件名推断的MIME类型
+                    final_mime_type = url_filename_mime
             
-            # 提取文件信息，传入MIME类型用于推断扩展名
-            _, content, file_size, _, _, file_hash = self._extract_file_info(file, mime_type)
+            # 提取文件信息，传入最终确定的MIME类型
+            _, content, file_size, extract_mime_type, extract_file_type, file_hash = self._extract_file_info(file, final_mime_type)
 
             # file_name已经在上面设置了（要么是用户指定的，要么是从URL提取的）
             extracted_file_name = file_name
+            
+            # 使用最终确定的MIME类型
+            mime_type = final_mime_type
 
-            # 基于文件名计算文件类型
-            file_type = Path(extracted_file_name).suffix.lstrip('.').lower() if Path(
-                extracted_file_name).suffix else 'dat'
+            # 基于最终MIME类型计算文件扩展名
+            file_type = extract_file_type
         else:
             # 解析文件参数，提取文件信息
             # 如果用户指定了文件名，先从文件名推断MIME类型，然后传给_extract_file_info
@@ -485,7 +500,9 @@ class AsyncBlobService(BaseFileService):
                 # 没有指定文件名，传入用户提供的MIME类型（如果有）
                 extracted_file_name, content, file_size, extract_mime_type, extract_file_type, file_hash = self._extract_file_info(
                     file, mime_type)
-                mime_type = extract_mime_type
+                # 如果用户指定了MIME类型，使用用户指定的，否则使用检测的
+                if not mime_type:
+                    mime_type = extract_mime_type
                 file_type = extract_file_type
 
         # 根据文件大小自动选择上传模式

{tamar_file_hub_client-0.1.6 → tamar_file_hub_client-0.1.7}/file_hub_client/services/file/sync_blob_service.py

@@ -452,18 +452,33 @@ class SyncBlobService(BaseFileService):
             # 使用下载的内容作为file参数
             file = downloaded_content
 
-            # 基于文件名计算MIME类型
-            mime_type = get_file_mime_type(Path(file_name))
+            # MIME类型优先级：用户指定 > 内容检测 > URL文件名推断
+            if mime_type:
+                # 用户明确提供的MIME类型优先级最高，无需进行内容检测
+                final_mime_type = mime_type
+            else:
+                # 用户未提供MIME类型，进行内容检测和文件名推断
+                content_detected_mime = self._detect_mime_from_content(downloaded_content)
+                url_filename_mime = get_file_mime_type(Path(file_name))
+                
+                if content_detected_mime != "application/octet-stream":
+                    # 内容检测到了具体的MIME类型，使用内容检测的
+                    final_mime_type = content_detected_mime
+                else:
+                    # 内容检测失败，使用从URL文件名推断的MIME类型
+                    final_mime_type = url_filename_mime
             
-            # 提取文件信息，传入MIME类型用于推断扩展名
-            _, content, file_size, _, _, file_hash = self._extract_file_info(file, mime_type)
+            # 提取文件信息，传入最终确定的MIME类型
+            _, content, file_size, extract_mime_type, extract_file_type, file_hash = self._extract_file_info(file, final_mime_type)
 
             # file_name已经在上面设置了（要么是用户指定的，要么是从URL提取的）
             extracted_file_name = file_name
+            
+            # 使用最终确定的MIME类型
+            mime_type = final_mime_type
 
-            # 基于文件名计算文件类型
-            file_type = Path(extracted_file_name).suffix.lstrip('.').lower() if Path(
-                extracted_file_name).suffix else 'dat'
+            # 基于最终MIME类型计算文件扩展名
+            file_type = extract_file_type
         else:
             # 解析文件参数，提取文件信息
             # 如果用户指定了文件名，先从文件名推断MIME类型，然后传给_extract_file_info
@@ -484,7 +499,9 @@ class SyncBlobService(BaseFileService):
                 # 没有指定文件名，传入用户提供的MIME类型（如果有）
                 extracted_file_name, content, file_size, extract_mime_type, extract_file_type, file_hash = self._extract_file_info(
                     file, mime_type)
-                mime_type = extract_mime_type
+                # 如果用户指定了MIME类型，使用用户指定的，否则使用检测的
+                if not mime_type:
+                    mime_type = extract_mime_type
                 file_type = extract_file_type
 
         # 根据文件大小自动选择上传模式

{tamar_file_hub_client-0.1.6 → tamar_file_hub_client-0.1.7}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="tamar-file-hub-client",
-    version="0.1.6",
+    version="0.1.7",
     description="A Python SDK for gRPC-based file management system",
     long_description=open("README.md", encoding="utf-8").read(),
     long_description_content_type="text/markdown",

{tamar_file_hub_client-0.1.6 → tamar_file_hub_client-0.1.7}/tamar_file_hub_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tamar-file-hub-client
-Version: 0.1.6
+Version: 0.1.7
 Summary: A Python SDK for gRPC-based file management system
 Home-page: https://github.com/Tamar-Edge-AI/file-hub-client
 Author: Oscar Ou