forked from erp-dev/erp
228 lines
6.1 KiB
Python
228 lines
6.1 KiB
Python
# import aiohttp
|
||
# from typing import Optional, Dict, Any, List
|
||
# from pydantic import BaseModel, Field, computed_field
|
||
# from enum import Enum
|
||
# import json
|
||
|
||
|
||
# class Product(BaseModel):
|
||
# """产品模型 - 请在此填入你的字段"""
|
||
# uid: str
|
||
# name: str
|
||
# unit: str | None
|
||
# color: str | None
|
||
# detail_str: str | None = Field(exclude=True) # 在序列化时排除此字段
|
||
# created_at: str
|
||
|
||
# @computed_field
|
||
# @property
|
||
# def detail(self) -> dict[str, Any]:
|
||
# """将 detail_str 转换为字典"""
|
||
# try:
|
||
# return json.loads(self.detail_str) if self.detail_str else {}
|
||
# except (json.JSONDecodeError, TypeError):
|
||
# return {}
|
||
|
||
|
||
# class ProductListResponse(BaseModel):
|
||
# """产品列表响应"""
|
||
# products: List[Product]
|
||
# total: int
|
||
|
||
|
||
# mdy_table_map = {
|
||
# 'product': 'spmx',
|
||
# 'customer': 'quanbu',
|
||
# }
|
||
|
||
|
||
# class HTTPMethod(Enum):
|
||
# """HTTP 请求方法枚举"""
|
||
# GET = "GET"
|
||
# POST = "POST"
|
||
|
||
|
||
# class MingDaoYunClient:
|
||
# """明道云 API 客户端"""
|
||
|
||
# def __init__(self, app_key: str, sign: str, base_url: str = ""):
|
||
# """
|
||
# 初始化客户端
|
||
|
||
# Args:
|
||
# app_key: 应用密钥
|
||
# sign: 签名
|
||
# base_url: API 基础地址
|
||
# """
|
||
# self.app_key = app_key
|
||
# self.sign = sign
|
||
# self.base_url = base_url
|
||
|
||
# async def request(
|
||
# self,
|
||
# method: HTTPMethod,
|
||
# endpoint: str,
|
||
# data: Optional[Dict[str, Any]] = None,
|
||
# params: Optional[Dict[str, Any]] = None,
|
||
# headers: Optional[Dict[str, str]] = None,
|
||
# ) -> Any:
|
||
# """
|
||
# 通用异步请求函数
|
||
|
||
# Args:
|
||
# method: 请求方法 (GET/POST)
|
||
# endpoint: API 端点路径
|
||
# data: POST 请求体数据（JSON）
|
||
# params: URL 查询参数
|
||
# headers: 自定义请求头
|
||
|
||
# Returns:
|
||
# 响应 JSON 数据
|
||
|
||
# Raises:
|
||
# aiohttp.ClientError: 请求失败时抛出
|
||
# """
|
||
# url = f"{self.base_url}{endpoint}"
|
||
|
||
# # 构建默认请求头
|
||
# default_headers = {
|
||
# "Content-Type": "application/json",
|
||
# }
|
||
|
||
# if headers:
|
||
# default_headers.update(headers)
|
||
|
||
# # 添加认证参数
|
||
# auth_params = {
|
||
# "appKey": self.app_key,
|
||
# "sign": self.sign
|
||
# }
|
||
|
||
# async with aiohttp.ClientSession() as session:
|
||
# if method == HTTPMethod.GET:
|
||
# async with session.get(url, params=params, json=auth_params) as response:
|
||
# response.raise_for_status()
|
||
# return await response.json()
|
||
|
||
# elif method == HTTPMethod.POST:
|
||
# if data is None:
|
||
# data = {}
|
||
|
||
# data.update(auth_params)
|
||
# async with session.post(url, json=data) as response:
|
||
# response.raise_for_status()
|
||
# return await response.json()
|
||
|
||
# else:
|
||
# raise ValueError(f"Unsupported HTTP method: {method}")
|
||
|
||
# async def get(
|
||
# self,
|
||
# endpoint: str,
|
||
# params: Optional[Dict[str, Any]] = None,
|
||
# headers: Optional[Dict[str, str]] = None,
|
||
# ) -> Any:
|
||
# """
|
||
# 发送 GET 请求
|
||
|
||
# Args:
|
||
# endpoint: API 端点路径
|
||
# params: URL 查询参数
|
||
# headers: 自定义请求头
|
||
|
||
# Returns:
|
||
# 响应 JSON 数据
|
||
# """
|
||
# return await self.request(
|
||
# method=HTTPMethod.GET,
|
||
# endpoint=endpoint,
|
||
# params=params,
|
||
# headers=headers,
|
||
# )
|
||
|
||
# async def post(
|
||
# self,
|
||
# endpoint: str,
|
||
# data: Optional[Dict[str, Any]] = None,
|
||
# params: Optional[Dict[str, Any]] = None,
|
||
# headers: Optional[Dict[str, str]] = None,
|
||
# ) -> Any:
|
||
# """
|
||
# 发送 POST 请求
|
||
|
||
# Args:
|
||
# endpoint: API 端点路径
|
||
# data: POST 请求体数据（JSON）
|
||
# params: URL 查询参数
|
||
# headers: 自定义请求头
|
||
|
||
# Returns:
|
||
# 响应 JSON 数据
|
||
# """
|
||
# return await self.request(
|
||
# method=HTTPMethod.POST,
|
||
# endpoint=endpoint,
|
||
# data=data,
|
||
# params=params,
|
||
# headers=headers,
|
||
# )
|
||
|
||
|
||
# product_type_map = {
|
||
# 'name': '668caa5eb80969563ecaee7d',
|
||
# 'color': '66ed3b9ae01d5599bdb45f6d',
|
||
# 'uid': '668caa5eb80969563ecaee7c',
|
||
# 'unit': '668e2ee2ca758c8cc5d0dc1d',
|
||
# 'detail_str': '668caa5eb80969563ecaee7e',
|
||
# 'created_at': 'ctime',
|
||
# }
|
||
|
||
|
||
# # json param example:
|
||
# # {
|
||
# # "appKey": "208e55fea5cea59f",
|
||
# # "sign": "MWU0YmViYjkwZmM1ZDIzYzRiN2U3ZGQ4MmE4ZGNkMjc0MWM1ZmQ2ZjkwMjljODE4YmNkZTBhMzA0OTU2YzE2NA==",
|
||
# # "worksheetId": "quanbu",
|
||
# # "listType": 1,
|
||
# # "sortId": "ctime",
|
||
# # "isAsc": false,
|
||
# # "notGetTotal": true
|
||
# # }
|
||
|
||
|
||
# def pick_product(fields: dict) -> Product:
|
||
# """
|
||
# 从字段字典中提取产品信息
|
||
# """
|
||
# data = {k: fields.get(v, '') for k, v in product_type_map.items()}
|
||
# return Product(**data)
|
||
|
||
|
||
# async def fetch_products_from_mingdaoyun(page: int = 1, page_size: int = 100) -> tuple[list[Product], int]:
|
||
# """
|
||
# 从明道云获取产品列表
|
||
|
||
# Returns:
|
||
# tuple: (产品列表, 总数量)
|
||
# """
|
||
# client = MingDaoYunClient(
|
||
# app_key='208e55fea5cea59f',
|
||
# sign='MWU0YmViYjkwZmM1ZDIzYzRiN2U3ZGQ4MmE4ZGNkMjc0MWM1ZmQ2ZjkwMjljODE4YmNkZTBhMzA0OTU2YzE2NA==',
|
||
# base_url="https://api.mingdao.com"
|
||
# )
|
||
# response = await client.post(
|
||
# endpoint='/v2/open/worksheet/getFilterRows',
|
||
# data={
|
||
# 'worksheetId': 'spmx',
|
||
# 'pageIndex': page,
|
||
# 'pageSize': page_size,
|
||
# }
|
||
# )
|
||
# data = response.get('data')
|
||
# if not data:
|
||
# return [], 0
|
||
|
||
# products = [pick_product(item) for item in data.get('rows', [])]
|
||
# total_count = data.get('total', 0)
|
||
# return products, total_count
|