from odoo import models, fields

class PosOrder(models.Model):
    _inherit = 'pos.order'

    creator_id = fields.Many2one('hr.employee', string='Order Taker', help='Employee who took the order')
    payer_id = fields.Many2one('hr.employee', string='Cashier', help='Employee who processed the payment')
    employee_id = fields.Many2one('hr.employee', string='Cash Open by', help='Employee who opened the cash register')

    def _export_for_ui(self, order):
        result = super()._export_for_ui(order)
        result['creator_id'] = order.creator_id.id
        result['payer_id'] = order.payer_id.id
        return result

    def _get_fields_for_draft_order(self):
        fields = super()._get_fields_for_draft_order()
        fields.extend(['creator_id', 'payer_id'])
        return fields