mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
1ac36d03da
A lot of campaigns want to define custom factions. This allows them to do so without us having to fill the built-in factions list with a bunch of campaign-specific factions. It also makes custom campaigns more portable as they don't need to also distribute the custom faction files.
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
from abc import ABC, abstractmethod
|
|
from typing import Any, TYPE_CHECKING
|
|
|
|
from game.factions import Faction
|
|
|
|
if TYPE_CHECKING:
|
|
from game.factions.factions import Factions
|
|
|
|
|
|
class FactionRecommendation(ABC):
|
|
def __init__(self, name: str) -> None:
|
|
self.name = name
|
|
|
|
@abstractmethod
|
|
def register_campaign_specific_faction(self, factions: Factions) -> None:
|
|
...
|
|
|
|
@abstractmethod
|
|
def get_faction(self, factions: Factions) -> Faction:
|
|
...
|
|
|
|
@staticmethod
|
|
def from_field(
|
|
data: str | dict[str, Any] | None, player: bool
|
|
) -> FactionRecommendation:
|
|
if data is None:
|
|
name = "USA 2005" if player else "Russia 1990"
|
|
return BuiltinFactionRecommendation(name)
|
|
if isinstance(data, str):
|
|
return BuiltinFactionRecommendation(data)
|
|
return CampaignDefinedFactionRecommendation(Faction.from_dict(data))
|
|
|
|
|
|
class BuiltinFactionRecommendation(FactionRecommendation):
|
|
def register_campaign_specific_faction(self, factions: Factions) -> None:
|
|
pass
|
|
|
|
def get_faction(self, factions: Factions) -> Faction:
|
|
return factions.get_by_name(self.name)
|
|
|
|
|
|
class CampaignDefinedFactionRecommendation(FactionRecommendation):
|
|
def __init__(self, faction: Faction) -> None:
|
|
super().__init__(faction.name)
|
|
self.faction = faction
|
|
|
|
def register_campaign_specific_faction(self, factions: Factions) -> None:
|
|
factions.add_campaign_defined(self.faction)
|
|
|
|
def get_faction(self, factions: Factions) -> Faction:
|
|
return self.faction
|