From 0ea144db17549b72dde938e736b730d6c315ac76 Mon Sep 17 00:00:00 2001 From: Eriks K Date: Mon, 30 Nov 2020 17:44:23 +0200 Subject: [PATCH] Added method to get division stats --- erepublik/access_points.py | 3 +++ erepublik/citizen.py | 5 +++++ 2 files changed, 8 insertions(+) diff --git a/erepublik/access_points.py b/erepublik/access_points.py index ef60329..1ac38a2 100644 --- a/erepublik/access_points.py +++ b/erepublik/access_points.py @@ -388,6 +388,9 @@ class ErepublikLocationAPI(CitizenBaseAPI): class ErepublikMilitaryAPI(CitizenBaseAPI): + def _get_military_battle_stats(self, battle_id: int, division: int, division_id: int): + return self.get(f"{self.url}/military/battle-stats/{battle_id}/{division}/{division_id}") + def _get_military_battlefield_choose_side(self, battle_id: int, side_id: int) -> Response: return self.get(f"{self.url}/military/battlefield-choose-side/{battle_id}/{side_id}") diff --git a/erepublik/citizen.py b/erepublik/citizen.py index f2200cb..bfe6857 100644 --- a/erepublik/citizen.py +++ b/erepublik/citizen.py @@ -2032,6 +2032,11 @@ class CitizenMilitary(CitizenTravel): return (r_json.get(str(battle.invader.id)).get("fighterData"), r_json.get(str(battle.defender.id)).get("fighterData")) + def get_battle_division_stats(self, division: classes.BattleDivision) -> Dict[str, Any]: + battle = division.battle + r = self._get_military_battle_stats(battle.id, division.div, division.id) + return r.json() + def schedule_attack(self, war_id: int, region_id: int, region_name: str, at_time: datetime): if at_time: self.sleep(utils.get_sleep_seconds(at_time))