init args change

This commit is contained in:
Walter 2020-10-30 16:42:48 -05:00 committed by Dan Albert
parent d519dfa5da
commit 65c85d7f0b
2 changed files with 6 additions and 5 deletions

View File

@ -159,7 +159,7 @@ class MissionInfoGenerator:
Examples of subtypes include briefing generators, kneeboard generators, etc.
"""
def __init__(self, mission: Mission, game: Optional['Game'] = None) -> None:
def __init__(self, mission: Mission, game: 'Game') -> None:
self.mission = mission
self.game = game
self.awacs: List[AwacsInfo] = []

View File

@ -26,7 +26,7 @@ import datetime
from collections import defaultdict
from dataclasses import dataclass
from pathlib import Path
from typing import Dict, List, Optional, Tuple
from typing import Dict, List, Optional, Tuple, TYPE_CHECKING
from PIL import Image, ImageDraw, ImageFont
from dcs.mission import Mission
@ -42,7 +42,8 @@ from .flights.flight import FlightWaypoint, FlightWaypointType
from .radios import RadioFrequency
from .runways import RunwayData
if TYPE_CHECKING:
from game import Game
class KneeboardPageWriter:
"""Creates kneeboard images."""
@ -310,8 +311,8 @@ class BriefingPage(KneeboardPage):
class KneeboardGenerator(MissionInfoGenerator):
"""Creates kneeboard pages for each client flight in the mission."""
def __init__(self, mission: Mission) -> None:
super().__init__(mission)
def __init__(self, mission: Mission, game: 'Game') -> None:
super().__init__(mission, game)
def generate(self) -> None:
"""Generates a kneeboard per client flight."""