++
This commit is contained in:
jack
@@ -0,0 +1,249 @@
|
||||
import xmlrpc.client
|
||||
import time
|
||||
import sys
|
||||
|
||||
# Odoo 连接配置
|
||||
ODOO_URL = "http://192.168.31.41:32000"
|
||||
DB_NAME = "quantify"
|
||||
USERNAME = "rpc"
|
||||
PASSWORD = "aaaAAA111"
|
||||
|
||||
|
||||
class OdooClient:
|
||||
def __init__(self, url, db, username, password):
|
||||
self.url = url
|
||||
self.db = db
|
||||
self.username = username
|
||||
self.password = password
|
||||
self.uid = None
|
||||
self.models = None
|
||||
self._connect()
|
||||
|
||||
def _connect(self):
|
||||
"""建立 XML-RPC 连接"""
|
||||
try:
|
||||
common = xmlrpc.client.ServerProxy(f"{self.url}/xmlrpc/2/common")
|
||||
self.uid = common.authenticate(self.db, self.username, self.password, {})
|
||||
if not self.uid:
|
||||
raise Exception("认证失败，请检查用户名和密码")
|
||||
self.models = xmlrpc.client.ServerProxy(f"{self.url}/xmlrpc/2/object")
|
||||
print(f"[INFO] 连接成功，UID: {self.uid}")
|
||||
except Exception as e:
|
||||
print(f"[ERROR] 连接失败: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
def search_draft_records(self):
|
||||
"""搜索所有状态为 draft 的 alpha.datasets 记录"""
|
||||
try:
|
||||
ids = self.models.execute_kw(
|
||||
self.db, self.uid, self.password,
|
||||
'alpha.datasets', 'search',
|
||||
[[('status', '=', 'draft')]]
|
||||
)
|
||||
print(f"[INFO] 找到 {len(ids)} 条 draft 记录: {ids}")
|
||||
return ids
|
||||
except xmlrpc.client.Fault as e:
|
||||
print(f"[ERROR] XML-RPC Fault 搜索记录失败: {e.faultCode} - {e.faultString}")
|
||||
return []
|
||||
except Exception as e:
|
||||
print(f"[ERROR] 搜索 draft 记录失败: {e}")
|
||||
return []
|
||||
|
||||
def get_record_status(self, record_id):
|
||||
"""获取单条记录的状态"""
|
||||
try:
|
||||
records = self.models.execute_kw(
|
||||
self.db, self.uid, self.password,
|
||||
'alpha.datasets', 'read',
|
||||
[[record_id], ['status', 'result_message', 'datasets_id', 'name']]
|
||||
)
|
||||
if records:
|
||||
return records[0]
|
||||
return None
|
||||
except xmlrpc.client.Fault as e:
|
||||
print(f"[ERROR] XML-RPC Fault 获取记录 {record_id} 状态失败: {e.faultCode} - {e.faultString}")
|
||||
return None
|
||||
except Exception as e:
|
||||
print(f"[ERROR] 获取记录 {record_id} 状态失败: {e}")
|
||||
return None
|
||||
|
||||
def trigger_btn_get_datasets(self, record_id):
|
||||
"""
|
||||
调用 btn_get_datasets 方法
|
||||
返回: (success, message)
|
||||
"""
|
||||
try:
|
||||
result = self.models.execute_kw(
|
||||
self.db, self.uid, self.password,
|
||||
'alpha.datasets', 'btn_get_datasets',
|
||||
[[record_id]]
|
||||
)
|
||||
|
||||
# 处理不同的返回值
|
||||
if result is True:
|
||||
print(f"[INFO] 记录 {record_id} 的 btn_get_datasets 调用成功")
|
||||
return True, "调用成功"
|
||||
elif isinstance(result, dict) and result.get('type') == 'ir.actions.client':
|
||||
# 返回的是 notification 字典
|
||||
message = result.get('params', {}).get('message', '')
|
||||
msg_type = result.get('params', {}).get('type', 'info')
|
||||
if msg_type == 'danger':
|
||||
print(f"[WARN] 记录 {record_id} 返回错误通知: {message}")
|
||||
return False, message
|
||||
else:
|
||||
print(f"[INFO] 记录 {record_id} 返回通知: {message}")
|
||||
return True, message
|
||||
else:
|
||||
print(f"[INFO] 记录 {record_id} 的 btn_get_datasets 调用完成，返回: {result}")
|
||||
return True, "调用成功"
|
||||
|
||||
except xmlrpc.client.Fault as e:
|
||||
error_msg = f"XML-RPC Fault: {e.faultCode} - {e.faultString}"
|
||||
print(f"[ERROR] 调用记录 {record_id} 的 btn_get_datasets 失败: {error_msg}")
|
||||
return False, error_msg
|
||||
except Exception as e:
|
||||
error_msg = f"Exception: {str(e)}"
|
||||
print(f"[ERROR] 调用记录 {record_id} 的 btn_get_datasets 失败: {error_msg}")
|
||||
return False, error_msg
|
||||
|
||||
def wait_for_status_change(self, record_id, check_interval=5, max_wait_minutes=30):
|
||||
"""
|
||||
轮询等待记录状态变为 done 或 failed
|
||||
返回: (status, result_message)
|
||||
"""
|
||||
max_attempts = (max_wait_minutes * 60) // check_interval
|
||||
attempt = 0
|
||||
|
||||
while attempt < max_attempts:
|
||||
record = self.get_record_status(record_id)
|
||||
if not record:
|
||||
return 'unknown', '无法获取记录状态'
|
||||
|
||||
status = record.get('status')
|
||||
result_message = record.get('result_message', '')
|
||||
datasets_id = record.get('datasets_id', 'N/A')
|
||||
name = record.get('name', 'N/A')
|
||||
|
||||
if status in ['done', 'failed']:
|
||||
print(f"[INFO] 记录 {record_id} (name={name}, datasets_id={datasets_id}) 状态变为: {status}")
|
||||
if status == 'failed':
|
||||
print(f"[WARN] 失败原因: {result_message}")
|
||||
return status, result_message
|
||||
|
||||
print(f"[INFO] 记录 {record_id} 当前状态: {status}，等待 {check_interval} 秒后重试...")
|
||||
time.sleep(check_interval)
|
||||
attempt += 1
|
||||
|
||||
print(f"[WARN] 记录 {record_id} 等待超时 ({max_wait_minutes} 分钟)")
|
||||
return 'timeout', '等待超时'
|
||||
|
||||
|
||||
def main():
|
||||
"""主函数"""
|
||||
print("=" * 60)
|
||||
print("Odoo 批量处理 draft 数据集脚本")
|
||||
print("=" * 60)
|
||||
|
||||
# 初始化客户端
|
||||
client = OdooClient(ODOO_URL, DB_NAME, USERNAME, PASSWORD)
|
||||
|
||||
# 统计信息
|
||||
total_processed = 0
|
||||
success_count = 0
|
||||
failed_count = 0
|
||||
timeout_count = 0
|
||||
|
||||
while True:
|
||||
# 搜索所有 draft 记录
|
||||
draft_ids = client.search_draft_records()
|
||||
|
||||
if not draft_ids:
|
||||
print("[INFO] 没有更多 draft 记录，处理完成！")
|
||||
break
|
||||
|
||||
print(f"\n[INFO] 开始处理 {len(draft_ids)} 条 draft 记录...")
|
||||
|
||||
for record_id in draft_ids:
|
||||
print(f"\n--- 处理记录 ID: {record_id} ---")
|
||||
|
||||
# 再次确认状态（防止在搜索后状态被改变）
|
||||
current_record = client.get_record_status(record_id)
|
||||
if not current_record:
|
||||
print(f"[ERROR] 无法获取记录 {record_id} 的信息，跳过")
|
||||
failed_count += 1
|
||||
total_processed += 1
|
||||
continue
|
||||
|
||||
if current_record.get('status') != 'draft':
|
||||
print(f"[INFO] 记录 {record_id} 状态已不是 draft (当前: {current_record.get('status')})，跳过")
|
||||
total_processed += 1
|
||||
continue
|
||||
|
||||
datasets_id = current_record.get('datasets_id', 'N/A')
|
||||
name = current_record.get('name', 'N/A')
|
||||
print(f"[INFO] 数据集名称: {name}")
|
||||
print(f"[INFO] 数据集 ID: {datasets_id}")
|
||||
|
||||
# 触发 btn_get_datasets
|
||||
success, message = client.trigger_btn_get_datasets(record_id)
|
||||
|
||||
if not success:
|
||||
print(f"[ERROR] 触发失败: {message}，跳过记录 {record_id}")
|
||||
failed_count += 1
|
||||
total_processed += 1
|
||||
|
||||
# 更新记录状态为 failed（如果触发失败）
|
||||
try:
|
||||
client.models.execute_kw(
|
||||
client.db, client.uid, client.password,
|
||||
'alpha.datasets', 'write',
|
||||
[[record_id], {
|
||||
'status': 'failed',
|
||||
'result_message': f'Trigger failed: {message}'
|
||||
}]
|
||||
)
|
||||
except Exception as e:
|
||||
print(f"[WARN] 无法更新记录状态: {e}")
|
||||
continue
|
||||
|
||||
# 等待状态变化
|
||||
status, result_msg = client.wait_for_status_change(record_id)
|
||||
|
||||
if status == 'done':
|
||||
success_count += 1
|
||||
print(f"[SUCCESS] 记录 {record_id} 处理成功")
|
||||
elif status == 'failed':
|
||||
failed_count += 1
|
||||
print(f"[FAILED] 记录 {record_id} 处理失败: {result_msg}")
|
||||
elif status == 'timeout':
|
||||
timeout_count += 1
|
||||
failed_count += 1
|
||||
print(f"[TIMEOUT] 记录 {record_id} 处理超时")
|
||||
else:
|
||||
failed_count += 1
|
||||
print(f"[UNKNOWN] 记录 {record_id} 状态未知: {status}")
|
||||
|
||||
total_processed += 1
|
||||
|
||||
# 短暂延迟，避免请求过快
|
||||
time.sleep(1)
|
||||
|
||||
# 输出统计
|
||||
print("\n" + "=" * 60)
|
||||
print("处理完成统计:")
|
||||
print(f" 总处理记录数: {total_processed}")
|
||||
print(f" 成功: {success_count}")
|
||||
print(f" 失败: {failed_count - timeout_count}")
|
||||
print(f" 超时: {timeout_count}")
|
||||
print("=" * 60)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
main()
|
||||
except KeyboardInterrupt:
|
||||
print("\n[INFO] 用户中断脚本")
|
||||
sys.exit(0)
|
||||
except Exception as e:
|
||||
print(f"[ERROR] 脚本异常: {e}")
|
||||
sys.exit(1)
|
||||
Reference in New Issue
Block a user