Fix type annotations for some UI code.

This commit is contained in:
Dan Albert 2021-06-19 15:03:39 -07:00
parent 703c68eb66
commit 9940dc8451
2 changed files with 6 additions and 4 deletions

View File

@ -1,9 +1,10 @@
"""Combo box for selecting a departure airfield.""" """Combo box for selecting a departure airfield."""
from typing import Iterable, Type from typing import Iterable
from PySide2.QtWidgets import QComboBox from PySide2.QtWidgets import QComboBox
from dcs.unittype import FlyingType from dcs.unittype import FlyingType
from game.dcs.aircrafttype import AircraftType
from game.theater.controlpoint import ControlPoint from game.theater.controlpoint import ControlPoint
@ -17,7 +18,7 @@ class QArrivalAirfieldSelector(QComboBox):
def __init__( def __init__(
self, self,
destinations: Iterable[ControlPoint], destinations: Iterable[ControlPoint],
aircraft: Type[FlyingType], aircraft: AircraftType,
optional_text: str, optional_text: str,
) -> None: ) -> None:
super().__init__() super().__init__()

View File

@ -1,10 +1,11 @@
"""Combo box for selecting a departure airfield.""" """Combo box for selecting a departure airfield."""
from typing import Iterable, Type from typing import Iterable
from PySide2.QtCore import Signal from PySide2.QtCore import Signal
from PySide2.QtWidgets import QComboBox from PySide2.QtWidgets import QComboBox
from dcs.unittype import FlyingType from dcs.unittype import FlyingType
from game.dcs.aircrafttype import AircraftType
from game.inventory import GlobalAircraftInventory from game.inventory import GlobalAircraftInventory
from game.theater.controlpoint import ControlPoint from game.theater.controlpoint import ControlPoint
@ -22,7 +23,7 @@ class QOriginAirfieldSelector(QComboBox):
self, self,
global_inventory: GlobalAircraftInventory, global_inventory: GlobalAircraftInventory,
origins: Iterable[ControlPoint], origins: Iterable[ControlPoint],
aircraft: Type[FlyingType], aircraft: AircraftType,
) -> None: ) -> None:
super().__init__() super().__init__()
self.global_inventory = global_inventory self.global_inventory = global_inventory