新增产品状态检查功能

- 处理前检查产品是否已存在（已发布/待审核） - 避免重复处理已有产品 - 新增 check_product_exists API 方法 - 更新 README 文档
2026-07-16 23:35:39 +08:00
parent a367887ace
commit 11edee581b
3 changed files with 156 additions and 2 deletions
@@ -127,6 +127,114 @@ class ParamHubClient:
        except Exception as e:
            print(f"发送通知失败: {str(e)}")
            return False
+    
+    def check_product_exists(self, product_name):
+        """
+        检查产品是否已存在于系统中（已发布或待审核）
+        
+        Args:
+            product_name: 产品名称
+        
+        Returns:
+            {
+                'exists': bool,
+                'status': str ('published'/'pending'/None),
+                'message': str
+            }
+        """
+        try:
+            if not self.session:
+                self.login()
+            
+            # 1. 检查已发布的产品（通过搜索API）
+            response = self.session.get(
+                f'{self.base_url}/api/search',
+                params={'q': product_name}
+            )
+            
+            if response.status_code == 200:
+                result = response.json()
+                
+                # 检查是否匹配（精确匹配或包含匹配）
+                product_name_lower = product_name.lower().strip()
+                
+                # 检查已发布的模型
+                for model in result.get('models', []):
+                    name = model.get('name', '').lower().strip()
+                    if product_name_lower == name or product_name_lower in name or name in product_name_lower:
+                        return {
+                            'exists': True,
+                            'status': 'published',
+                            'message': f'产品 "{product_name}" 已在已发布的模型中',
+                            'category': 'ai-models',
+                            'data': model
+                        }
+                
+                # 检查已发布的GPU
+                for gpu in result.get('gpus', []):
+                    name = gpu.get('name', '').lower().strip()
+                    if product_name_lower == name or product_name_lower in name or name in product_name_lower:
+                        return {
+                            'exists': True,
+                            'status': 'published',
+                            'message': f'产品 "{product_name}" 已在已发布的GPU中',
+                            'category': 'gpus',
+                            'data': gpu
+                        }
+                
+                # 检查已发布的CPU
+                for cpu in result.get('cpus', []):
+                    name = cpu.get('name', '').lower().strip()
+                    if product_name_lower == name or product_name_lower in name or name in product_name_lower:
+                        return {
+                            'exists': True,
+                            'status': 'published',
+                            'message': f'产品 "{product_name}" 已在已发布的CPU中',
+                            'category': 'cpus',
+                            'data': cpu
+                        }
+            
+            
+            # 2. 检查待审核列表
+            response = self.session.get(
+                f'{self.base_url}/api/reviews',
+                params={'status': 'pending'}
+            )
+            
+            if response.status_code == 200:
+                reviews = response.json()
+                product_name_lower = product_name.lower().strip()
+                
+                for review in reviews:
+                    review_data = review.get('data', {})
+                    review_name = review_data.get('name', '').lower().strip()
+                    
+                    if product_name_lower == review_name or product_name_lower in review_name or review_name in product_name_lower:
+                        return {
+                            'exists': True,
+                            'status': 'pending',
+                            'message': f'产品 "{product_name}" 已在待审核列表中',
+                            'review_id': review.get('id'),
+                            'category': review.get('category_id'),
+                            'data': review
+                        }
+            
+            
+            # 产品不存在
+            return {
+                'exists': False,
+                'status': None,
+                'message': f'产品 "{product_name}" 未在系统中找到'
+            }
+            
+        except Exception as e:
+            print(f"检查产品是否存在失败: {str(e)}")
+            # 出错时返回不存在，允许后续处理
+            return {
+                'exists': False,
+                'status': None,
+                'message': f'检查失败: {str(e)}'
+            }

 # 全局客户端实例
 paramhub_client = ParamHubClient()