62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
from odoo import models, fields, api, _, SUPERUSER_ID
|
|
from odoo.exceptions import ValidationError, UserError
|
|
from pypinyin import pinyin, Style
|
|
import xlrd
|
|
import base64
|
|
import datetime
|
|
from datetime import datetime, timedelta
|
|
|
|
class SalaryPartyWizard(models.TransientModel):
|
|
_name = 'salary.party.wizard'
|
|
_description = '党建导入弹窗'
|
|
|
|
excel_file = fields.Binary(string="上传Excel表")
|
|
excel_name = fields.Char(string="文件名")
|
|
|
|
def confirm(self):
|
|
if not self.excel_file:
|
|
raise ValidationError('请上传文件。')
|
|
book = xlrd.open_workbook(file_contents=base64.decodebytes(self.excel_file))
|
|
sh = book.sheet_by_index(0)
|
|
for rx in range(1, sh.nrows):
|
|
row = sh.row(rx)
|
|
self.salary_year_ttjt_data(row)
|
|
|
|
def salary_party_data(self, row):
|
|
select_year = row[0].value
|
|
select_time = row[1].value
|
|
employee_id = self.env['hr.employee'].search([('identification_id', '=', row[3].value)], limit=1)
|
|
card = row[3].value
|
|
fee = row[4].value
|
|
payment = row[5].value
|
|
total_payment = row[6].value
|
|
if employee_id:
|
|
party_id = self.env['yuthon.salary.party'].search([
|
|
('employee_id', '=', employee_id.id),
|
|
('select_year', '=', select_year),
|
|
('select_time', '=', select_time)
|
|
])
|
|
if not party_id:
|
|
self.env['yuthon.salary.party'].create({
|
|
'employee_id': employee_id.id,
|
|
'card': card,
|
|
'select_year': select_year,
|
|
'select_time': select_time,
|
|
'fee': fee,
|
|
'payment': payment,
|
|
'total_payment': total_payment,
|
|
})
|
|
else:
|
|
party_id.write({
|
|
'card': card,
|
|
'fee': fee,
|
|
'payment': payment,
|
|
'total_payment': total_payment,
|
|
}
|
|
)
|
|
|
|
|
|
|
|
|