From a8f5ae145a20907d0f2828bf4ebd5adf1d1654fc Mon Sep 17 00:00:00 2001 From: nexy7574 Date: Fri, 5 Jul 2024 00:27:59 +0100 Subject: [PATCH] Properly report gains and losses --- src/cogs/election.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/src/cogs/election.py b/src/cogs/election.py index 7cb47f2..d35fb1e 100644 --- a/src/cogs/election.py +++ b/src/cogs/election.py @@ -113,7 +113,15 @@ class ElectionCog(commands.Cog): try: party, extra = span.get_text().strip().split(":", 1) seats, extra = extra.split(",", 1) - seats = change = int(seats.split()[0]) + extra = extra.strip() + if extra.lower() == "no change": + change = 0 + else: + _values = extra.split() + change = int(_values[0]) + if _values[-1] != "gained": + change *= -1 + seats = int(seats.split()[0]) except ValueError: self.log.error("failed to parse %r", span) continue