mirror of https://github.com/dirtbags/moth.git
64 lines
1.1 KiB
Python
64 lines
1.1 KiB
Python
|
#! /usr/bin/python3
|
||
|
|
|
||
|
|
import sys
|
||
|
|
|
||
|
|
limits = {
|
||
|
|
"net": 100,
|
||
|
|
"mal": 60,
|
||
|
|
"hst": 60,
|
||
|
|
"icc": 40,
|
||
|
|
"nil": 9000,
|
||
|
|
":-(": 9000,
|
||
|
|
}
|
||
|
|
|
||
|
|
# Read in allowed substring list
|
||
|
|
allowed = []
|
||
|
|
for line in open("approved.txt"):
|
||
|
|
allowed.append(line.strip())
|
||
|
|
|
||
|
|
# Read in registration data
|
||
|
|
registrants = []
|
||
|
|
regs = {}
|
||
|
|
for line in open("reg.txt"):
|
||
|
|
line = line.strip('\n')
|
||
|
|
ok = False
|
||
|
|
for a in allowed:
|
||
|
|
if a in line:
|
||
|
|
ok = True
|
||
|
|
break
|
||
|
|
name, email, org, c1, c2, _ = line.split('\t')
|
||
|
|
if not ok:
|
||
|
|
c1 = c2 = "nil"
|
||
|
|
|
||
|
|
if email not in registrants:
|
||
|
|
registrants.append(email)
|
||
|
|
regs[email] = (name, org, c1, c2)
|
||
|
|
|
||
|
|
# Divvy out classes
|
||
|
|
which = {}
|
||
|
|
counts = [0, 0, 0]
|
||
|
|
for email in registrants:
|
||
|
|
r = regs.get(email)
|
||
|
|
name, org, c1, c2 = regs[email]
|
||
|
|
c = [c1, c2, ":-("]
|
||
|
|
|
||
|
|
for i in range(3):
|
||
|
|
w = c[i]
|
||
|
|
if limits[w] > 0:
|
||
|
|
which[email] = i
|
||
|
|
limits[w] -= 1
|
||
|
|
counts[i] += 1
|
||
|
|
break
|
||
|
|
|
||
|
|
try:
|
||
|
|
oldreg = open(email).read()
|
||
|
|
except:
|
||
|
|
oldreg = None
|
||
|
|
|
||
|
|
if oldreg != w:
|
||
|
|
print(w, email)
|
||
|
|
open(email, "w").write(w)
|
||
|
|
|
||
|
|
print("%d got 1st choice, %d got 2nd choice, %d got screwed" %
|
||
|
|
(counts[0], counts[1], counts[2]))
|