Add typing for _getPrinterOutputDevices

Then my IDE will give hints on where a printer of the wrong type is being added.

Contributes to issue CURA-5061.
This commit is contained in:
Ghostkeeper 2018-03-08 14:23:14 +01:00
parent 2a10c9a2d7
commit 375770818b
No known key found for this signature in database
GPG key ID: 5252B696FB5E7C7A

View file

@ -1,3 +1,8 @@
# Copyright (c) 2018 Ultimaker B.V.
# Cura is released under the terms of the LGPLv3 or higher.
from typing import List
from cura.MachineAction import MachineAction
from cura.PrinterOutputDevice import PrinterOutputDevice
@ -32,7 +37,7 @@ class BedLevelMachineAction(MachineAction):
printer_output_devices[0].moveHead(0, 0, 3)
printer_output_devices[0].homeHead()
def _getPrinterOutputDevices(self):
def _getPrinterOutputDevices(self) -> List[PrinterOutputDevice]:
return [printer_output_device for printer_output_device in Application.getInstance().getOutputDeviceManager().getOutputDevices() if isinstance(printer_output_device, PrinterOutputDevice)]
@pyqtSlot()