Darsana Anker lottery

darsana_lottery.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-

from __future__ import print_function
import sys
import random
import time
import csv
import datetime

seed = time.time()


def uniq_choice(members, num):
    results = []
    for i in range(num):
        try:
            t = random.Random(seed).choice(members)
        except IndexError:
            sys.exit(1)
        results.append(t)
        members.remove(t)
    return results

if __name__ == '__main__':
    if len(sys.argv) < 3:
        print("darsana_lottery.py [Path to userlist csv] [Path to prizelist csv]")
        sys.exit(1)

    members_file = sys.argv[1]
    prizes_file = sys.argv[2]

    # Timestamp(seed) check
    print("Time based seed(%s) means datatime(%s)" % (
        seed,
        datetime.datetime.fromtimestamp(seed).strftime('%Y/%m/%d %H:%M:%S')))

    with open(members_file) as mb, open(prizes_file) as pz:
        members = [i for i in csv.DictReader(mb)]
        members_index = list(range(len(members)))

        # each line include Anker item name, num and num
        prizes = [line[:-1].split(",") for line in pz]

        type_ = (("*当選者*", 1), ("*当選予備*", 2))
        for i in prizes:
            print("*%s*" % i[0])  # item name
            for n, j in type_:
                print(n)
                results = uniq_choice(members_index, int(i[j]))
                for k in results:
                    print("%s,%s,%s" % (members[k]["name"],  # Agent name
                                        members[k]["gname"],  # Google+ name
                                        members[k]["url"]))  # Google+ url

members_sample.csv

prizes.csv