| # |
| # Licensed to the Apache Software Foundation (ASF) under one or more |
| # contributor license agreements. See the NOTICE file distributed with |
| # this work for additional information regarding copyright ownership. |
| # The ASF licenses this file to You under the Apache License, Version 2.0 |
| # (the "License"); you may not use this file except in compliance with |
| # the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| # |
| """ D'Hondt (Jefferson) Based Voting Plugin """ |
| import re, json, random |
| |
| from lib import constants |
| |
| def validateDH(vote, issue): |
| "Tries to validate a vote, returns why if not valid, None otherwise" |
| letters = [chr(i) for i in range(ord('a'), ord('a') + len(issue['candidates']))] |
| if len(vote) > 1: |
| return "Vote may only contain one letter!" |
| for char in vote: |
| if char not in letters: |
| return "Invalid characters in vote. Accepted are: %s" % ", ".join(letters) |
| return None |
| |
| |
| def tallyDH(votes, issue): |
| m = re.match(r"dh(\d+)", issue['type']) |
| if not m: |
| raise Exception("Not a D'Hondt vote!") |
| |
| numseats = int(m.group(1)) |
| candidates = [] |
| for c in issue['candidates']: |
| candidates.append(c['name']) |
| |
| |
| debug = [] |
| |
| # Set up letters for mangling |
| letters = [chr(i) for i in range(ord('a'), ord('a') + len(candidates))] |
| cc = "".join(letters) |
| |
| # Set up seats won |
| winners = [] |
| |
| # Set up vote matrix |
| matrix = {} |
| for key in votes: |
| vote = votes[key] |
| if not vote in matrix: |
| matrix[vote] = [0,1] |
| matrix[vote][0] += 1 |
| |
| # Start counting |
| while len(winners) < numseats: |
| m = [] |
| for c in matrix: |
| quotient = (matrix[c][0]/matrix[c][1]) |
| m.append(quotient) |
| for c in matrix: |
| quotient = (matrix[c][0]/matrix[c][1]) |
| if quotient == max(m): |
| winners.append(c) |
| matrix[c][1] += 1 |
| break |
| |
| # Compile list of winner names |
| winnernames = [] |
| for c in winners: |
| i = ord(c) - ord('a') |
| winnernames.append(candidates[i]) |
| |
| # Return the data |
| return { |
| 'votes': len(votes), |
| 'winners': winners, |
| 'winnernames': winnernames, |
| }, """ |
| Winners: |
| - %s |
| """ % "\n - ".join(winnernames) |
| |
| |
| constants.appendVote ( |
| { |
| 'key': "dh1", |
| 'description': "D'Hondt Election with 1 seat", |
| 'category': 'dh', |
| 'validate_func': validateDH, |
| 'vote_func': None, |
| 'tally_func': tallyDH |
| }, |
| ) |
| |
| # Add ad nauseam |
| for i in range(2,constants.MAX_NUM+1): |
| constants.appendVote ( |
| { |
| 'key': "dh%u" % i, |
| 'description': "D'Hondt Election with %u seats" % i, |
| 'category': 'dh', |
| 'validate_func': validateDH, |
| 'vote_func': None, |
| 'tally_func': tallyDH |
| }, |
| ) |