feature: add custom class support for war-related entities

Author: lukasthaler

coc/war_clans.py

@@ -66,6 +66,11 @@ class WarClan(BaseClan):
         :class:`int`: Total attacks used by clan this war.
     max_stars:
         :class:`int`: Total possible stars achievable.
+    total_attacks:
+        :class:`int`: Total possible number of attacks usable by clan this war
+    member_cls: :class:`coc.ClanMember`
+        The type which the members found in :attr:`Clan.members` will be of.
+        Ensure any overriding of this inherits from :class:`coc.ClanMember`.
     """
 
     __slots__ = (
@@ -76,6 +81,7 @@ class WarClan(BaseClan):
         "exp_earned",
         "max_stars",
         "total_attacks",
+        "member_cls",
         "_war",
         "_client",
 
@@ -86,9 +92,10 @@ class WarClan(BaseClan):
         "_cs_members",
     )
 
-    def __init__(self, *, data, client, war):
+    def __init__(self, *, data, client, war, **kwargs):
         self._war = war
         self._members = {}
+        self.member_cls = kwargs.pop('member_cls', ClanWarMember)
 
         super().__init__(data=data, client=client)
         self._from_data(data)
@@ -110,7 +117,7 @@ def _from_data(self, data: dict) -> None:
             self.total_attacks: int = (data_get("teamSize") or 0) * 3
 
         self._iter_members = (
-            ClanWarMember(data=mdata, client=self._client, war=self._war, clan=self)
+            self.member_cls(data=mdata, client=self._client, war=self._war, clan=self)
             for mdata in data_get("members", [])
         )
 

coc/war_members.py

@@ -52,6 +52,9 @@ class ClanWarMember(BasePlayer):
         :class:`ClanWar`: The current war this member is in.
     clan:
         :class:`WarClan`: The member's clan.
+    attack_cls: :class:`coc.WarAttack`
+        The type which the attacks in :attr:`ClanWarMember.attacks` will be of.
+        Ensure any overriding of this inherits from :class:`coc.WarAttack`.
     """
 
     __slots__ = (
@@ -65,16 +68,18 @@ class ClanWarMember(BasePlayer):
         "map_position",
         "war",
         "clan",
+        "attack_cls",
         "_client",
         "_attacks",
     )
 
-    def __init__(self, *, data, client, war, clan):
+    def __init__(self, *, data, client, war, clan, **kwargs):
         super().__init__(data=data, client=client)
         self._client = client
         self._attacks = []
         self.war = war  # type: ClanWar
         self.clan = clan  # type: WarClan
+        self.attack_cls = kwargs.pop('attack_cls', WarAttack)
         self._from_data(data)
 
     def _from_data(self, data):
@@ -87,7 +92,7 @@ def _from_data(self, data):
         self.defense_count: int = data_get("opponentAttacks")
 
         self.__iter_attacks = (
-            WarAttack(data=adata, client=self._client, war=self.war) for adata in data_get("attacks", [])
+            self.attack_cls(data=adata, client=self._client, war=self.war) for adata in data_get("attacks", [])
         )
 
         self._best_opponent_attacker: str = data_get("bestOpponentAttack", {}).get("attackerTag")

coc/wars.py

@@ -66,6 +66,9 @@ class ClanWar:
         :class:`ClanWarLeagueGroup`: The war's league group. This is ``None`` unless this is a Clan League War.
     attacks_per_member:
         :class:`int`: The number of attacks each member has this war.
+    clan_cls: :class:`WarClan`
+        the type `war.clan` and `war.opponent` will be of.
+        Ensure any overriding of this inherits from :class:`coc.WarClan`.
     """
 
     __slots__ = (
@@ -81,6 +84,7 @@ class ClanWar:
         "opponent",
         "league_group",
         "attacks_per_member",
+        "clan_cls",
         "_response_retry",
         "_raw_data"
     )
@@ -90,6 +94,7 @@ def __init__(self, *, data, client, **kwargs):
         self._client = client
         self._raw_data = data if client and client.raw_attribute else None
         self.clan_tag = kwargs.pop("clan_tag", None)
+        self.clan_cls = kwargs.pop('clan_cls', WarClan)
         self._from_data(data)
 
         self.clan_tag = self.clan and self.clan.tag or self.clan_tag
@@ -117,14 +122,14 @@ def _from_data(self, data: dict) -> None:
         # depending on the way the game stores it internally. This isn't very helpful as we always want it
         # from the perspective of the tag we provided, so switch them around if it isn't correct.
         if clan_data and clan_data.get("tag", self.clan_tag) == self.clan_tag:
-            self.clan = try_enum(WarClan, data=clan_data, client=self._client,
+            self.clan = try_enum(self.clan_cls, data=clan_data, client=self._client,
                                  war=self)
-            self.opponent = try_enum(WarClan, data=data_get("opponent"),
+            self.opponent = try_enum(self.clan_cls, data=data_get("opponent"),
                                      client=self._client, war=self)
         else:
-            self.clan = try_enum(WarClan, data=data_get("opponent"),
+            self.clan = try_enum(self.clan_cls, data=data_get("opponent"),
                                  client=self._client, war=self)
-            self.opponent = try_enum(WarClan, data=clan_data,
+            self.opponent = try_enum(self.clan_cls, data=clan_data,
                                      client=self._client, war=self)
 
     @property