good/staff_expense/models/hr_expense.py

# Copyright 2016 上海开阖软件有限公司 (http://www.osbzr.com)
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

from odoo import api, fields, models, _
from odoo.exceptions import UserError
class HrExpense(models.Model):
    _name = 'hr.expense'
    _inherit = ['mail.thread', 'mail.activity.mixin']
    _description = '报销'
    name = fields.Char('单据编号', copy=False)
    date = fields.Date(string='报销日期', default=fields.Date.today, copy=False, required=True, help='报销发生日期')
    staff = fields.Many2one('staff', '报销员工', required=True, help='选择员工')
    invoice_all_total = fields.Float(string='费用金额合计', store=True, readonly=True, compute='_compute_invoice_all_total', tracking=True, digits='Amount')
    state = fields.Selection([('draft', '草稿'), ('confirm', '已提交'), ('done', '已支付')], '状态', default='draft', tracking=True, store=True, compute='_state_to_done')
    type = fields.Selection([('company', '付给公司'), ('my', '付给报销人')], string='支付方式', default='my', required=True, help='支付给个人时走其他付款单，支付给公司时走结算单')
    line_ids = fields.One2many('hr.expense.line', 'order_id', '明细发票行', copy=False)
    note = fields.Text('备注')
    bank_account_id = fields.Many2one('bank.account', '账号', ondelete='restrict', help='付给个人/公司')
    partner_id = fields.Many2one('partner', '供应商', help='直接支付给供应商')
    money_invoice = fields.Many2one('money.invoice', '对应结算单', readonly=True, ondelete='restrict', copy=False)
    other_money_order = fields.Many2one('other.money.order', '对应其他付款单', readonly=True, ondelete='restrict', copy=False)
    @api.depends('other_money_order.state')
    def _state_to_done(self):
        self.state = self.state or 'draft'
        if self.other_money_order and self.other_money_order.state == 'done':
            self.state = 'done'
        if self.other_money_order and self.other_money_order.state == 'draft':
            self.state = 'confirm'

    @api.depends('line_ids')
    def _compute_invoice_all_total(self):
        for bill in self:
            total = sum(i.invoice_total for i in bill.line_ids)
            bill.invoice_all_total = total

    def check_consistency(self):
        for bill in self:
            category_id = []
            for line in bill.line_ids:
                if line.category_id and line.category_id.id not in category_id:
                    category_id.append(line.category_id.id)
                    if line.category_id.type != 'expense' and bill.type == 'company':
                        raise UserError('申报支付给公司的费用必须属于采购类别！')
                if line.staff.id != bill.staff.id:
                    raise UserError('费用明细必须是同一人！')
            if bill.type == 'company' and len(category_id) > 1:
                raise UserError('申报支付给公司的费用必须同一类别！')

    @api.model_create_multi
    def create(self, vals_list):
        res = super().create(vals_list)
        res.check_consistency()
        return res

    def write(self, vals):
        res = super().write(vals)
        self.check_consistency()
        return res

    def unlink(self):
        for record in self:
            if record.state != 'draft':
                raise UserError('只能删除草稿状态的费用报销单')
        super().unlink()
    def hr_expense_confirm(self):
        for bill in self:
            if bill.state == 'confirm':
                raise UserError('请不要重复确认')
            if bill.type == 'company':
                bill.to_money_invoice()
            if bill.type == 'my':
                bill.to_other_money_order()
            bill.state = 'confirm'

    def hr_expense_draft(self):
        for bill in self:
            if bill.state == 'draft':
                raise UserError('请不要重复撤销 %s' % bill.name)
            if bill.other_money_order:
                other_money_order = bill.other_money_order
                bill.other_money_order = False
                if other_money_order.state == 'done':
                    other_money_order.other_money_draft()
                other_money_order.unlink()
            if bill.money_invoice:
                money_invoice = bill.money_invoice
                bill.money_invoice = False
                if money_invoice.state == 'done':
                    money_invoice.money_invoice_draft()
                money_invoice.unlink()
            bill.state = 'draft'
    def to_money_invoice(self):
        for bill in self:
            tax = 0
            bill_number = ''
            for line in bill.line_ids:
                if line.invoice_type == 'zy':
                    tax += line.invoice_tax
                if line.invoice_name:
                    bill_number = '%s, %s' % (line.invoice_name, bill_number)
            money_invoice = bill.env['money.invoice'].create({
                'name': bill.name,
                'partner_id': bill.partner_id.id,
                'category_id': bill.line_ids and bill.line_ids[0].category_id.id or False,
                'date': bill.date,
                'amount': bill.invoice_all_total,
                'reconciled': 0,
                'to_reconcile': bill.invoice_all_total,
                'date_due': fields.Date.context_today(bill),
                'state': 'draft',
                'tax_amount': tax,
                'bill_number': bill_number,
            })
            bill.write({'money_invoice': money_invoice.id})
    def to_other_money_order(self):
        for bill in self:
            other_money_order = bill.with_context(type='other_pay').env['other.money.order'].create({
                'state': 'draft',
                'partner_id': bill.partner_id and bill.partner_id.id or False,
                'date': bill.date,
                'bank_id': bill.bank_account_id.id,
            })
            bill.write({'other_money_order': other_money_order.id})
            for line in bill.line_ids:
                if line.invoice_type != 'zy':
                    invoice_tax = 0
                    invoice_amount = line.invoice_amount + line.invoice_tax
                else:
                    invoice_tax = line.invoice_tax
                    invoice_amount = line.invoice_amount
                bill.env['other.money.order.line'].create({
                    'other_money_id': other_money_order.id,
                    'amount': invoice_amount,
                    'tax_rate': round(line.invoice_amount and invoice_tax / line.invoice_amount * 100, 0) or 0,
                    'tax_amount': invoice_tax,
                    'category_id': line.category_id and line.category_id.id or False,
                })

