#!/usr/bin/env python
# -*- coding: iso-8859-1 -*-

from csv import reader
from optparse import OptionParser

class LotteryNumber:
	rank = None
	lottery = None
	def __init__(self, rank, lottery):
		self.rank = rank
		self.lottery = lottery
	def __cmp__(self, other):
		if self.rank < other.rank:
			return -1
		elif self.rank > other.rank:
			return 1
		elif self.lottery > other.lottery:
			return -1
		elif self.lottery < other.lottery:
			return 1
		else:
			return 0
	def __str__(self):
		return "%s/%s" % (("%.4f" % (self.rank,)).rstrip('0').rstrip('.'), self.lottery)
	def __hash__(self):
		return (int)((40 - self.rank) * 100000000) + self.lottery

class Contestant:
	lotteryNumber = None
	lotteryClass = None
	name = None
	def __init__(self, lotteryNumber, lotteryClass, name):
		self.lotteryNumber = lotteryNumber
		self.lotteryClass = lotteryClass
		self.name = name
	def __cmp__(self, other):
		return cmp(self.lotteryNumber, other.lotteryNumber)
	def __str__(self):
		return self.name

class Suite:
	lotteryNumber = None
	contestants = None
	def __init__(self, lotteryNumber):
		self.lotteryNumber = lotteryNumber
		self.contestants = []
	def addContestant(self, contestant):
		if contestant.lotteryNumber == self.lotteryNumber:
			self.contestants.append(contestant)
	def __cmp__(self, other):
		if type(self) != type(other):
			return 1
		return cmp(self.lotteryNumber, other.lotteryNumber)
	def __len__(self):
		return len(self.contestants)
	def __str__(self):
		return ", ".join([str(contestant) for contestant in self.contestants])

class Lottery:
	contestants = []
	suites = {}
	def __init__(self, csvfile):
		data = reader(open(csvfile))
		for row in data:
			nextContestant = Contestant(LotteryNumber(float(row[2]), int(row[3])), row[1], row[0])
			self.contestants.append(nextContestant)
			if nextContestant.lotteryClass == "Suite":
				if nextContestant.lotteryNumber in self.suites:
					self.suites[nextContestant.lotteryNumber].addContestant(nextContestant)
				else:
					suite = Suite(nextContestant.lotteryNumber)
					suite.addContestant(nextContestant)
					self.suites[nextContestant.lotteryNumber] = suite
			nextContestant = None
	def printContestants(self):
		index = 1
		for contestant in sorted(self.contestants, reverse=True):
			print "%s: %s - %s in %s" % (index, contestant.lotteryNumber, contestant.name, contestant.lotteryClass)
			index += 1
	def printGeneral(self):
		index = 1
		for contestant in sorted(self.contestants, reverse=True):
			if contestant.lotteryClass != "General":
				continue
			print "%s: %s - %s" % (index, contestant.lotteryNumber, contestant.name)
			index += 1
	def printSuites(self):
		index = 1
		for suite in sorted(self.suites.values(), reverse=True):
			print "%s: %s - %s" % (index, suite.lotteryNumber, suite)
			index += 1
	def printSizedSuites(self, size):
		index = 1
		for suite in sorted(self.suites.values(), reverse=True):
			if len(suite) != size:
				continue
			print "%s: %s - %s" % (index, suite.lotteryNumber, suite)
			index += 1


parser = OptionParser(usage = "usage: %prog [options] csvfile")
parser.add_option("-g", "--general", action="store_true", help="Print general selection contestants.")
parser.add_option("-s", "--suite", type="int", help="Print suite selection contestants in suites of SIZE. If SIZE is 0, print all suites.", metavar="SIZE")
parser.add_option("-a", "--all", action="store_true", help="Print all contestants.")
(options, args) = parser.parse_args()
if len(args) != 1:
	parser.error("You must give 1 csv file to parse.")
try:
	lotto = Lottery(args[0])
except:
	parser.error("Could not parse csv.")
if options.all == True:
	lotto.printContestants()
if options.general == True:
	lotto.printGeneral()
if options.suite != None:
	if options.suite == 0:
		lotto.printSuites()
	else:
		lotto.printSizedSuites(options.suite)