123 lines
4.6 KiB
Python
123 lines
4.6 KiB
Python
from odoo import models, fields, api, _, SUPERUSER_ID
|
|
from odoo.exceptions import UserError, ValidationError
|
|
from dateutil.relativedelta import relativedelta
|
|
from datetime import datetime, time, timedelta
|
|
import xlrd
|
|
import base64
|
|
import pandas as pd
|
|
import re
|
|
|
|
class InitializeWizard(models.TransientModel):
|
|
_name = 'initialize.wizard'
|
|
_description = '应收账单初始化数据弹窗'
|
|
|
|
excel_file = fields.Binary(string="上传Excel表")
|
|
excel_name = fields.Char(string="文件名")
|
|
|
|
def confirm(self):
|
|
"""
|
|
导入excel文件,按行读取数据并处理，去重显示未找到的合同
|
|
"""
|
|
if not self.excel_file:
|
|
raise ValidationError('请上传文件。')
|
|
|
|
book = xlrd.open_workbook(file_contents=base64.decodebytes(self.excel_file))
|
|
sh = book.sheet_by_index(0)
|
|
success_count = 0
|
|
missing_contracts = set()
|
|
other_errors = [] # 记录其他错误
|
|
|
|
for rx in range(1, sh.nrows):
|
|
row = sh.row(rx)
|
|
contract_code = row[0].value
|
|
|
|
# 检查合同是否存在
|
|
contract_id = self.env['property.lease.contract'].search([('contract_code', '=', contract_code)], limit=1)
|
|
if not contract_id:
|
|
# 合同不存在，添加到集合（自动去重）
|
|
missing_contracts.add(contract_code)
|
|
continue
|
|
|
|
# 合同存在，处理该行数据
|
|
try:
|
|
self.property_initializer_data(row, contract_id)
|
|
success_count += 1
|
|
except Exception as e:
|
|
other_errors.append(f"合同 {contract_code} 处理失败：{str(e)}")
|
|
|
|
# 生成处理结果信息
|
|
result_msg = f"处理完成，成功导入 {success_count} 条数据"
|
|
|
|
# 添加未找到的合同信息（去重后）
|
|
if missing_contracts:
|
|
contracts_list = ", ".join(str(contract) for contract in missing_contracts)
|
|
result_msg += f"，以下合同不存在：{contracts_list}"
|
|
|
|
# 添加其他错误信息
|
|
if other_errors:
|
|
result_msg += "\n其他错误：\n" + "\n".join(other_errors)
|
|
|
|
# 显示处理结果
|
|
return {
|
|
'type': 'ir.actions.client',
|
|
'tag': 'display_notification',
|
|
'params': {
|
|
'message': result_msg,
|
|
'type': 'info',
|
|
'sticky': True,
|
|
}
|
|
}
|
|
|
|
def property_initializer_data(self, row, contract_id):
|
|
# 直接使用传入的已验证存在的contract_id
|
|
property_id = self.env['yuthon.property'].search([('name', '=', row[1].value)], limit=1)
|
|
lessee_id = self.env['property.tenant.information'].search([('name', '=', row[2].value)], limit=1)
|
|
|
|
due_date_value = row[3].value
|
|
if isinstance(due_date_value, float):
|
|
due_date = datetime(1899, 12, 30) + timedelta(days=due_date_value)
|
|
else:
|
|
due_date = datetime.strptime(due_date_value, '%Y-%m-%d')
|
|
|
|
due_date_end = False
|
|
if row[4].value:
|
|
end_date_value = row[4].value
|
|
if isinstance(end_date_value, float):
|
|
due_date_end = datetime(1899, 12, 30) + timedelta(days=end_date_value)
|
|
else:
|
|
due_date_end = datetime.strptime(end_date_value, '%Y-%m-%d')
|
|
|
|
original_amount = row[5].value
|
|
submitted = row[6].value
|
|
management_type_id = self.env['yuthon.management.type'].search([('name', '=', row[8].value)], limit=1)
|
|
remark = row[7].value
|
|
|
|
# 查找应收款记录
|
|
receivable = self.env['property.accounts.receivable'].search([
|
|
('contract_id', '=', contract_id.id),
|
|
('due_date', '=', due_date),
|
|
('due_date_end', '=', due_date_end),
|
|
])
|
|
|
|
if not receivable:
|
|
self.env['property.accounts.receivable'].create({
|
|
'contract_id': contract_id.id,
|
|
'lessee_id': lessee_id.id,
|
|
'due_date': due_date,
|
|
'due_date_end': due_date_end,
|
|
'original_amount': original_amount,
|
|
'submitted': submitted,
|
|
'management_type_id': management_type_id.id,
|
|
'remark': remark,
|
|
})
|
|
else:
|
|
receivable.write({
|
|
'contract_id': contract_id.id,
|
|
'lessee_id': lessee_id.id,
|
|
'due_date': due_date,
|
|
'due_date_end': due_date_end,
|
|
'original_amount': original_amount,
|
|
'submitted': submitted,
|
|
'management_type_id': management_type_id.id,
|
|
'remark': remark,
|
|
}) |