class HrExpenseLine(models.Model):
    _name = 'hr.expense.line'
    _inherit = ['mail.thread']
    _description = '报销行'
    _order = 'date'
    name = fields.Char('单据编号', index=True, copy=False, help='报销单的唯一编号')
    staff = fields.Many2one('staff', required=True, string='报销员工', help='用关键字段查找并关联类别')
    invoice_type = fields.Selection([('pt', '增值税普通发票'), ('zy', '增值税专用发票'), ('dz', '电子普通发票')], '发票类别')
    invoice_code = fields.Char('发票代码', copy=False)
    invoice_name = fields.Char('发票号码', copy=False)
    invoice_amount = fields.Float(string='发票金额', digits='Amount', required=True, help='如是增值税发票请填不含税金额')
    invoice_tax = fields.Float(string='发票税额', digits='Amount', help='如是增值税发票请填不含税金额')
    invoice_total = fields.Float(string='发票金额合计', store=True, readonly=True, compute='_compute_cost_total', digits='Amount')
    invoice_heck_code = fields.Char('发票校验码', copy=False)
    invoice_date = fields.Char('发票日期', copy=False)
    note = fields.Text('备注')
    state = fields.Selection([('draft', '草稿'), ('confirm', '已提交')], '状态', default='draft', tracking=True, copy=False)
    category_id = fields.Many2one('core.category', '类别', ondelete='restrict', required=True, help='类型：运费、咨询费等')
    date = fields.Date(string='费用日期', default=fields.Date.context_today, copy=False, required=True, help='费用发生日期')
    order_id = fields.Many2one('hr.expense', '报销单号', index=True, copy=False, readonly=True)
    is_refused = fields.Boolean(string='已被使用', store=True, compute='_compute_is_choose', readonly=True, copy=False)
    is_pay = fields.Boolean(string='已付款', store=True, compute='_compute_is_pay', readonly=True, copy=False)
    _sql_constraints = [('unique_invoice_code_name', 'unique (invoice_code, invoice_name)', '发票代码+发票号码不能相同!')]
    attachment_number = fields.Integer(compute='_compute_attachment_number', string='附件号')
    def hr_expense_line_confirm(self):
        for bill in self:
            if bill.state == 'confirm':
                raise UserError('请不要重复确认')
            bill.state = 'confirm'

    def hr_expense_line_draft(self):
        for bill in self:
            if bill.state == 'draft':
                raise UserError('请不要重复撤销 %s' % bill.name)
            if bill.state == 'confirm' and not bill.order_id:
                bill.state = 'draft'
                bill.is_refused = False
            else:
                raise UserError('请先解除关联单据%s' % bill.order_id.name)

    @api.depends('invoice_amount', 'invoice_tax')
    def _compute_cost_total(self):
        for bill in self:
            bill.invoice_total = bill.invoice_amount + bill.invoice_tax

    @api.depends('order_id')
    def _compute_is_choose(self):
        for bill in self:
            if bill.order_id:
                bill.is_refused = True
            else:
                bill.is_refused = False
    @api.model
    def saomiaofapiao(self, model_name, barcode, order_id):
        barcode = barcode.replace('，', ',').replace('。', '.')
        code = barcode.split(',')
        invoice_heck_code = ''
        if len(code) < 5:
            raise UserError('请确认扫描是否正确%s' % code)
        if code[0] == '01':
            if code[1] == '10':
                invoice_type = 'dz'
                invoice_heck_code = code[6]
            if code[1] == '01':
                invoice_type = 'zy'
            if code[1] == '04':
                invoice_type = 'pt'
                invoice_heck_code = code[6]
            invoice_code = code[2]
            invoice_name = code[3]
            invoice_amount = round(float(code[4]), 2)
            invoice_date = code[5]
            invoice_tax = 0
        self.browse(order_id).write({
            'invoice_type': invoice_type, 'invoice_code': invoice_code, 'invoice_name': invoice_name,
            'invoice_amount': invoice_amount, 'invoice_tax': invoice_tax,
            'invoice_heck_code': invoice_heck_code, 'invoice_date': invoice_date,
        })

    def action_get_attachment_view(self):
        res = self.env.ref('base.action_attachment')
        res['domain'] = [('res_model', '=', 'hr.expense.line'), ('res_id', 'in', self.ids)]
        res['context'] = {'default_res_model': 'hr.expense.line', 'default_res_id': self.id}
        return res

    def _compute_attachment_number(self):
        attachment_data = self.env['ir.attachment'].read_group(
            [('res_model', '=', 'hr.expense.line'), ('res_id', 'in', self.ids)], ['res_id'], ['res_id'])
        attachment = dict((data['res_id'], data['res_id_count']) for data in attachment_data)
        for bill in self:
            bill.attachment_number = attachment.get(bill.id, 0)

    def unlink(self):
        for bill in self:
            if bill.state != 'draft':
                raise UserError('只能删除草稿状态的费用发票')
        super().unlink()

    @api.depends('order_id.state')
    def _compute_is_pay(self):
        for bill in self:
            if bill.order_id and bill.order_id.state == 'done':
                bill.is_pay = True
            else:
                bill.is_pay = False