if __name__ == "__main__": # start up the default logging logging.basicConfig(level=logging.DEBUG) # connect to mongodb try: connect('tweet-store') except: logging.critical('couldnt connect to mongodb!') sys.exit(-1) else: logging.info('connected to mongodb') while True: battles = Battle.objects(active = True) for battle in battles: if (battle.end - datetime.now()) <= timedelta(seconds=0): logging.info("battle '%s' ended!" % battle) battle.active = False battle.save() # find the winner max_votes = 0 max_choice = None total_votes = 0 for choice, voters in battle.choices.items(): num_votes = len(voters) total_votes += num_votes
from mongoengine import * from db.document import Battle from datetime import datetime import sys connect('tweet-store') print 'enter tag: ' tag = sys.stdin.readline().strip() print 'enter choices: ' c = sys.stdin.readline().split(',') d = {c[0].strip():[], c[1].strip():[]} b = Battle(tag = tag, start = datetime.now(), suggester = "someone", choices = d, active = True) b.save() b = Battle.objects.get(tag = tag) print b.tag, b.choices