1import math
 2
 3class Currency:
 4    def __init__(self, name, curs):
 5        self.name = name
 6        self.curs = curs
 7
 8    def __eq__(self, other):
 9        if not isinstance(other, Currency):
10            return NotImplemented
11        return self.name == other.name and self.curs == other.curs
12    
13class PoliticalParty:
14    def __init__(self, name, motto, members = None, preferred_currency : Currency = None):
15        self.name = name
16        self._motto = motto
17        self.members = members if members is not None else []
18        self.preferred_currency = preferred_currency
19    
20    @property
21    def motto(self):
22        return self._motto
23    
24    def convert_currency_to_voters(self, voters, currency: Currency):
25        quantity = math.floor(voters / currency.curs)
26        if self.preferred_currency is not None and currency == self.preferred_currency:
27            quantity *= 2
28        return quantity
29    
30    def __str__(self):
31        return self.name
32    
33    def __add__(self, other):
34        if isinstance(other, PoliticalParty):
35            return Coalition(self, other)
36        return NotImplemented
37    
38class Coalition:
39    def __init__(self, *parties : PoliticalParty):
40        self.parties = list(parties)
41
42    @property
43    def members(self):
44        res = {party.name: party.members for party in self.parties}
45        return res
46    
47    def __add__(self, other : PoliticalParty):
48        if isinstance(other, PoliticalParty):
49            return Coalition(*self.parties, other)
50        if isinstance(other, Coalition):
51            return Coalition(*self.parties, *other.parties)
52        return NotImplemented
53    
54    def __str__(self):
55        return "-".join(party.name for party in self.parties)
56    
57class Elections:
58    all_results = {}
59
60    def __init__(self, year):
61        self.year = year
62        if year not in Elections.all_results:
63            Elections.all_results[year] = {}
64        self.results = Elections.all_results[year]
65    
66    def register_party_or_coalition(self, party):
67        if party not in self.results:
68            self.results[party] = 0
69    
70    def vote(self, party):
71        if party in self.results:
72            self.results[party] += 1
73    
74    def rig_elections(self, party, quantity, currency):
75        votes = 0
76        if isinstance(party, PoliticalParty):
77            votes = party.convert_currency_to_voters(quantity, currency)
78        if isinstance(party, Coalition):
79            for p in party.parties:
80                votes = max(votes, p.convert_currency_to_voters(quantity, currency))
81        self.results[party] += votes
82    
83    def get_results(self):
84        res = {str(p): votes for p,votes in self.results.items()}
85        return res
86    
87    @classmethod
88    def get_results_by_year(cls, year):
89        res = {}
90        if year in cls.all_results:
91            for p, votes in cls.all_results[year].items():
92                res[str(p)] = votes
93        return res