import odoorpc
import base64
import os
from datetime import datetimeclass OdooAttachmentDownloader:def __init__(self, host, port, database, username, password):self.host = hostself.port = portself.database = databaseself.username = usernameself.password = passwordself.odoo = Nonedef connect(self):"""连接到Odoo服务器"""try:self.odoo = odoorpc.ODOO(self.host, port=self.port)self.odoo.login(self.database, self.username, self.password)print("成功连接到Odoo服务器")return Trueexcept Exception as e:print(f"连接失败: {e}")return Falsedef download_attachments_by_record(self, model_name, record_id, save_dir="./attachments"):"""下载指定单据的所有附件Args:model_name: 模型名称，如 'sale.order', 'account.move'record_id: 记录IDsave_dir: 保存目录"""if not self.odoo:print("请先连接到Odoo服务器")return []os.makedirs(save_dir, exist_ok=True)try:Attachment = self.odoo.env['ir.attachment']attachment_ids = Attachment.search([('res_model', '=', model_name),('res_id', '=', record_id)])if not attachment_ids:print(f"未找到模型 {model_name} 记录 {record_id} 的附件")return []attachments = Attachment.read(attachment_ids, ['name', 'datas', 'mimetype', 'create_date'])downloaded_files = []for attachment in attachments:if attachment.get('datas'):file_data = base64.b64decode(attachment['datas'])file_name = attachment['name']save_path = os.path.join(save_dir, file_name)counter = 1while os.path.exists(save_path):name, ext = os.path.splitext(file_name)save_path = os.path.join(save_dir, f"{name}_{counter}{ext}")counter += 1with open(save_path, 'wb') as f:f.write(file_data)file_info = {'file_path': save_path,'file_name': os.path.basename(save_path),'original_name': file_name,'mime_type': attachment.get('mimetype', ''),'create_date': attachment.get('create_date', ''),'size': len(file_data)}downloaded_files.append(file_info)print(f"✓ 已下载: {file_info['file_name']} ({file_info['size']} bytes)")else:print(f"⚠ 附件 {attachment['name']} 无数据内容")print(f"\n总共下载了 {len(downloaded_files)} 个文件到目录: {save_dir}")return downloaded_filesexcept Exception as e:print(f"下载附件时出错: {e}")return []def main():downloader = OdooAttachmentDownloader(host='10.86.113.88',port=8069,database='odoo17',username='admin',password='123123sdt')if downloader.connect():files = downloader.download_attachments_by_record(model_name='cbd.huohao.price.form.head',record_id=659,save_dir='./downloaded_attachments')for file in files:print(f"文件: {file['file_name']}, 大小: {file['size']} bytes")if __name__ == "__main__":main()
