diff --git a/printer_zpl2/tests/test_wizard_print_record_label.py b/printer_zpl2/tests/test_wizard_print_record_label.py index 874c7d9..5333294 100644 --- a/printer_zpl2/tests/test_wizard_print_record_label.py +++ b/printer_zpl2/tests/test_wizard_print_record_label.py @@ -39,6 +39,7 @@ class TestWizardPrintRecordLabel(TransactionCase): active_model='printing.printer', active_id=self.printer.id, active_ids=[self.printer.id], + printer_zpl2_id=self.printer.id, ) wizard = wizard_obj.create({}) self.assertEqual(wizard.printer_id, self.printer) @@ -87,6 +88,7 @@ class TestWizardPrintRecordLabel(TransactionCase): active_model='printing.printer', active_id=self.printer.id, active_ids=[self.printer.id], + printer_zpl2_id=self.printer.id, ) wizard = wizard_obj.create({}) self.assertEqual(wizard.label_id, self.label) diff --git a/printer_zpl2/wizard/print_record_label.py b/printer_zpl2/wizard/print_record_label.py index c1ad750..bf39b0f 100644 --- a/printer_zpl2/wizard/print_record_label.py +++ b/printer_zpl2/wizard/print_record_label.py @@ -23,7 +23,10 @@ class PrintRecordLabel(models.TransientModel): values = super(PrintRecordLabel, self).default_get(fields_list) # Automatically select the printer and label, if only one is available - printers = self.env['printing.printer'].search([]) + printers = self.env['printing.printer'].search( + [('id', '=', self.env.context.get('printer_zpl2_id'))]) + if not printers: + printers = self.env['printing.printer'].search([]) if len(printers) == 1: values['printer_id'] = printers.id